From 06f26e586fb2652d2ea1258d3862f3ddd27ab2e0 Mon Sep 17 00:00:00 2001
From: Jacob Heun <jacobheun@gmail.com>
Date: Fri, 23 Oct 2020 15:34:59 +0200
Subject: [PATCH] fix: cleanup open streams on conn close (#791)

---
 src/upgrader.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/upgrader.js b/src/upgrader.js
index fd8fe119..92997eb1 100644
--- 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)