From e42892f9a5c3c4dd3cf97e88c79c8926bb44b5f4 Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Thu, 14 Dec 2023 10:47:51 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=20=E6=96=87=E6=A1=A3=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/document.ts          | 25 ++++++++++++++++++++++++-
 ui/src/views/document/index.vue | 29 +++++++++++++++++++++++------
 2 files changed, 47 insertions(+), 7 deletions(-)

diff --git a/ui/src/api/document.ts b/ui/src/api/document.ts
index 2080c562e..4e16522cf 100644
--- a/ui/src/api/document.ts
+++ b/ui/src/api/document.ts
@@ -121,6 +121,28 @@ const getDocumentDetail: (dataset_id: string, document_id: string) => Promise<Re
   return get(`${prefix}/${dataset_id}/document/${document_id}`)
 }
 
+/**
+ * 刷新文档向量库
+ * @param 参数 
+ * dataset_id, document_id, 
+ * {
+      "name": "string",
+      "is_active": true
+    }
+ */
+const putDocumentRefresh: (
+  dataset_id: string,
+  document_id: string,
+  loading?: Ref<boolean>
+) => Promise<Result<any>> = (dataset_id, document_id, loading) => {
+  return put(
+    `${prefix}/${dataset_id}/document/${document_id}/refresh`,
+    undefined,
+    undefined,
+    loading
+  )
+}
+
 export default {
   postSplitDocument,
   getDocument,
@@ -129,5 +151,6 @@ export default {
   putDocument,
   delDocument,
   getDocumentDetail,
-  listSplitPattern
+  listSplitPattern,
+  putDocumentRefresh
 }
diff --git a/ui/src/views/document/index.vue b/ui/src/views/document/index.vue
index 27c58794d..df1596111 100644
--- a/ui/src/views/document/index.vue
+++ b/ui/src/views/document/index.vue
@@ -80,14 +80,14 @@
           </el-table-column>
           <el-table-column label="操作" align="center">
             <template #default="{ row }">
-              <span v-if="row.status === 2">
+              <span v-if="row.status === '2'" class="mr-4">
                 <el-tooltip effect="dark" content="刷新" placement="top">
-                  <el-button type="primary" text>
+                  <el-button type="primary" text @click.stop="refreshDocument(row)">
                     <el-icon><RefreshRight /></el-icon>
                   </el-button>
                 </el-tooltip>
               </span>
-              <span class="ml-4">
+              <span>
                 <el-tooltip effect="dark" content="删除" placement="top">
                   <el-button type="primary" text @click.stop="deleteDocument(row)">
                     <el-icon><Delete /></el-icon>
@@ -102,7 +102,7 @@
   </LayoutContainer>
 </template>
 <script setup lang="ts">
-import { ref, onMounted, reactive, computed } from 'vue'
+import { ref, onMounted, reactive, watch } from 'vue'
 import { useRouter, useRoute } from 'vue-router'
 import documentApi from '@/api/document'
 import { numberFormat } from '@/utils/utils'
@@ -125,6 +125,23 @@ const paginationConfig = reactive({
   total: 0
 })
 
+watch(documentData, (list) => {
+  let interval
+  if (list.every((item) => item.status === '0')) {
+    interval = setInterval(() => {
+      getList(true)
+    }, 6000)
+  } else {
+    clearInterval(interval)
+  }
+})
+
+function refreshDocument(row: any) {
+  documentApi.putDocumentRefresh(row.dataset_id, row.id).then((res) => {
+    getList()
+  })
+}
+
 function rowClickHandle(row: any) {
   router.push({ path: `/dataset/${id}/${row.id}` })
 }
@@ -215,13 +232,13 @@ function handleSizeChange() {
   getList()
 }
 
-function getList() {
+function getList(bool?: boolean) {
   documentApi
     .getDocument(
       id as string,
       paginationConfig,
       filterText.value && { name: filterText.value },
-      loading
+      bool ? undefined : loading
     )
     .then((res) => {
       documentData.value = res.data.records