fix: handling disconnects

ipfs-shipyard · Dec 7, 2017 · 47a8e33 · 47a8e33
1 parent 3f649a8
commit 47a8e33
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/connection.js b/src/connection.js
@@ -58,7 +58,10 @@ module.exports = class Connection extends EventEmitter {
         pull(
           pushable,
           conn,
-          pull.onEnd(() => delete this._connection)
+          pull.onEnd(() => {
+            delete this._connection
+            this.emit('disconnect')
+          })
         )
         this.emit('connect', pushable)
       })

diff --git a/src/index.js b/src/index.js
@@ -75,6 +75,12 @@ class PubSubRoom extends EventEmitter {
       conn = new Connection(peer, this._ipfs, this)
       conn.on('error', (err) => this.emit('error', err))
       this._connections[peer] = conn
+
+      conn.once('disconnect', () => {
+        delete this._connections[peer]
+        this._peers = this._peers.filter((p) => p !== peer)
+        this.emit('peer left', peer)
+      })
     }
 
     // We should use the same sequence number generation as js-libp2p-floosub does: