reject prevote explicitly when in lease

Signed-off-by: Jay Lee <[email protected]>
tikv · Mar 7, 2021 · 158f7f1 · 158f7f1
1 parent 6a895ab
commit 158f7f1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/raft.rs b/src/raft.rs
@@ -1289,6 +1289,18 @@ impl<T: Storage> Raft<T> {
                         "msg type" => ?m.get_msg_type(),
                     );
 
+                    // When judge_split_prevote, reject explicitly to make candidate exit PreCandiate early
+                    // so it will vote for other peer later.
+                    if self.judge_split_prevote
+                        && m.get_msg_type() == MessageType::MsgRequestPreVote
+                    {
+                        let mut to_send =
+                            new_message(m.from, MessageType::MsgRequestPreVoteResponse, None);
+                        to_send.term = m.term;
+                        to_send.reject = true;
+                        self.r.send(to_send, &mut self.msgs);
+                    }
+
                     return Ok(());
                 }
             }