Interface of querying the information of progress

Signed-off-by: LintianShi <lintian.shi@pingcap.com>

Add interface of query matched of progress

Signed-off-by: LintianShi <lintian.shi@pingcap.com>

Interface of querying last_idx of raft node

Signed-off-by: LintianShi <lintian.shi@pingcap.com>

Interface of query whether a progress lags behind

Signed-off-by: LintianShi <lintian.shi@pingcap.com>

Author: LintianShi

src/raft.rs

@@ -3045,4 +3045,52 @@ impl<T: Storage> Raft<T> {
             pr.ins.set_cap(cap);
         }
     }
+
+    /// Whether this RawNode is active recently.
+    pub fn is_recent_active(&self, id: u64) -> bool {
+        if let Some(pr) = self.prs().get(id) {
+            if pr.recent_active {
+                return true;
+            }
+        }
+        false
+    }
+
+    /// Get the next idx of peer.
+    #[inline]
+    pub fn get_next_idx(&self, id: u64) -> Option<u64> {
+        self.prs().get(id).map(|pr| pr.next_idx)
+    }
+
+    /// Get the matched of peer.
+    #[inline]
+    pub fn get_matched(&self, id: u64) -> Option<u64> {
+        self.prs().get(id).map(|pr| pr.matched)
+    }
+
+    /// Determine whether a progress is in Replicate state.
+    pub fn is_replicate_state(&self, id: u64) -> bool {
+        if let Some(pr) = self.prs().get(id) {
+            if pr.is_replicate_state() {
+                return true;
+            }
+        }
+        false
+    }
+
+    /// Determine whether a progress lags.
+    pub fn is_lag(&self, id: u64) -> bool {
+        if let Some(pr) = self.prs().get(id) {
+            if pr.lag {
+                return true;
+            }
+        }
+        false
+    }
+
+    /// Determine whether this is the leader.
+    #[inline]
+    pub fn is_leader(&self) -> bool {
+        self.id == self.leader_id
+    }
 }

src/raw_node.rs

@@ -786,12 +786,6 @@ impl<T: Storage> RawNode<T> {
     pub fn skip_bcast_commit(&mut self, skip: bool) {
         self.raft.skip_bcast_commit(skip)
     }
-
-    /// Set whether to batch append msg at runtime.
-    #[inline]
-    pub fn set_batch_append(&mut self, batch_append: bool) {
-        self.raft.set_batch_append(batch_append)
-    }
 }
 
 #[cfg(test)]

src/tracker/progress.rs

@@ -212,6 +212,12 @@ impl Progress {
         true
     }
 
+    /// Determine whether progress is in the Replicate state;
+    #[inline]
+    pub fn is_replicate_state(&self) -> bool {
+        self.state == ProgressState::Replicate
+    }
+
     /// Determine whether progress is paused.
     #[inline]
     pub fn is_paused(&self) -> bool {