@teamkeel/functions-runtime 0.452.0 → 0.452.1

package/dist/index.js

@@ -4060,6 +4060,23 @@ var defaultOpts = {
   retries: 4,
   timeout: 6e4
 };
+async function insertNewStep(db, runId, name, stage) {
+  await db.transaction().execute(async (trx) => {
+    await trx.selectFrom("keel.flow_run").select("id").where("id", "=", runId).forUpdate().executeTakeFirst();
+    const existing = await trx.selectFrom("keel.flow_step").select("id").where("run_id", "=", runId).where("name", "=", name).where("status", "=", "NEW" /* NEW */).executeTakeFirst();
+    if (existing) {
+      return;
+    }
+    await trx.insertInto("keel.flow_step").values({
+      run_id: runId,
+      name,
+      stage,
+      status: "NEW" /* NEW */,
+      type: "FUNCTION" /* FUNCTION */
+    }).execute();
+  });
+}
+__name(insertNewStep, "insertNewStep");
 function createFlowContext(runId, data, action, callback, element, spanId, ctx) {
   const usedNames = /* @__PURE__ */ new Set();
   return {
@@ -4150,13 +4167,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
                 }
                 throw new ExhuastedRetriesDisrupt();
               }
-              await db.insertInto("keel.flow_step").values({
-                run_id: runId,
-                name,
-                stage: options.stage,
-                status: "NEW" /* NEW */,
-                type: "FUNCTION" /* FUNCTION */
-              }).returningAll().executeTakeFirst();
+              await insertNewStep(db, runId, name, options.stage);
               span.setAttribute("step.status", "NEW" /* NEW */);
               throw new StepCreatedDisrupt(
                 options.retryPolicy ? new Date(
@@ -4173,13 +4184,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
             span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
             return result;
           }
-          await db.insertInto("keel.flow_step").values({
-            run_id: runId,
-            name,
-            stage: options.stage,
-            status: "NEW" /* NEW */,
-            type: "FUNCTION" /* FUNCTION */
-          }).returningAll().executeTakeFirst();
+          await insertNewStep(db, runId, name, options.stage);
           span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
           span.setAttribute("step.status", "NEW" /* NEW */);
           throw new StepCreatedDisrupt();