update success tip when create task info return

actiontech · Nov 15, 2023 · bfc9eb5 · bfc9eb5
1 parent 3eec557
commit bfc9eb5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/page/SqlAuditRecord/Create/index.tsx b/src/page/SqlAuditRecord/Create/index.tsx
@@ -16,7 +16,7 @@ import { useCurrentProjectName } from '../../ProjectManage/ProjectDetail';
 import AuditResult from '../../Order/AuditResult';
 import sql_audit_record from '../../../api/sql_audit_record';
 import { ICreateSQLAuditRecordV1Params } from '../../../api/sql_audit_record/index.d';
-import { useRef, useState } from 'react';
+import { useRef, useState, useEffect } from "react";
 import { IAuditTaskResV1, ISQLAuditRecordResData } from '../../../api/common';
 import { ResponseCode } from '../../../data/common';
 import EmptyBox from '../../../components/EmptyBox';
@@ -65,8 +65,6 @@ const SQLAuditCreate: React.FC = () => {
           return updateTags(res.data.data, baseValues);
         } else {
           setTask(res.data.data.task);
-          message.success(t('sqlAudit.create.SQLInfo.successTips'));
-          scrollToAuditResult();
         }
       }
     });
@@ -85,8 +83,6 @@ const SQLAuditCreate: React.FC = () => {
       .then((res) => {
         if (res.data.code === ResponseCode.SUCCESS) {
           setTask(record.task);
-          message.success(t('sqlAudit.create.SQLInfo.successTips'));
-          scrollToAuditResult();
         }
       });
   };
@@ -97,6 +93,13 @@ const SQLAuditCreate: React.FC = () => {
     setTask(undefined);
   };
 
+  useEffect(() => {
+    if (typeof task?.task_id === 'number') {
+      message.success(t("sqlAudit.create.SQLInfo.successTips"));
+      scrollToAuditResult();
+    }
+  }, [task, t]);
+
   return (
     <>
       <PageHeader title={t('sqlAudit.create.title')} ghost={false}>