From 2b032229e46293af92db798a36793c6b8b97baee Mon Sep 17 00:00:00 2001 From: hozan23 Date: Wed, 29 Nov 2023 00:15:10 +0300 Subject: p2p/protocol: improve the Protocol API --- p2p/src/peer_pool.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'p2p/src/peer_pool.rs') diff --git a/p2p/src/peer_pool.rs b/p2p/src/peer_pool.rs index dd7e669..ee9ebf9 100644 --- a/p2p/src/peer_pool.rs +++ b/p2p/src/peer_pool.rs @@ -189,8 +189,7 @@ impl PeerPool { } }; - self.task_group - .spawn(peer.run(self.executor.clone()), on_disconnect); + self.task_group.spawn(peer.run(), on_disconnect); info!("Add new peer {pid}, direction: {conn_direction}, endpoint: {endpoint}"); @@ -230,8 +229,9 @@ impl PeerPool { /// Attach the core protocols. async fn setup_protocols(&self) -> Result<()> { - self.attach_protocol::(Box::new(PingProtocol::new)) - .await + let executor = self.executor.clone(); + let c = move |peer| PingProtocol::new(peer, executor.clone()); + self.attach_protocol::(Box::new(c)).await } /// Initiate a handshake with a connection. -- cgit v1.2.3