Merge pull request #186 from dfinity/shiling/router

feat: Block startup until route provider is healthy

Author: shilingwang

src/routing/ic/route_provider.rs

@@ -1,4 +1,4 @@
-use std::sync::Arc;
+use std::{sync::Arc, time::Duration};
 
 use anyhow::anyhow;
 use derive_new::new;
@@ -13,7 +13,8 @@ use ic_bn_lib::ic_agent::agent::{
     },
 };
 use ic_bn_lib_common::{principal, traits::Healthy};
-use tracing::info;
+use tokio::time::{sleep, timeout};
+use tracing::{info, warn};
 use url::Url;
 
 use crate::{
@@ -96,5 +97,17 @@ pub async fn setup_route_provider(
         Arc::new(RoundRobinRouteProvider::new(urls_str)?)
     };
 
+    let wrapper = RouteProviderWrapper::new(route_provider.clone());
+    if timeout(Duration::from_secs(120), async {
+        while !wrapper.healthy() {
+            sleep(Duration::from_secs(1)).await;
+        }
+    })
+    .await
+    .is_err()
+    {
+        warn!("Route provider did not become healthy within 2 minutes, continuing anyway");
+    }
+
     Ok(route_provider)
 }