@dudousxd/nestjs-durable-store-mikro-orm 0.9.1 → 0.11.0

package/dist/index.js

@@ -2,18 +2,17 @@ var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 
 // src/entities.ts
-import { Entity, PrimaryKey, Property } from "@mikro-orm/core";
-function _ts_decorate(decorators, target, key, desc) {
-  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
-  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
-  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
-  return c > 3 && r && Object.defineProperty(target, key, r), r;
+import { EntitySchema } from "@mikro-orm/core";
+function snakeCase(property) {
+  return property.replace(/[A-Z]/g, (c) => `_${c.toLowerCase()}`);
 }
-__name(_ts_decorate, "_ts_decorate");
-function _ts_metadata(k, v) {
-  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+__name(snakeCase, "snakeCase");
+function columnNamer(naming) {
+  if (naming === "snake_case") return snakeCase;
+  if (naming === "preserve") return (property) => property;
+  return naming;
 }
-__name(_ts_metadata, "_ts_metadata");
+__name(columnNamer, "columnNamer");
 var WorkflowRunEntity = class {
   static {
     __name(this, "WorkflowRunEntity");
@@ -34,95 +33,6 @@ var WorkflowRunEntity = class {
   createdAt;
   updatedAt;
 };
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "id", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "workflow", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "workflowVersion", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof RunStatus === "undefined" ? Object : RunStatus)
-], WorkflowRunEntity.prototype, "status", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], WorkflowRunEntity.prototype, "input", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], WorkflowRunEntity.prototype, "output", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], WorkflowRunEntity.prototype, "error", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "wakeAt", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "lockedBy", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "lockedUntil", void 0);
-_ts_decorate([
-  Property({
-    type: "integer",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Number)
-], WorkflowRunEntity.prototype, "recoveryAttempts", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], WorkflowRunEntity.prototype, "tags", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], WorkflowRunEntity.prototype, "searchAttributes", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "createdAt", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "updatedAt", void 0);
-WorkflowRunEntity = _ts_decorate([
-  Entity({
-    tableName: "durable_workflow_runs"
-  })
-], WorkflowRunEntity);
 var StepCheckpointEntity = class {
   static {
     __name(this, "StepCheckpointEntity");
@@ -144,93 +54,6 @@ var StepCheckpointEntity = class {
   startedAt;
   finishedAt;
 };
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "runId", void 0);
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", Number)
-], StepCheckpointEntity.prototype, "seq", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "name", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof StepKind === "undefined" ? Object : StepKind)
-], StepCheckpointEntity.prototype, "kind", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "stepId", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "status", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], StepCheckpointEntity.prototype, "input", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], StepCheckpointEntity.prototype, "output", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], StepCheckpointEntity.prototype, "error", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], StepCheckpointEntity.prototype, "events", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", Number)
-], StepCheckpointEntity.prototype, "attempts", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "workerGroup", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "wakeAt", void 0);
-_ts_decorate([
-  Property({
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "enqueuedAt", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "startedAt", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "finishedAt", void 0);
-StepCheckpointEntity = _ts_decorate([
-  Entity({
-    tableName: "durable_step_checkpoints"
-  })
-], StepCheckpointEntity);
 var RunAttributeEntity = class {
   static {
     __name(this, "RunAttributeEntity");
@@ -238,36 +61,8 @@ var RunAttributeEntity = class {
   runId;
   key;
   strValue;
-  // `float`/`double` is portable across SQLite/MySQL/Postgres for numeric range scans.
   numValue;
 };
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", String)
-], RunAttributeEntity.prototype, "runId", void 0);
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", String)
-], RunAttributeEntity.prototype, "key", void 0);
-_ts_decorate([
-  Property({
-    type: "string",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], RunAttributeEntity.prototype, "strValue", void 0);
-_ts_decorate([
-  Property({
-    type: "float",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], RunAttributeEntity.prototype, "numValue", void 0);
-RunAttributeEntity = _ts_decorate([
-  Entity({
-    tableName: "durable_run_attributes"
-  })
-], RunAttributeEntity);
 var SignalWaiterEntity = class {
   static {
     __name(this, "SignalWaiterEntity");
@@ -276,23 +71,6 @@ var SignalWaiterEntity = class {
   runId;
   seq;
 };
-_ts_decorate([
-  PrimaryKey(),
-  _ts_metadata("design:type", String)
-], SignalWaiterEntity.prototype, "token", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", String)
-], SignalWaiterEntity.prototype, "runId", void 0);
-_ts_decorate([
-  Property(),
-  _ts_metadata("design:type", Number)
-], SignalWaiterEntity.prototype, "seq", void 0);
-SignalWaiterEntity = _ts_decorate([
-  Entity({
-    tableName: "durable_signal_waiters"
-  })
-], SignalWaiterEntity);
 var BufferedSignalEntity = class {
   static {
     __name(this, "BufferedSignalEntity");
@@ -301,44 +79,248 @@ var BufferedSignalEntity = class {
   token;
   payload;
 };
-_ts_decorate([
-  PrimaryKey({
-    autoincrement: true
-  }),
-  _ts_metadata("design:type", Number)
-], BufferedSignalEntity.prototype, "id", void 0);
-_ts_decorate([
-  Property({
-    index: true
-  }),
-  _ts_metadata("design:type", String)
-], BufferedSignalEntity.prototype, "token", void 0);
-_ts_decorate([
-  Property({
-    type: "json",
-    nullable: true
-  }),
-  _ts_metadata("design:type", Object)
-], BufferedSignalEntity.prototype, "payload", void 0);
-BufferedSignalEntity = _ts_decorate([
-  Entity({
-    tableName: "durable_buffered_signals"
-  })
-], BufferedSignalEntity);
-var ENTITIES = [
-  WorkflowRunEntity,
-  StepCheckpointEntity,
-  RunAttributeEntity,
-  SignalWaiterEntity,
-  BufferedSignalEntity
-];
+function durableEntities(options = {}) {
+  const col = columnNamer(options.naming ?? "snake_case");
+  const workflowRuns = new EntitySchema({
+    class: WorkflowRunEntity,
+    tableName: "durable_workflow_runs",
+    properties: {
+      id: {
+        type: "string",
+        primary: true,
+        fieldName: col("id")
+      },
+      workflow: {
+        type: "string",
+        fieldName: col("workflow")
+      },
+      workflowVersion: {
+        type: "string",
+        fieldName: col("workflowVersion")
+      },
+      status: {
+        type: "string",
+        fieldName: col("status")
+      },
+      input: {
+        type: "json",
+        nullable: true,
+        fieldName: col("input")
+      },
+      output: {
+        type: "json",
+        nullable: true,
+        fieldName: col("output")
+      },
+      error: {
+        type: "json",
+        nullable: true,
+        fieldName: col("error")
+      },
+      wakeAt: {
+        type: "Date",
+        nullable: true,
+        fieldName: col("wakeAt")
+      },
+      lockedBy: {
+        type: "string",
+        nullable: true,
+        fieldName: col("lockedBy")
+      },
+      lockedUntil: {
+        type: "Date",
+        nullable: true,
+        fieldName: col("lockedUntil")
+      },
+      recoveryAttempts: {
+        type: "integer",
+        nullable: true,
+        fieldName: col("recoveryAttempts")
+      },
+      tags: {
+        type: "json",
+        nullable: true,
+        fieldName: col("tags")
+      },
+      searchAttributes: {
+        type: "json",
+        nullable: true,
+        fieldName: col("searchAttributes")
+      },
+      createdAt: {
+        type: "Date",
+        fieldName: col("createdAt")
+      },
+      updatedAt: {
+        type: "Date",
+        fieldName: col("updatedAt")
+      }
+    }
+  });
+  const stepCheckpoints = new EntitySchema({
+    class: StepCheckpointEntity,
+    tableName: "durable_step_checkpoints",
+    properties: {
+      runId: {
+        type: "string",
+        primary: true,
+        fieldName: col("runId")
+      },
+      seq: {
+        type: "integer",
+        primary: true,
+        fieldName: col("seq")
+      },
+      name: {
+        type: "string",
+        fieldName: col("name")
+      },
+      kind: {
+        type: "string",
+        fieldName: col("kind")
+      },
+      stepId: {
+        type: "string",
+        fieldName: col("stepId")
+      },
+      status: {
+        type: "string",
+        fieldName: col("status")
+      },
+      input: {
+        type: "json",
+        nullable: true,
+        fieldName: col("input")
+      },
+      output: {
+        type: "json",
+        nullable: true,
+        fieldName: col("output")
+      },
+      error: {
+        type: "json",
+        nullable: true,
+        fieldName: col("error")
+      },
+      events: {
+        type: "json",
+        nullable: true,
+        fieldName: col("events")
+      },
+      attempts: {
+        type: "integer",
+        fieldName: col("attempts")
+      },
+      workerGroup: {
+        type: "string",
+        nullable: true,
+        fieldName: col("workerGroup")
+      },
+      wakeAt: {
+        type: "Date",
+        nullable: true,
+        fieldName: col("wakeAt")
+      },
+      enqueuedAt: {
+        type: "Date",
+        nullable: true,
+        fieldName: col("enqueuedAt")
+      },
+      startedAt: {
+        type: "Date",
+        fieldName: col("startedAt")
+      },
+      finishedAt: {
+        type: "Date",
+        fieldName: col("finishedAt")
+      }
+    }
+  });
+  const runAttributes = new EntitySchema({
+    class: RunAttributeEntity,
+    tableName: "durable_run_attributes",
+    properties: {
+      runId: {
+        type: "string",
+        primary: true,
+        fieldName: col("runId")
+      },
+      key: {
+        type: "string",
+        primary: true,
+        fieldName: col("key")
+      },
+      strValue: {
+        type: "string",
+        nullable: true,
+        fieldName: col("strValue")
+      },
+      // `float`/`double` is portable across SQLite/MySQL/Postgres for numeric range scans.
+      numValue: {
+        type: "float",
+        nullable: true,
+        fieldName: col("numValue")
+      }
+    }
+  });
+  const signalWaiters = new EntitySchema({
+    class: SignalWaiterEntity,
+    tableName: "durable_signal_waiters",
+    properties: {
+      token: {
+        type: "string",
+        primary: true,
+        fieldName: col("token")
+      },
+      runId: {
+        type: "string",
+        fieldName: col("runId")
+      },
+      seq: {
+        type: "integer",
+        fieldName: col("seq")
+      }
+    }
+  });
+  const bufferedSignals = new EntitySchema({
+    class: BufferedSignalEntity,
+    tableName: "durable_buffered_signals",
+    properties: {
+      id: {
+        type: "integer",
+        primary: true,
+        autoincrement: true,
+        fieldName: col("id")
+      },
+      token: {
+        type: "string",
+        index: true,
+        fieldName: col("token")
+      },
+      payload: {
+        type: "json",
+        nullable: true,
+        fieldName: col("payload")
+      }
+    }
+  });
+  return [
+    workflowRuns,
+    stepCheckpoints,
+    runAttributes,
+    signalWaiters,
+    bufferedSignals
+  ];
+}
+__name(durableEntities, "durableEntities");
+var ENTITIES = durableEntities();
 
 // src/mikro-orm-state-store.ts
 import { attributeColumnFor, attributeOperand, normalizeAttributeRows, sqlComparator } from "@dudousxd/nestjs-durable-core";
 
 // src/schema.ts
 async function ensureMikroOrmDurableSchema(orm) {
-  await orm.schema.updateSchema({
+  await orm.schema.update({
     safe: true
   });
 }
@@ -554,9 +536,9 @@ var MikroOrmStateStore = class {
   *  the active naming strategy (underscore vs camelCase) rather than guessing. The run PK column is
   *  reused for BOTH the outer correlation and the side-table FK (they share the same fieldName). */
   attributeColumns(em) {
-    const meta = em.getMetadata().get(RunAttributeEntity.name);
-    const field = /* @__PURE__ */ __name((prop) => meta.properties[prop]?.fieldNames?.[0] ?? prop, "field");
-    const runMeta = em.getMetadata().get(WorkflowRunEntity.name);
+    const meta = em.getMetadata().get(RunAttributeEntity);
+    const field = /* @__PURE__ */ __name((prop) => meta.props.find((p) => p.name === prop)?.fieldNames?.[0] ?? prop, "field");
+    const runMeta = em.getMetadata().get(WorkflowRunEntity);
     const runPk = runMeta.properties.id?.fieldNames?.[0] ?? "id";
     return {
       table: meta.tableName,
@@ -569,7 +551,7 @@ var MikroOrmStateStore = class {
   }
   /** Resolve the `tags` column name from metadata (per the active naming strategy), for the raw LIKE. */
   tagsColumn(em) {
-    const runMeta = em.getMetadata().get(WorkflowRunEntity.name);
+    const runMeta = em.getMetadata().get(WorkflowRunEntity);
     return runMeta.properties.tags?.fieldNames?.[0] ?? "tags";
   }
   /** Wrap a JSON column reference so it can be `LIKE`d as text. Postgres `jsonb`/`json` rejects `LIKE`
@@ -646,7 +628,7 @@ var MikroOrmStateStore = class {
       runId: entity.runId,
       seq: entity.seq
     };
-    await em.removeAndFlush(entity);
+    await em.remove(entity).flush();
     return waiter;
   }
   async listSignalWaiters(prefix) {
@@ -681,7 +663,7 @@ var MikroOrmStateStore = class {
     });
     if (!entity) return null;
     const payload = entity.payload ?? void 0;
-    await em.removeAndFlush(entity);
+    await em.remove(entity).flush();
     return {
       payload
     };
@@ -777,6 +759,7 @@ export {
   SignalWaiterEntity,
   StepCheckpointEntity,
   WorkflowRunEntity,
+  durableEntities,
   ensureMikroOrmDurableSchema
 };
 //# sourceMappingURL=index.js.map