store: Wait for the db to be up in set_manifest_raw_yaml

Author: lutter

store/postgres/src/deployment_store.rs

@@ -1844,12 +1844,23 @@ impl DeploymentStore {
     }
 
     pub(crate) async fn set_manifest_raw_yaml(
-        &self,
+        self: Arc<Self>,
         site: Arc<Site>,
         raw_yaml: String,
     ) -> Result<(), StoreError> {
-        self.with_conn(move |conn, _| {
-            deployment::set_manifest_raw_yaml(conn, &site, &raw_yaml).map_err(Into::into)
+        let logger = self.logger.cheap_clone();
+        retry::forever_async(&logger, "set_manifest_raw_yaml", move || {
+            let site = site.cheap_clone();
+            let raw_yaml = raw_yaml.clone();
+            let store = self.cheap_clone();
+            async move {
+                store
+                    .with_conn(move |conn, _| {
+                        deployment::set_manifest_raw_yaml(conn, &site, &raw_yaml)
+                            .map_err(Into::into)
+                    })
+                    .await
+            }
         })
         .await
     }

store/postgres/src/subgraph_store.rs

@@ -1615,7 +1615,10 @@ impl SubgraphStoreTrait for SubgraphStore {
         raw_yaml: String,
     ) -> Result<(), StoreError> {
         let (store, site) = self.store(hash)?;
-        store.set_manifest_raw_yaml(site, raw_yaml).await
+        store
+            .cheap_clone()
+            .set_manifest_raw_yaml(site, raw_yaml)
+            .await
     }
 
     fn instrument(&self, deployment: &DeploymentLocator) -> Result<bool, StoreError> {