do not send messages

Signed-off-by: Little-Wallace <[email protected]>

Author: Little-Wallace

src/raft.rs

@@ -128,6 +128,9 @@ pub struct Raft<T: Storage> {
     /// The list of messages.
     pub msgs: Vec<Message>,
 
+    /// The list of heartbeat messages.
+    pub heartbeats: Vec<Message>,
+
     /// The leader id
     pub leader_id: u64,
 
@@ -246,6 +249,7 @@ impl<T: Storage> Raft<T> {
             election_timeout: c.election_tick,
             votes: Default::default(),
             msgs: Default::default(),
+            heartbeats: Default::default(),
             leader_id: Default::default(),
             lead_transferee: None,
             term: Default::default(),
@@ -710,10 +714,12 @@ impl<T: Storage> Raft<T> {
         m.set_msg_type(MessageType::MsgHeartbeat);
         let commit = cmp::min(pr.matched, self.raft_log.committed);
         m.commit = commit;
+        m.from = self.id;
+        m.term = self.term;
         if let Some(context) = ctx {
             m.context = context;
         }
-        self.send(m);
+        self.heartbeats.push(m);
     }
 
     /// Sends RPC, with entries to all peers that are not up-to-date
@@ -2005,7 +2011,16 @@ impl<T: Storage> Raft<T> {
         to_send.to = m.from;
         to_send.context = m.take_context();
         to_send.commit = self.raft_log.committed;
-        self.send(to_send);
+        if self.raft_log.committed > self.raft_log.store.last_index().unwrap()
+        {
+            // If there is some entries that has committed in memory but not persisted, the message
+            // shall not be sent until all entries before committed_index have been persisted.
+            self.send(to_send);
+        } else {
+            to_send.term = self.term;
+            to_send.from = self.id;
+            self.heartbeats.push(to_send);
+        }
     }
 
     fn handle_snapshot(&mut self, mut m: Message) {