fix(ws): add destory method to WS Provider

vitelabs · Jul 26, 2020 · dfcf417 · dfcf417
1 parent 54f02f9
commit dfcf417
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/WS/index.js b/src/WS/index.js
@@ -25,6 +25,7 @@ class WsRpc extends IPC_WS {
         this.clientConfig = options.clientConfig;
 
         this._timeout = null;
+        this._destroyed = false;
 
         this.reconnect();
 
@@ -45,6 +46,7 @@ class WsRpc extends IPC_WS {
     }
 
     reconnect() {
+        if (this._destroyed) return;
         this.disconnect();
         this.socket = new Websocket(this.path, this.protocol, null, this.headers, null, this.clientConfig);
         this.socket.onopen = () => {
@@ -67,6 +69,14 @@ class WsRpc extends IPC_WS {
         clearTimeout(this._timeout);
         this.socket = null;
     }
+
+    destroy() {
+        this.disconnect();
+        this.remove('error');
+        this.remove('close');
+        this.remove('connect');
+        this._destroyed = true;
+    }
 }
 
 export const WS_RPC = WsRpc;