fix: add worker message trigger

2026-01-20 10:36:38 +08:00
parent b4f69a4e2e
commit 9d13b27c9e
5 changed files with 30 additions and 23 deletions
@@ -60,8 +60,7 @@ const AddWorker: React.FC<AddWorkerProps> = (props) => {
    stepList = []
  } = props || {};
  const intl = useIntl();
-  const { addedCount, createModelsChunkRequest, chunkRequestRef } =
-    useAddWorkerMessage();
+  const { addedCount, createModelsChunkRequest, reset } = useAddWorkerMessage();
  const firstLoad = React.useRef(true);
  const axiosTokenRef = React.useRef<any>(null);
  const [registrationInfo, setRegistrationInfo] = React.useState<{
@@ -100,8 +99,8 @@ const AddWorker: React.FC<AddWorkerProps> = (props) => {
    }
    return () => {
      firstLoad.current = true;
-      chunkRequestRef.current?.current?.cancel?.();
      axiosTokenRef.current?.cancel?.();
+      reset();
    };
  }, [open, cluster_id]);

@@ -109,7 +108,7 @@ const AddWorker: React.FC<AddWorkerProps> = (props) => {
    if (open) {
      createModelsChunkRequest();
    } else {
-      chunkRequestRef.current?.current?.cancel?.();
+      reset();
      axiosTokenRef.current?.cancel?.();
    }
  }, [open]);