(박서영) 설계 Document Numbering Rule 수정

1 files changed, 5 insertions, 2 deletions

diff --git a/lib/docu-list-rule/document-class/table/document-class-table.tsx b/lib/docu-list-rule/document-class/table/document-class-table.tsx
index bbe79800..e3daac8a 100644
--- a/lib/docu-list-rule/document-class/table/document-class-table.tsx
+++ b/lib/docu-list-rule/document-class/table/document-class-table.tsx
@@ -1,6 +1,7 @@
 "use client"
 
 import * as React from "react"
+import { useRouter } from "next/navigation"
 import { useDataTable } from "@/hooks/use-data-table"
 import { DataTable } from "@/components/data-table/data-table"
 import { DataTableAdvancedToolbar } from "@/components/data-table/data-table-advanced-toolbar"
@@ -18,13 +19,15 @@ interface DocumentClassTableProps {
 }
 
 export function DocumentClassTable({ promises }: DocumentClassTableProps) {
+  const router = useRouter()
   const rawData = React.use(promises!)
   const [rowAction, setRowAction] = React.useState<DataTableRowAction<typeof documentClasses.$inferSelect> | null>(null)
   const [selectedDocumentClass, setSelectedDocumentClass] = React.useState<typeof documentClasses.$inferSelect | null>(null)
 
   const refreshData = React.useCallback(() => {
-    window.location.reload()
-  }, [])
+    // 전체 페이지 새로고침 대신 router.refresh() 사용 (성능 개선)
+    router.refresh()
+  }, [router])
 
   const columns = React.useMemo(() => getColumns({ setRowAction }), [setRowAction])