disconnect the peer only once

vacp2p · Mar 21, 2024 · 484c25d · 484c25d
1 parent 8c9a6ae
commit 484c25d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/libp2p/protocols/pubsub/pubsubpeer.nim b/libp2p/protocols/pubsub/pubsubpeer.nim
@@ -87,6 +87,7 @@ type
 
     rpcmessagequeue: RpcMessageQueue
     maxNumElementInNonPriorityQueue*: int # The max number of elements allowed in the non-priority queue.
+    disconnected: bool
 
   RPCHandler* = proc(peer: PubSubPeer, data: seq[byte]): Future[void]
     {.gcsafe, raises: [].}
@@ -345,8 +346,12 @@ proc sendEncoded*(p: PubSubPeer, msg: seq[byte], isHighPriority: bool): Future[v
     f
   else:
     if len(p.rpcmessagequeue.nonPriorityQueue) == p.maxNumElementInNonPriorityQueue:
-      libp2p_pubsub_disconnects_over_non_priority_queue_limit.inc()
-      p.closeSendConn(PubSubPeerEventKind.DisconnectionRequested)
+      if not p.disconnected:
+        p.disconnected = true
+        libp2p_pubsub_disconnects_over_non_priority_queue_limit.inc()
+        p.closeSendConn(PubSubPeerEventKind.DisconnectionRequested)
+      else:
+        Future[void].completed()
     else:
       let f = p.rpcmessagequeue.nonPriorityQueue.addLast(msg)
       when defined(pubsubpeer_queue_metrics):