[Assembly] wait_servers_ready() instead of sleep 3s when start engine…

… on vineyard (#2136)
alibaba · Oct 13, 2022 · d4d942c · d4d942c
1 parent 2aa3cfc
commit d4d942c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/interactive_engine/executor/assembly/v6d/src/bin/gaia_executor.rs b/interactive_engine/executor/assembly/v6d/src/bin/gaia_executor.rs
@@ -21,7 +21,7 @@ use gaia_runtime::error::{StartServerError, StartServerResult};
 use global_query::{FFIGraphStore, GraphPartitionManager};
 use log::info;
 use pegasus::api::{Fold, Sink};
-use pegasus::{Configuration, JobConf, ServerConf};
+use pegasus::{wait_servers_ready, Configuration, JobConf, ServerConf};
 use pegasus_network::config::NetworkConfig;
 use pegasus_network::config::ServerAddr;
 use pegasus_server::rpc::{start_rpc_server, RPCServerConfig, ServiceStartListener};
@@ -92,7 +92,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     info!("process_partition_list: {:?}", process_partition_list);
 
     pegasus::startup(server_config)?;
-    std::thread::sleep(std::time::Duration::from_secs(3));
+    wait_servers_ready(&ServerConf::All);
     let partition_server_index_map = get_global_partition_server_mapping(process_partition_list)?;
 
     let query_vineyard = QueryVineyard::new(