feat: wait for loki connection

Author: David Gomes

server/src/tracing_setup.rs

@@ -29,21 +29,25 @@ pub async fn setup_loki() {
         .with(fmt::layer().with_writer(io::stdout));
 
     let http = Client::new();
-    let loki_base_url = env::var("LOKI_BASE_URL").unwrap_or("http://127.0.0.1:3100".into());
-
-    match http.get(loki_base_url.parse().unwrap()).await {
-        Ok(_) => {
-            let (layer, task) = build_loki_layer(&loki_base_url);
-
-            registry.with(layer).init();
-            tokio::spawn(task);
-
-            tracing::info!("Loki initialized");
-        }
-        Err(_) => {
+    let mut is_url_provided = true;
+    let loki_base_url = env::var("LOKI_BASE_URL").unwrap_or_else(|_| {
+        is_url_provided = false;
+        "http://127.0.0.1:3100".into()
+    });
+
+    if !is_url_provided {
+        if http.get(loki_base_url.parse().unwrap()).await.is_err() {
             registry.init();
 
             tracing::warn!("Couldn't connect to Loki. Continuing without it.");
+            return;
         }
-    };
+    }
+
+    let (layer, task) = build_loki_layer(&loki_base_url);
+
+    registry.with(layer).init();
+    tokio::spawn(task);
+
+    tracing::info!("Loki initialized");
 }