Make example a tiny bit more general (#78)

tikv · Jun 29, 2018 · b8d746a · b8d746a
1 parent fb292a7
commit b8d746a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/single_mem_node/main.rs b/examples/single_mem_node/main.rs
@@ -117,7 +117,7 @@ fn on_ready(r: &mut RawNode<MemStorage>, cbs: &mut HashMap<u8, ProposeCallback>)
     // The Raft is ready, we can do something now.
     let mut ready = r.ready();
 
-    let is_leader = r.raft.leader_id == 1;
+    let is_leader = r.raft.leader_id == r.raft.id;
     if is_leader {
         // If the peer is leader, the leader can send messages to other followers ASAP.
         let msgs = ready.messages.drain(..);