Merge pull request #4115 from driftluo/fix-missing-await

fix: fix disconnect miss await
nervosnetwork · Aug 25, 2023 · 44d8a76 · 44d8a76
2 parents b82b336 + d7fe438
commit 44d8a76
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/network/src/protocols/mod.rs b/network/src/protocols/mod.rs
@@ -290,15 +290,15 @@ impl ServiceProtocol for CKBHandler {
             .network_state
             .ckb2023
             .load(std::sync::atomic::Ordering::SeqCst)
-            && version != "3"
+            && version != crate::protocols::support_protocols::LASTEST_VERSION
             && context.proto_id != SupportProtocols::RelayV2.protocol_id()
         {
             debug!(
                 "session {}, protocol {} with version {}, not 3, so disconnect it",
                 context.session.id, context.proto_id, version
             );
             let id = context.session.id;
-            let _ignore = context.disconnect(id);
+            let _ignore = context.disconnect(id).await;
             return;
         }
         self.network_state.with_peer_registry_mut(|reg| {

diff --git a/network/src/protocols/support_protocols.rs b/network/src/protocols/support_protocols.rs
@@ -6,7 +6,7 @@ use p2p::{
 };
 use tokio_util::codec::length_delimited;
 
-const LASTEST_VERSION: &str = "3";
+pub const LASTEST_VERSION: &str = "3";
 
 /// All supported protocols
 ///