fix: Ensure state is updated after starting data layer provisioning (#…

…817)

- Need to write the state to move it out of "new" and in to "existing"
- Added some logs to `DataLayerService`

coordinator/src/synchroniser.rs

@@ -86,6 +86,14 @@ impl<'a> Synchroniser<'a> {
             .start_provisioning_task(config)
             .await?;
 
+        self.state_manager
+            .set_provisioning(config)
+            .await
+            .map_err(|err| {
+                tracing::warn!(?err, "Failed to set provisioning state");
+                err
+            })?;
+
         Ok(())
     }
 
@@ -594,6 +602,11 @@ mod test {
 
             let mut state_manager = IndexerStateManager::default();
             state_manager.expect_list().returning(|| Ok(vec![]));
+            state_manager
+                .expect_set_provisioning()
+                .with(eq(config.clone()))
+                .returning(|_| Ok(()))
+                .once();
 
             let mut data_layer_handler = DataLayerHandler::default();
             data_layer_handler

runner/src/server/services/data-layer/data-layer-service.ts

@@ -2,6 +2,7 @@ import { type ServerUnaryCall, type sendUnaryData, status, StatusBuilder } from
 
 import Provisioner from '../../../provisioner';
 import { ProvisioningConfig } from '../../../indexer-config/indexer-config';
+import parentLogger from '../../../logger';
 
 import { type CheckProvisioningTaskStatusRequest__Output } from '../../../generated/data_layer/CheckProvisioningTaskStatusRequest';
 import { type DataLayerHandlers } from '../../../generated/data_layer/DataLayer';
@@ -72,6 +73,12 @@ export function createDataLayerService (
 
       const provisioningConfig = new ProvisioningConfig(accountId, functionName, schema);
 
+      const logger = parentLogger.child({
+        service: 'DataLayerService',
+        accountId: provisioningConfig.accountId,
+        functionName: provisioningConfig.functionName,
+      });
+
       const task = tasks[generateTaskId(accountId, functionName)];
 
       if (task) {
@@ -91,7 +98,19 @@ export function createDataLayerService (
           return;
         }
 
-        tasks[generateTaskId(accountId, functionName)] = new ProvisioningTask(provisioner.provisionUserApi(provisioningConfig));
+        logger.info('Provisioning Data Layer');
+
+        tasks[generateTaskId(accountId, functionName)] = new ProvisioningTask(
+          provisioner
+            .provisionUserApi(provisioningConfig)
+            .then(() => {
+              logger.info('Successfully provisioned Data Layer');
+            })
+            .catch((err) => {
+              logger.error('Failed to provision Data Layer', err);
+              throw err;
+            })
+        );
 
         callback(null, { status: ProvisioningStatus.PENDING });
       }).catch((error) => {