p2p/protocol: improve the Protocol API

1 files changed, 4 insertions, 4 deletions

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::<PingProtocol>(Box::new(PingProtocol::new))
-            .await
+        let executor = self.executor.clone();
+        let c = move |peer| PingProtocol::new(peer, executor.clone());
+        self.attach_protocol::<PingProtocol>(Box::new(c)).await
     }
 
     /// Initiate a handshake with a connection.