hypha-rpc 0.20.59 → 0.20.60

package/dist/hypha-rpc-websocket.js

@@ -566,8 +566,24 @@ class RPC extends _utils__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
     this._services = {};
   }
 
-  async disconnect() {
+  close() {
+    // Clear all heartbeat intervals
+    for (const session_id in this._object_store) {
+      if (this._object_store.hasOwnProperty(session_id)) {
+        const session = this._object_store[session_id];
+        if (session && session.heartbeat_task) {
+          clearInterval(session.heartbeat_task);
+        }
+        if (session && session.timer) {
+          session.timer.clear();
+        }
+      }
+    }
     this._fire("disconnected");
+  }
+
+  async disconnect() {
+    this.close();
     await this._connection.disconnect();
   }
 
@@ -1402,6 +1418,13 @@ class RPC extends _utils__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
             }
           }
           heartbeat_task = setInterval(heartbeat, promise.interval * 1000);
+          // Store the heartbeat task in the session store for cleanup
+          if (data.session) {
+            const session_store = this._get_session_store(data.session, false);
+            if (session_store) {
+              session_store.heartbeat_task = heartbeat_task;
+            }
+          }
         }
       }
 
@@ -1528,6 +1551,12 @@ class RPC extends _utils__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
         method.apply(null, args);
         clearInterval(heartbeat_task);
       }
+      if (data.session) {
+        const session_store = this._get_session_store(data.session, false);
+        if (session_store) {
+          delete session_store.heartbeat_task;
+        }
+      }
     } catch (err) {
       if (reject) {
         reject(err);