From 20d54f4d3cad5732859e4594f3bc54cc0b7278dd Mon Sep 17 00:00:00 2001
From: tuyennhv <vutuyen2636@gmail.com>
Date: Thu, 19 Jan 2023 21:40:10 +0700
Subject: [PATCH] fix: ignore new closed connection (#399)

* fix: ignore new closed connection

* fix: update src/metrics.ts

Co-authored-by: Cayman <caymannava@gmail.com>
---
 src/index.ts   | 5 ++++-
 src/metrics.ts | 6 ++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/index.ts b/src/index.ts
index 3a9bb99a..2b8c044c 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -698,7 +698,10 @@ export class GossipSub extends EventEmitter<GossipsubEvents> implements PubSub<G
    * Registrar notifies an established connection with pubsub protocol
    */
   private onPeerConnected(peerId: PeerId, connection: Connection): void {
-    if (!this.isStarted()) {
+    this.metrics?.newConnectionCount.inc({ status: connection.stat.status })
+    // libp2p may emit a closed connection and never issue peer:disconnect event
+    // see https://github.com/ChainSafe/js-libp2p-gossipsub/issues/398
+    if (!this.isStarted() || connection.stat.status !== 'OPEN') {
       return
     }
 
diff --git a/src/metrics.ts b/src/metrics.ts
index 2f01e9e8..caa29ba6 100644
--- a/src/metrics.ts
+++ b/src/metrics.ts
@@ -512,6 +512,12 @@ export function getMetrics(
       help: 'Total count of key collisions on fastmsgid cache put'
     }),
 
+    newConnectionCount: register.gauge<{ status: string }>({
+      name: 'gossipsub_new_connection_total',
+      help: 'Total new connection by status',
+      labelNames: ['status']
+    }),
+
     topicStrToLabel: topicStrToLabel,
 
     toTopic(topicStr: TopicStr): TopicLabel {