From 72c920c6e8e4bd24b254ad9e56514e1fa9a66283 Mon Sep 17 00:00:00 2001
From: Potato <tanxinyu@apache.org>
Date: Mon, 13 Jan 2025 09:52:28 +0800
Subject: [PATCH] RATIS-2237 Add getFollowerMatchIndices interface for
 DivisionInfo

Signed-off-by: OneSizeFitQuorum <tanxinyu@apache.org>
---
 .../main/java/org/apache/ratis/server/DivisionInfo.java   | 6 ++++++
 .../org/apache/ratis/server/impl/LeaderStateImpl.java     | 4 ++++
 .../java/org/apache/ratis/server/impl/RaftServerImpl.java | 8 ++++++++
 3 files changed, 18 insertions(+)

diff --git a/ratis-server-api/src/main/java/org/apache/ratis/server/DivisionInfo.java b/ratis-server-api/src/main/java/org/apache/ratis/server/DivisionInfo.java
index c07317bd69..78073a21c5 100644
--- a/ratis-server-api/src/main/java/org/apache/ratis/server/DivisionInfo.java
+++ b/ratis-server-api/src/main/java/org/apache/ratis/server/DivisionInfo.java
@@ -80,4 +80,10 @@ default boolean isAlive() {
    *         otherwise, return null.
    */
   long[] getFollowerNextIndices();
+
+  /**
+   * @return an array of match indices of the followers if this server division is the leader;
+   *         otherwise, return null.
+   */
+  long[] getFollowerMatchIndices();
 }
diff --git a/ratis-server/src/main/java/org/apache/ratis/server/impl/LeaderStateImpl.java b/ratis-server/src/main/java/org/apache/ratis/server/impl/LeaderStateImpl.java
index f5682e35f0..bb705fb9fb 100644
--- a/ratis-server/src/main/java/org/apache/ratis/server/impl/LeaderStateImpl.java
+++ b/ratis-server/src/main/java/org/apache/ratis/server/impl/LeaderStateImpl.java
@@ -1214,6 +1214,10 @@ long[] getFollowerNextIndices() {
     return getLogAppenders().mapToLong(s -> s.getFollower().getNextIndex()).toArray();
   }
 
+  long[] getFollowerMatchIndices() {
+    return getLogAppenders().mapToLong(s -> s.getFollower().getMatchIndex()).toArray();
+  }
+
   static Map<RaftPeerId, RaftPeer> newMap(Collection<RaftPeer> peers, String str) {
     Objects.requireNonNull(peers, () -> str + " == null");
     final Map<RaftPeerId, RaftPeer> map = new HashMap<>();
diff --git a/ratis-server/src/main/java/org/apache/ratis/server/impl/RaftServerImpl.java b/ratis-server/src/main/java/org/apache/ratis/server/impl/RaftServerImpl.java
index 4c752d43bf..c67329cae1 100644
--- a/ratis-server/src/main/java/org/apache/ratis/server/impl/RaftServerImpl.java
+++ b/ratis-server/src/main/java/org/apache/ratis/server/impl/RaftServerImpl.java
@@ -206,6 +206,14 @@ public long[] getFollowerNextIndices() {
           .map(LeaderStateImpl::getFollowerNextIndices)
           .orElse(null);
     }
+
+    @Override
+    public long[] getFollowerMatchIndices() {
+      return role.getLeaderState()
+          .filter(leader -> isLeader())
+          .map(LeaderStateImpl::getFollowerMatchIndices)
+          .orElse(null);
+    }
   }
 
   private final RaftServerProxy proxy;