Query Membership in service

Node simply has to call the membership service with None members
pool2win · Oct 9, 2024 · 423511d · 423511d
1 parent 9a9b0db
commit 423511d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 22 deletions.
diff --git a/src/node.rs b/src/node.rs
@@ -234,27 +234,17 @@ impl Node {
 
     pub(crate) async fn send_membership(&self, sender: ReliableSenderHandle) {
         log::info!("Sending membership information");
-        match self.state.membership_handle.get_members().await {
-            Err(_) => {
-                log::debug!("Error reading membership");
-            }
-            Ok(members) => {
-                let members_name = Some(members.into_keys().collect());
-                let protocol_service =
-                    protocol::Protocol::new(self.get_node_id().clone(), self.state.clone());
-                let reliable_sender_service = ReliableSend::new(protocol_service, sender);
-                let timeout_layer = tower::timeout::TimeoutLayer::new(
-                    tokio::time::Duration::from_millis(self.delivery_timeout),
-                );
-                let res = timeout_layer
-                    .layer(reliable_sender_service)
-                    .oneshot(
-                        MembershipMessage::new(self.get_node_id().clone(), members_name).into(),
-                    )
-                    .await;
-                log::debug!("Membership sending result {:?}", res);
-            }
-        }
+        let protocol_service =
+            protocol::Protocol::new(self.get_node_id().clone(), self.state.clone());
+        let reliable_sender_service = ReliableSend::new(protocol_service, sender);
+        let timeout_layer = tower::timeout::TimeoutLayer::new(tokio::time::Duration::from_millis(
+            self.delivery_timeout,
+        ));
+        let res = timeout_layer
+            .layer(reliable_sender_service)
+            .oneshot(MembershipMessage::new(self.get_node_id().clone(), None).into())
+            .await;
+        log::debug!("Membership sending result {:?}", res);
     }
 
     /// Connect to all peers and start reader writer tasks

diff --git a/src/node/protocol/membership.rs b/src/node/protocol/membership.rs
@@ -68,7 +68,6 @@ impl Service<Message> for Membership {
     /// node when anyone connects to it.
     fn call(&mut self, msg: Message) -> Self::Future {
         let state = self.state.clone();
-        log::debug!("MSG {:?}", msg);
         async move {
             match msg {
                 Message::UnicastMessage(Unicast::Membership(MembershipMessage {