Deploy-queue heartbeat: Added retry when fetching deployment

deploy-queue/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "deploy-queue"
-version = "0.8.2"
+version = "0.8.3"
 edition = "2024"
 
 [dependencies]

deploy-queue/src/constants.rs

@@ -7,3 +7,5 @@ pub const BUSY_RETRY: Duration = Duration::from_secs(5);
 pub const HEARTBEAT_INTERVAL: Duration = Duration::from_secs(30);
 pub const HEARTBEAT_TIMEOUT: Duration = Duration::from_secs(15 * 60); // 15 minutes
 pub const HEARTBEAT_UPDATE_TIMEOUT: Duration = Duration::from_secs(20);
+pub const DEPLOYMENT_ID_LOOKUP_RETRY: Duration = Duration::from_secs(10);
+pub const DEPLOYMENT_ID_LOOKUP_TIMEOUT: Duration = Duration::from_secs(5 * 60);

deploy-queue/src/lib.rs

@@ -1,4 +1,4 @@
-use anyhow::{Context, Result};
+use anyhow::{Context, Result, bail};
 use clap::Parser;
 
 pub mod cli;
@@ -109,9 +109,24 @@ pub async fn run_deploy_queue(mode: cli::Mode, skip_migrations: bool) -> Result<
                     })?;
             }
             cli::HeartbeatTarget::Url { url } => {
-                let deployment_id = handler::fetch::deployment_id_by_url(&db_client, &url)
-                    .await?
-                    .with_context(|| format!("No deployment found with URL: {}", url))?;
+                let start = std::time::Instant::now();
+                let deployment_id = loop {
+                    if let Some(deployment_id) =
+                        handler::fetch::deployment_id_by_url(&db_client, &url).await?
+                    {
+                        break deployment_id;
+                    }
+
+                    if start.elapsed() >= constants::DEPLOYMENT_ID_LOOKUP_TIMEOUT {
+                        bail!(
+                            "No deployment found with URL after {}s: {}",
+                            constants::DEPLOYMENT_ID_LOOKUP_TIMEOUT.as_secs(),
+                            url
+                        );
+                    }
+
+                    tokio::time::sleep(constants::DEPLOYMENT_ID_LOOKUP_RETRY).await;
+                };
 
                 handler::run_heartbeat_loop(&db_client, deployment_id)
                     .await