Await condition

Author: EliXorZz

src/controller/definitions/epsilon_instance.rs

@@ -51,7 +51,7 @@ impl EpsilonInstance {
             name: self.metadata.name.as_ref().unwrap().clone(),
             template: self.spec.template.clone(),
             state: self.get_state().clone(),
-            slots: self.get_slots(),
+            slots: self.status.as_ref().unwrap().slots,
             online_count: self.get_online_count().await.unwrap_or(0),
         }
     }
@@ -63,13 +63,6 @@ impl EpsilonInstance {
         }
     }
 
-    pub fn get_slots(&self) -> i32 {
-        match &self.status {
-            None => 0,
-            Some(status) => status.slots,
-        }
-    }
-
     pub async fn get_info(&self) -> EResult<StatusResponse> {
         let status = self.status.as_ref().unwrap();
         let address = status.ip.as_ref().unwrap();

src/epsilon/server/instances/instance_provider.rs

@@ -10,6 +10,7 @@ use anyhow::format_err;
 use futures::StreamExt;
 use k8s_openapi::apimachinery::pkg::apis::meta::v1::ObjectMeta;
 use kube::api::{DeleteParams, ListParams, PostParams};
+use kube::runtime::wait::await_condition;
 use kube::Api;
 use serde_json::json;
 use std::env;
@@ -66,6 +67,16 @@ impl InstanceProvider {
     ) -> EResult<Vec<Arc<EpsilonInstance>>> {
         let mut instances = self.epsilon_controller.get_epsilon_instance_store().state();
 
+        for instance in &instances {
+            let condition = await_condition(
+                self.epsilon_controller.get_epsilon_instance_api().clone(),
+                instance.metadata.name.as_ref().unwrap(),
+                move |instance: Option<&EpsilonInstance>| instance.is_some(),
+            );
+
+            let _ = tokio::time::timeout(std::time::Duration::from_secs(3), condition).await?;
+        }
+
         instances = instances
             .into_iter()
             .filter(|instance| {