Start sync workflow after transaction.commit() (#1967)

dust-tt · Oct 5, 2023 · 9dda1f7 · 9dda1f7
1 parent f6d46ef
commit 9dda1f7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 13 deletions.
diff --git a/connectors/src/connectors/github/index.ts b/connectors/src/connectors/github/index.ts
@@ -36,9 +36,8 @@ export async function createGithubConnector(
   if (!(await validateInstallationId(githubInstallationId))) {
     return new Err(new Error("Github installation id is invalid"));
   }
-
-  return await sequelize_conn.transaction(async (transaction) => {
-    try {
+  try {
+    const connector = await sequelize_conn.transaction(async (transaction) => {
       const connector = await Connector.create(
         {
           type: "github",
@@ -57,14 +56,16 @@ export async function createGithubConnector(
         },
         { transaction }
       );
-      await launchGithubFullSyncWorkflow(connector.id.toString());
-      return new Ok(connector.id.toString());
-    } catch (err) {
-      logger.error({ error: err }, "Error creating github connector");
 
-      return new Err(err as Error);
-    }
-  });
+      return connector;
+    });
+    await launchGithubFullSyncWorkflow(connector.id.toString());
+    return new Ok(connector.id.toString());
+  } catch (err) {
+    logger.error({ error: err }, "Error creating github connector");
+
+    return new Err(err as Error);
+  }
 }
 
 export async function updateGithubConnector(

diff --git a/connectors/src/connectors/notion/index.ts b/connectors/src/connectors/notion/index.ts
@@ -55,7 +55,7 @@ export async function createNotionConnector(
   }
 
   try {
-    return await sequelize_conn.transaction(async (transaction) => {
+    const connector = await sequelize_conn.transaction(async (transaction) => {
       const connector = await Connector.create(
         {
           type: "notion",
@@ -73,9 +73,11 @@ export async function createNotionConnector(
         },
         { transaction }
       );
-      await launchNotionSyncWorkflow(connector.id.toString());
-      return new Ok(connector.id.toString());
+
+      return connector;
     });
+    await launchNotionSyncWorkflow(connector.id.toString());
+    return new Ok(connector.id.toString());
   } catch (e) {
     logger.error({ error: e }, "Error creating notion connector.");
     return new Err(e as Error);