fix: process hangs when using websocket provider

Fixes #46.
TripleSpeeder · Dec 10, 2019 · 85a26ef · 85a26ef
1 parent 6428ec7
commit 85a26ef
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -126,11 +126,15 @@ module.exports = function Updater() {
     }
 
     async function stop() {
-        // Perform a clean shutdown. Especially needed for HDWalletProvider to prevent dangling process
+        // Perform a clean shutdown. Especially needed for HDWalletProvider engine to prevent dangling process
         if (web3 && web3.currentProvider && web3.currentProvider.engine && web3.currentProvider.engine.stop) {
-            verbose && console.log("Stopping provider")
+            verbose && console.log("Stopping provider engine")
             await web3.currentProvider.engine.stop()
         }
+        if(web3 && web3.currentProvider && web3.currentProvider.connection && web3.currentProvider.connection.close){
+            verbose && console.log("Closing provider connection")
+            await web3.currentProvider.connection.close()
+        }
     }
 
     function isMethodConstant(contract, method) {