fix: cleanup open streams on conn close (#791)

libp2p · Oct 23, 2020 · 06f26e5 · 06f26e5
1 parent 8879634
commit 06f26e5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/upgrader.js b/src/upgrader.js
@@ -306,7 +306,13 @@ class Upgrader {
       },
       newStream: newStream || errConnectionNotMultiplexed,
       getStreams: () => muxer ? muxer.streams : errConnectionNotMultiplexed,
-      close: (err) => maConn.close(err)
+      close: async (err) => {
+        await maConn.close(err)
+        // Ensure remaining streams are aborted
+        if (muxer) {
+          muxer.streams.map(stream => stream.abort(err))
+        }
+      }
     })
 
     this.onConnection(connection)