HDDS-8111. ReplicationManager: Add RatisMisReplicationHandler into rm…

….processUnderReplicatedContainer (#4414)
apache · Mar 17, 2023 · c14200c · c14200c
1 parent 1c30a35
commit c14200c
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/...cm/src/main/java/org/apache/hadoop/hdds/scm/container/replication/ReplicationManager.java b/...cm/src/main/java/org/apache/hadoop/hdds/scm/container/replication/ReplicationManager.java
@@ -169,6 +169,7 @@ public class ReplicationManager implements SCMService {
   private final ECMisReplicationHandler ecMisReplicationHandler;
   private final RatisUnderReplicationHandler ratisUnderReplicationHandler;
   private final RatisOverReplicationHandler ratisOverReplicationHandler;
+  private final RatisMisReplicationHandler ratisMisReplicationHandler;
   private final int maintenanceRedundancy;
   private final int ratisMaintenanceMinReplicas;
   private Thread underReplicatedProcessorThread;
@@ -239,6 +240,8 @@ public ReplicationManager(final ConfigurationSource conf,
         ratisContainerPlacement, conf, this);
     ratisOverReplicationHandler =
         new RatisOverReplicationHandler(ratisContainerPlacement, this);
+    ratisMisReplicationHandler = new RatisMisReplicationHandler(
+        ratisContainerPlacement, conf, this, rmConf.isPush());
     underReplicatedProcessor =
         new UnderReplicatedProcessor(this,
             rmConf.getUnderReplicatedInterval());
@@ -682,8 +685,18 @@ int processUnderReplicatedContainer(
             + result.getHealthState());
       }
     }
-    return ratisUnderReplicationHandler.processAndSendCommands(replicas,
-        pendingOps, result, ratisMaintenanceMinReplicas);
+    if (result.getHealthState()
+        == ContainerHealthResult.HealthState.UNDER_REPLICATED) {
+      return ratisUnderReplicationHandler.processAndSendCommands(replicas,
+          pendingOps, result, ratisMaintenanceMinReplicas);
+    } else if (result.getHealthState()
+        == ContainerHealthResult.HealthState.MIS_REPLICATED) {
+      return ratisMisReplicationHandler.processAndSendCommands(replicas,
+          pendingOps, result, ratisMaintenanceMinReplicas);
+    } else {
+      throw new IllegalArgumentException("Unexpected health state: "
+          + result.getHealthState());
+    }
   }
 
   int processOverReplicatedContainer(