implementing disconnectNode

waku-org · Oct 11, 2024 · 32c1439 · 32c1439
1 parent 553d449
commit 32c1439
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/library/waku_thread/inter_thread_communication/requests/peer_manager_request.nim b/library/waku_thread/inter_thread_communication/requests/peer_manager_request.nim
@@ -95,6 +95,10 @@ proc process*(
       .join(",")
     return ok(connectedPeers)
   of DISCONNECT_PEER_BY_ID:
+    let peerId = PeerId.init($self[].peerId).valueOr:
+      error "DISCONNECT_PEER_BY_ID failed", error = $error
+      return err($error)
+    await waku.node.peerManager.disconnectNode(peerId)
     return ok("")
 
   return ok("")
diff --git a/waku/node/peer_manager/peer_manager.nim b/waku/node/peer_manager/peer_manager.nim
@@ -381,9 +381,12 @@ proc connectToNodes*(
   # later.
   await sleepAsync(chronos.seconds(5))
 
+proc disconnectNode*(pm: PeerManager, peerId: PeerId) {.async.} =
+  await pm.switch.disconnect(peerId)
+
 proc disconnectNode*(pm: PeerManager, peer: RemotePeerInfo) {.async.} =
   let peerId = peer.peerId
-  await pm.switch.disconnect(peerId)
+  await pm.disconnectNode(peerId)
 
 # Dialing should be used for just protocols that require a stream to write and read
 # This shall not be used to dial Relay protocols, since that would create