From 22035d1383f55f0200e4609bde5b9a6925ec756a Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Thu, 12 Dec 2024 17:00:04 +0800
Subject: [PATCH] chore: fetch model list after deleting action

---
 src/hooks/use-update-chunk-list.ts           | 9 ++++-----
 src/pages/llmodels/components/table-list.tsx | 4 ++++
 src/pages/llmodels/index.tsx                 | 8 +++++---
 3 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/src/hooks/use-update-chunk-list.ts b/src/hooks/use-update-chunk-list.ts
index 074aed8c..62272d45 100644
--- a/src/hooks/use-update-chunk-list.ts
+++ b/src/hooks/use-update-chunk-list.ts
@@ -27,11 +27,10 @@ export function useUpdateChunkedList(options: {
   }, [options.dataList]);
 
   const debounceUpdateChunckedList = () => {
-    // clearTimeout(timerRef.current);
-    // timerRef.current = setTimeout(() => {
-    //   options.setDataList?.([...cacheDataListRef.current]);
-    // }, 80);
-    options.setDataList?.([...cacheDataListRef.current]);
+    clearTimeout(timerRef.current);
+    timerRef.current = setTimeout(() => {
+      options.setDataList?.([...cacheDataListRef.current]);
+    }, 200);
   };
   const updateChunkedList = (data: ChunkedCollection) => {
     console.log('updateChunkedList=====', {
diff --git a/src/pages/llmodels/components/table-list.tsx b/src/pages/llmodels/components/table-list.tsx
index d8e6def1..2a09b5b8 100644
--- a/src/pages/llmodels/components/table-list.tsx
+++ b/src/pages/llmodels/components/table-list.tsx
@@ -55,6 +55,7 @@ interface ModelsProps {
   handleNameChange: (e: any) => void;
   handleShowSizeChange?: (page: number, size: number) => void;
   handlePageChange: (page: number, pageSize: number | undefined) => void;
+  handleDeleteSuccess: () => void;
   queryParams: {
     page: number;
     perPage: number;
@@ -71,6 +72,7 @@ const Models: React.FC<ModelsProps> = ({
   handleNameChange,
   handleSearch,
   handlePageChange,
+  handleDeleteSuccess,
   dataSource,
   gpuDeviceList,
   workerList,
@@ -347,6 +349,7 @@ const Models: React.FC<ModelsProps> = ({
         await deleteModel(row.id);
         removeExpandedRowKey([row.id]);
         rowSelection.removeSelectedKey(row.id);
+        handleDeleteSuccess();
       }
     });
   };
@@ -358,6 +361,7 @@ const Models: React.FC<ModelsProps> = ({
         await handleBatchRequest(rowSelection.selectedRowKeys, deleteModel);
         rowSelection.clearSelections();
         removeExpandedRowKey(rowSelection.selectedRowKeys);
+        handleDeleteSuccess();
       }
     });
   };
diff --git a/src/pages/llmodels/index.tsx b/src/pages/llmodels/index.tsx
index ca3b3b1a..4c94dcd9 100644
--- a/src/pages/llmodels/index.tsx
+++ b/src/pages/llmodels/index.tsx
@@ -87,7 +87,6 @@ const Models: React.FC = () => {
       const res: any = await queryModelsList(params, {
         cancelToken: axiosToken.token
       });
-
       setDataSource({
         dataList: res.items || [],
         loading: false,
@@ -173,8 +172,10 @@ const Models: React.FC = () => {
 
   useEffect(() => {
     if (!firstLoad) {
-      createModelsChunkRequest();
-      createModelsInstanceChunkRequest();
+      setTimeout(() => {
+        createModelsChunkRequest();
+        createModelsInstanceChunkRequest();
+      }, 100);
     }
     return () => {
       chunkRequedtRef.current?.current?.cancel?.();
@@ -205,6 +206,7 @@ const Models: React.FC = () => {
         handleNameChange={handleNameChange}
         handleSearch={handleSearch}
         handlePageChange={handlePageChange}
+        handleDeleteSuccess={fetchData}
         queryParams={queryParams}
         loading={dataSource.loading}
         total={dataSource.total}