remove leaked event listeners

1 files changed, 14 insertions, 2 deletions

diff --git a/lib/peer.js b/lib/peer.js
index 2f291d8..5471773 100644
--- a/lib/peer.js
+++ b/lib/peer.js
@@ -20,8 +20,20 @@ exports.createWebRTCPeer = (conn, swarm) => {
   if (peer.conn.connected) {
     peer.onConnect()
   } else {
-    peer.conn.once('connect', () => { peer.onConnect() })
-    peer.conn.once('error', err => { peer.destroy(err) })
+    const cleanup = () => {
+      peer.conn.removeListener('connect', onConnect)
+      peer.conn.removeListener('error', onError)
+    }
+    const onConnect = () => {
+      cleanup()
+      peer.onConnect()
+    }
+    const onError = err => {
+      cleanup()
+      peer.destroy(err)
+    }
+    peer.conn.once('connect', onConnect)
+    peer.conn.once('error', onError)
     peer.startConnectTimeout()
   }