fix: ensure table data reset successfully

ajthinking · Sep 27, 2024 · d458831 · d458831
1 parent 7f2222c
commit d458831
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/packages/ui/src/components/DataStory/clients/WorkspaceSocketClient.tsx b/packages/ui/src/components/DataStory/clients/WorkspaceSocketClient.tsx
@@ -14,7 +14,7 @@ export class WorkspaceSocketClient implements WorkspaceApiClient {
   private maxReconnectTries = 100;
   private reconnectTimeoutMs = 1000;
   private updateEdgeCounts?: ClientRunParams['updateEdgeCounts'];
-  private observers: ServerClientObservationConfig | undefined;
+  private observers: ServerClientObservationConfig|undefined;
 
   constructor() {
     this.socket$ = webSocket({
@@ -47,6 +47,7 @@ export class WorkspaceSocketClient implements WorkspaceApiClient {
       filter(data => data.type === 'NotifyObservers'),
       clientBuffer()
     ).subscribe((data) => {
+      console.log('workspaceSocketClient: NotifyObservers', data);
       this?.observers?.onDataChange(
         data.items,
         data.inputObservers,
@@ -65,6 +66,10 @@ export class WorkspaceSocketClient implements WorkspaceApiClient {
       inputObservers: observers?.inputObservers || [],
     };
 
+    eventManager.emit({
+      type: DataStoryEvents.RUN_START
+    });
+
     this.socketSendMsg(message);
   }
 

diff --git a/packages/vscode-extension/src/app/VsCodeClient.tsx b/packages/vscode-extension/src/app/VsCodeClient.tsx
@@ -27,6 +27,10 @@ export class VsCodeClient implements WorkspaceApiClient {
       inputObservers: observers?.inputObservers || [],
     };
 
+    eventManager.emit({
+      type: DataStoryEvents.RUN_START
+    });
+
     this.sendMessage(message);
   }
 
@@ -143,4 +147,4 @@ export async function waitForResponse(params: any): Promise<any> {
       }
     }, 10000);
   });
-}
+}