waiting for the pod to be ready

ppgia-unifor · Jul 28, 2024 · df678ce · df678ce
1 parent 4cb7cc7
commit df678ce
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/...ence-bench/operator/src/main/java/io/resiliencebench/execution/steps/EnvironmentStep.java b/...ence-bench/operator/src/main/java/io/resiliencebench/execution/steps/EnvironmentStep.java
@@ -1,18 +1,20 @@
 package io.resiliencebench.execution.steps;
 
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Service;
+
 import com.fasterxml.jackson.databind.JsonNode;
+
 import io.fabric8.kubernetes.api.model.apps.Deployment;
 import io.fabric8.kubernetes.client.KubernetesClient;
 import io.resiliencebench.resources.queue.ExecutionQueue;
 import io.resiliencebench.resources.scenario.Scenario;
 import io.resiliencebench.resources.service.ResilientService;
 import io.resiliencebench.support.CustomResourceRepository;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-import org.springframework.stereotype.Service;
-
-import java.util.Map;
-import java.util.concurrent.TimeUnit;
 
 @Service
 public class EnvironmentStep extends ExecutorStep<Deployment> {
@@ -22,7 +24,7 @@ public class EnvironmentStep extends ExecutorStep<Deployment> {
   private final CustomResourceRepository<ResilientService> resilientServiceRepository;
 
   public EnvironmentStep(KubernetesClient kubernetesClient,
-                         CustomResourceRepository<ResilientService> resilientServiceRepository) {
+                        CustomResourceRepository<ResilientService> resilientServiceRepository) {
     super(kubernetesClient);
     this.resilientServiceRepository = resilientServiceRepository;
   }
@@ -97,7 +99,9 @@ public void applyNewVariables(Deployment deployment, String containerName, Map<S
             .deployments()
             .inNamespace(deployment.getMetadata().getNamespace())
             .withName(deployment.getMetadata().getName())
-            .waitUntilCondition(pod -> pod.getStatus().getReadyReplicas().equals(pod.getStatus().getReplicas()), 1, TimeUnit.MINUTES);
+            .waitUntilCondition(pod -> pod.getStatus().getConditions().stream().anyMatch(
+              condition -> "Ready".equals(condition.getType()) && "True".equals(condition.getStatus()
+            )), 1, TimeUnit.MINUTES);
     logger.info("Pods restarted successfully");
   }
 }