fix: unsubscribe from RpcBrowser

containers · Jan 16, 2024 · d55d9bd · d55d9bd
1 parent 40aaf8c
commit d55d9bd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/packages/frontend/src/stores/playground-queries.ts b/packages/frontend/src/stores/playground-queries.ts
@@ -6,8 +6,11 @@ import { MSG_NEW_PLAYGROUND_QUERIES_STATE } from '@shared/Messages';
 import { studioClient } from '/@/utils/client';
 
 export const playgroundQueries: Readable<QueryState[]> = readable<QueryState[]>([], (set) => {  
-  rpcBrowser.subscribe(MSG_NEW_PLAYGROUND_QUERIES_STATE, (msg) => {
+  const sub = rpcBrowser.subscribe(MSG_NEW_PLAYGROUND_QUERIES_STATE, (msg) => {
     set(msg);
   });
   studioClient.askPlaygroundQueries();
+  return () => {
+    sub.unsubscribe();
+  }
 });
diff --git a/packages/shared/MessageProxy.ts b/packages/shared/MessageProxy.ts
@@ -83,6 +83,10 @@ export class RpcExtension {
   }
 }
 
+export interface Subscriber {
+  unsubscribe(): void;
+}
+
 export class RpcBrowser {
   counter: number = 0;
   promises: Map<number, {resolve: (value: unknown) => any, reject: (value: unknown) => void}> = new Map();
@@ -166,10 +170,14 @@ export class RpcBrowser {
     })
   }
 
-  // TODO(feloy) need to unsubscribe
   // TODO(feloy) need to subscribe several times?
-  subscribe(msgId: string, f: (msg: any) => void) {
+  subscribe(msgId: string, f: (msg: any) => void): Subscriber {
     this.subscribers.set(msgId, f);
+    return {
+      unsubscribe: () => {
+        this.subscribers.delete(msgId);
+      } 
+    }
   }
 
   isSubscribedMessage(content: any): content is ISubscribedMessage {