Added destroy to WebSocketProvider.

ethers-io · Jul 12, 2020 · d0a79c6 · d0a79c6
1 parent c660176
commit d0a79c6
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/packages/providers/src.ts/websocket-provider.ts b/packages/providers/src.ts/websocket-provider.ts
@@ -274,4 +274,22 @@ export class WebSocketProvider extends JsonRpcProvider {
             this.send("eth_unsubscribe", [ subId ]);
         });
     }
+
+    async destroy(): Promise<void> {
+        // Wait until we have connected before trying to disconnect
+        if (this._websocket.readyState === WebSocket.CONNECTING) {
+            await new Promise((resolve) => {
+                this._websocket.on("open", () => {
+                    resolve(true);
+                });
+
+                this._websocket.on("error", () => {
+                    resolve(false);
+                });
+            });
+        }
+
+        // Hangup (navigating away from the page that opened the connection)
+        this._websocket.close(1001);
+    }
 }