Fix nits

lablup · Feb 1, 2024 · 32ac6f4 · 32ac6f4
1 parent 003c296
commit 32ac6f4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/examples/memstore/dynamic-members/src/main.rs b/examples/memstore/dynamic-members/src/main.rs
@@ -64,7 +64,6 @@ async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
                 .await
                 .unwrap();
             let node_id = ticket.reserved_id;
-            println!("ticket.peers {:?}", ticket.peers);
             cfg.initial_peers = Some(ticket.peers.clone().into());
 
             let raft = Raft::bootstrap(

diff --git a/raftify/src/raft_node/mod.rs b/raftify/src/raft_node/mod.rs
@@ -457,7 +457,7 @@ impl<
             .clone()
             .inner
             .into_iter()
-            .filter(|(_, peer)| peer.role == InitialRole::Voter)
+            .filter(|(_, peer)| peer.role == InitialRole::Voter || peer.role == InitialRole::Leader)
             .map(|(key, _)| key)
             .collect::<Vec<_>>();
 

diff --git a/raftify/src/raft_node/role.rs b/raftify/src/raft_node/role.rs
@@ -35,10 +35,11 @@ impl FromStr for InitialRole {
     type Err = ();
 
     fn from_str(s: &str) -> Result<Self, Self::Err> {
-        match s {
-            "Leader" => Ok(InitialRole::Leader),
-            "Voter" => Ok(InitialRole::Voter),
-            "Learner" => Ok(InitialRole::Learner),
+        let s = s.to_lowercase();
+        match s.as_str() {
+            "leader" => Ok(InitialRole::Leader),
+            "voter" => Ok(InitialRole::Voter),
+            "learner" => Ok(InitialRole::Learner),
             _ => Err(()),
         }
     }