@teamkeel/functions-runtime 0.454.2 → 0.455.0

package/dist/index.cjs

@@ -588,9 +588,9 @@ var InstrumentedClient = class extends import_pg.Client {
   }
   async query(...args) {
     const _super = super.query.bind(this);
-    const sql5 = args[0];
+    const sql4 = args[0];
     let sqlAttribute = false;
-    let spanName = txStatements[sql5.toLowerCase()];
+    let spanName = txStatements[sql4.toLowerCase()];
     if (!spanName) {
       spanName = "Database Query";
       sqlAttribute = true;
@@ -658,9 +658,9 @@ function getDialect(connString) {
       pool.on("connect", (client) => {
         const originalQuery = client.query;
         client.query = function(...args) {
-          const sql5 = args[0];
+          const sql4 = args[0];
           let sqlAttribute = false;
-          let spanName = txStatements[sql5.toLowerCase()];
+          let spanName = txStatements[sql4.toLowerCase()];
           if (!spanName) {
             spanName = "Database Query";
             sqlAttribute = true;
@@ -1151,23 +1151,23 @@ var TimePeriod = class _TimePeriod {
     return new _TimePeriod(period, value, offset, complete2);
   }
   periodStartSQL() {
-    let sql5 = "NOW()";
+    let sql4 = "NOW()";
     if (this.offset !== 0) {
-      sql5 = `${sql5} + INTERVAL '${this.offset} ${this.period}'`;
+      sql4 = `${sql4} + INTERVAL '${this.offset} ${this.period}'`;
     }
     if (this.complete) {
-      sql5 = `DATE_TRUNC('${this.period}', ${sql5})`;
+      sql4 = `DATE_TRUNC('${this.period}', ${sql4})`;
     } else {
-      sql5 = `(${sql5})`;
+      sql4 = `(${sql4})`;
     }
-    return sql5;
+    return sql4;
   }
   periodEndSQL() {
-    let sql5 = this.periodStartSQL();
+    let sql4 = this.periodStartSQL();
     if (this.value != 0) {
-      sql5 = `(${sql5} + INTERVAL '${this.value} ${this.period}')`;
+      sql4 = `(${sql4} + INTERVAL '${this.value} ${this.period}')`;
     }
-    return sql5;
+    return sql4;
   }
 };
 
@@ -3320,24 +3320,11 @@ var import_json_rpc_26 = require("json-rpc-2.0");
 var opentelemetry6 = __toESM(require("@opentelemetry/api"), 1);
 
 // src/tryExecuteFlow.js
-var import_kysely6 = require("kysely");
 function tryExecuteFlow(db, request, cb) {
-  return withDatabase(db, false, async ({ sDb }) => {
-    const runId = request?.meta?.runId;
-    if (runId && sDb) {
-      await import_kysely6.sql`SELECT pg_advisory_lock(hashtextextended(${runId}, 0))`.execute(
-        sDb
-      );
-    }
-    try {
-      return await withAuditContext(request, async () => {
-        return cb();
-      });
-    } finally {
-      if (runId && sDb) {
-        await import_kysely6.sql`SELECT pg_advisory_unlock(hashtextextended(${runId}, 0))`.execute(sDb).catch(() => void 0);
-      }
-    }
+  return withDatabase(db, false, async () => {
+    return withAuditContext(request, async () => {
+      return cb();
+    });
   });
 }
 __name(tryExecuteFlow, "tryExecuteFlow");
@@ -4192,7 +4179,8 @@ var scan = /* @__PURE__ */ __name((name, options) => {
     validate: /* @__PURE__ */ __name(async (data, action) => {
       return options?.validate?.(data, action) ?? true;
     }, "validate"),
-    getData: /* @__PURE__ */ __name((x) => x, "getData")
+    getData: /* @__PURE__ */ __name((x) => x, "getData"),
+    lookup: options?.lookup
   };
 }, "scan");
 
@@ -4401,7 +4389,22 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               throw new Error(`Duplicate step name: ${name}`);
             }
             usedNames.add(name);
-            let step = await db.selectFrom("keel.flow_step").where("run_id", "=", runId).where("name", "=", name).selectAll().executeTakeFirst();
+            const { step, inserted } = 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").where("run_id", "=", runId).where("name", "=", name).selectAll().executeTakeFirst();
+              if (existing) {
+                return { step: existing, inserted: false };
+              }
+              const created = await trx.insertInto("keel.flow_step").values({
+                run_id: runId,
+                name,
+                stage: options.stage,
+                status: "PENDING" /* PENDING */,
+                type: "UI" /* UI */,
+                startTime: /* @__PURE__ */ new Date()
+              }).returningAll().executeTakeFirstOrThrow();
+              return { step: created, inserted: true };
+            });
             if (step && step.status === "COMPLETED" /* COMPLETED */) {
               span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
               span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
@@ -4414,15 +4417,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               }
               return parsedData2;
             }
-            if (!step) {
-              step = await db.insertInto("keel.flow_step").values({
-                run_id: runId,
-                name,
-                stage: options.stage,
-                status: "PENDING" /* PENDING */,
-                type: "UI" /* UI */,
-                startTime: /* @__PURE__ */ new Date()
-              }).returningAll().executeTakeFirst();
+            if (inserted) {
               span.setAttribute("rendered", true);
               span.setAttribute("step.status", "PENDING" /* PENDING */);
               throw new UIRenderDisrupt(