Interface of querying the information of progress which supports agen…

…t selection

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

src/raft.rs

@@ -3036,4 +3036,30 @@ impl<T: Storage> Raft<T> {
             pr.ins.set_cap(cap);
         }
     }
+
+    /// Whether this RawNode is active recently.
+    #[inline]
+    pub fn is_recent_active(&self, id: u64) -> bool {
+        self.prs().get(id).map_or(false, |pr| pr.recent_active)
+    }
+
+    /// 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.
+    #[inline]
+    pub fn is_replicate_state(&self, id: u64) -> bool {
+        self.prs()
+            .get(id)
+            .map_or(false, |pr| pr.is_replicate_state())
+    }
 }

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 {