Interface of querying the information of progress which supports agent selection

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

Author: LintianShi

src/raft.rs

@@ -3036,4 +3036,36 @@ 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
+    }
 }

src/tracker/progress.rs

@@ -203,6 +203,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 {