Add retry mechanism to connect to PostgreSQL before starting maplibre#1539

Author: teyotan

Cargo.lock

@@ -113,6 +113,7 @@ tilejson.workspace = true
 tokio = { workspace = true, features = ["io-std"] }
 tokio-postgres-rustls = { workspace = true, optional = true }
 url.workspace = true
+tokio-retry = "0.3.0"
 
 [build-dependencies]
 static-files = { workspace = true, optional = true }

martin/Cargo.toml

@@ -15,7 +15,9 @@ use crate::pg::utils::on_slow;
 use crate::pg::PgResult;
 use crate::source::TileInfoSources;
 use crate::utils::{IdResolver, OptBoolObj, OptOneMany};
-use crate::MartinResult;
+use crate::{MartinError, MartinResult};
+use tokio_retry::strategy::{jitter, FixedInterval};
+use tokio_retry::Retry;
 
 pub trait PgInfo {
     fn format_id(&self) -> String;
@@ -114,7 +116,18 @@ impl PgConfig {
     }
 
     pub async fn resolve(&mut self, id_resolver: IdResolver) -> MartinResult<TileInfoSources> {
-        let pg = PgBuilder::new(self, id_resolver).await?;
+        // Retry strategy: Fixed 5 seconds interval backoff with jitter (random variation)
+        let retry_strategy = FixedInterval::from_millis(5000)
+            .map(jitter) // Add random jitter to avoid "thundering herd" problem
+            .take(3); // Retry up to 3 times
+
+        // Create PgBuilder using retry_strategy
+        let pg = Retry::spawn(retry_strategy, || async {
+            PgBuilder::new(self, id_resolver.clone()).await
+        })
+        .await
+        .map_err(|e| MartinError::PostgresError(e))?;
+
         let inst_tables = on_slow(
             pg.instantiate_tables(),
             // warn only if default bounds timeout has already passed