Merge pull request #201 from code0-tech/#189-retry-sending-definitions

retry sending definitions

Author: raphael-goetz

Cargo.lock

@@ -7,7 +7,7 @@ version = "0.0.0"
 edition = "2024"
 
 [workspace.dependencies]
-code0-flow = { version = "0.0.30" }
+code0-flow = { version = "0.0.31" }
 tucana = { version = "0.0.68", features = ["aquila"] }
 serde_json = { version = "1.0.138" }
 log = "0.4.27"

Cargo.toml

@@ -5,8 +5,8 @@ use crate::{
     traits::{LoadConfig, Server as AdapterServer},
 };
 use code0_flow::flow_service::FlowUpdateService;
-use std::sync::Arc;
-use tokio::signal;
+use std::{sync::Arc, time::Duration};
+use tokio::{signal, time::sleep};
 use tonic::transport::Server;
 use tonic_health::pb::health_server::HealthServer;
 use tucana::shared::{AdapterConfiguration, RuntimeFeature};
@@ -91,7 +91,22 @@ impl<C: LoadConfig> ServerRunner<C> {
                 config.aquila_token.clone(),
             )
             .await;
-            definition_service.send().await;
+
+            let mut success = false;
+            let mut count = 1;
+            while !success {
+                success = definition_service.send_with_status().await;
+                if success {
+                    break;
+                }
+
+                log::warn!(
+                    "Updating definitions failed, trying again in 2 secs (retry number {})",
+                    count
+                );
+                count += 1;
+                sleep(Duration::from_secs(3)).await;
+            }
         }
 
         let health_task = if config.with_health_service {