fix(raft): update role metrics when closing or transitioning to inactive

(cherry picked from commit 35ea7ba)
camunda · Nov 29, 2023 · b6c6c11 · b6c6c11
1 parent d25c1ae
commit b6c6c11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/atomix/cluster/src/main/java/io/atomix/raft/impl/RaftContext.java b/atomix/cluster/src/main/java/io/atomix/raft/impl/RaftContext.java
@@ -853,6 +853,7 @@ public void checkThread() {
   private RaftRole createRole(final Role role) {
     switch (role) {
       case INACTIVE:
+        raftRoleMetrics.becomingInactive();
         return new InactiveRole(this);
       case PASSIVE:
         return new PassiveRole(this);
@@ -915,6 +916,7 @@ public void transition(final Type type) {
 
   @Override
   public void close() {
+    raftRoleMetrics.becomingInactive();
     started = false;
     // Unregister protocol listeners.
     unregisterHandlers(protocol);

diff --git a/atomix/cluster/src/main/java/io/atomix/raft/metrics/RaftRoleMetrics.java b/atomix/cluster/src/main/java/io/atomix/raft/metrics/RaftRoleMetrics.java
@@ -67,6 +67,10 @@ public RaftRoleMetrics(final String partitionName) {
     electionLatency = ELECTION_LATENCY.labels(partitionGroupName, partition);
   }
 
+  public void becomingInactive() {
+    role.set(0);
+  }
+
   public void becomingFollower() {
     role.set(1);
   }