fix: handle closing outbound stream (#314)

ChainSafe · Aug 8, 2022 · 74c08b1 · 74c08b1
1 parent 5e6ce1f
commit 74c08b1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -2736,6 +2736,8 @@ export class GossipSub extends EventEmitter<GossipsubEvents> implements Initiali
     metrics.cacheSize.set({ cache: 'fanout' }, this.fanout.size)
     // Bounded by peer
     metrics.cacheSize.set({ cache: 'peers' }, this.peers.size)
+    metrics.cacheSize.set({ cache: 'streamsOutbound' }, this.streamsOutbound.size)
+    metrics.cacheSize.set({ cache: 'streamsInbound' }, this.streamsInbound.size)
     metrics.cacheSize.set({ cache: 'acceptFromWhitelist' }, this.acceptFromWhitelist.size)
     metrics.cacheSize.set({ cache: 'gossip' }, this.gossip.size)
     metrics.cacheSize.set({ cache: 'control' }, this.control.size)

diff --git a/src/stream.ts b/src/stream.ts
@@ -33,6 +33,8 @@ export class OutboundStream {
 
   close(): void {
     this.closeController.abort()
+    // similar to pushable.end() but clear the internal buffer
+    this.pushable.return()
     this.rawStream.close()
   }
 }