fix: persist nodeId between WS reconnections

paipe-labs · Mar 31, 2024 · 5fed471 · 5fed471
1 parent 83aac06
commit 5fed471
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/client-package/src/SessionManager.ts b/client-package/src/SessionManager.ts
@@ -45,13 +45,16 @@ export class SessionManager {
   private _tasksManager: TasksManager;
   private _inferenceServer: InferenceServer;
   private _ws?: WebSocket | WebSocketNode;
-
+  private _nodeId: string;
+
   constructor(options: SessionManagerOptions) {
     const { backendServerWebSocketUrl, inferenceServerUrl, inferenceServerType } = options;
 
     this._inferenceServerUrl = inferenceServerUrl;
     this._backendServerWebSocketUrl = backendServerWebSocketUrl;
 
+    this._nodeId = uuidv4();
+
     switch (inferenceServerType) {
       case 'automatic':
         this._inferenceServer = new AutomaticInferenceServer({ inferenceServerUrl });
@@ -102,7 +105,7 @@ export class SessionManager {
     const { _ws: ws } = this;
 
     const onSocketOpen = () => {
-      ws.send(JSON.stringify({ type: 'register', node_id: uuidv4() }));
+      ws.send(JSON.stringify({ type: 'register', node_id: this._nodeId }));
       console.log('Server WebSocket connection opened');
     };