fix(ws): disconnect ws before reconnect

vitelabs · Jul 26, 2020 · 54f02f9 · 54f02f9
1 parent a3f12d9
commit 54f02f9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/WS/index.js b/src/WS/index.js
@@ -24,6 +24,8 @@ class WsRpc extends IPC_WS {
         this.headers = options.headers;
         this.clientConfig = options.clientConfig;
 
+        this._timeout = null;
+
         this.reconnect();
 
         // Try to reconnect.
@@ -35,14 +37,15 @@ class WsRpc extends IPC_WS {
             if (times > options.retryTimes) {
                 return;
             }
-            setTimeout(() => {
+            this._timeout = setTimeout(() => {
                 times++;
                 this.reconnect();
             }, options.retryInterval);
         });
     }
 
     reconnect() {
+        this.disconnect();
         this.socket = new Websocket(this.path, this.protocol, null, this.headers, null, this.clientConfig);
         this.socket.onopen = () => {
             (this.socket.readyState === this.socket.OPEN) && this._connected();
@@ -61,6 +64,8 @@ class WsRpc extends IPC_WS {
 
     disconnect() {
         this.socket && this.socket.close && this.socket.close();
+        clearTimeout(this._timeout);
+        this.socket = null;
     }
 }