@dudousxd/nestjs-durable-store-mikro-orm 0.10.0 → 0.11.1

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/decorators/legacy";
-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,110 +33,6 @@ var WorkflowRunEntity = class {
   createdAt;
   updatedAt;
 };
-_ts_decorate([
-  PrimaryKey({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "id", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "workflow", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "workflowVersion", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _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({
-    type: "Date",
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "wakeAt", void 0);
-_ts_decorate([
-  Property({
-    type: "string",
-    nullable: true
-  }),
-  _ts_metadata("design:type", String)
-], WorkflowRunEntity.prototype, "lockedBy", void 0);
-_ts_decorate([
-  Property({
-    type: "Date",
-    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({
-    type: "Date"
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], WorkflowRunEntity.prototype, "createdAt", void 0);
-_ts_decorate([
-  Property({
-    type: "Date"
-  }),
-  _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");
@@ -159,114 +54,6 @@ var StepCheckpointEntity = class {
   startedAt;
   finishedAt;
 };
-_ts_decorate([
-  PrimaryKey({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "runId", void 0);
-_ts_decorate([
-  PrimaryKey({
-    type: "integer"
-  }),
-  _ts_metadata("design:type", Number)
-], StepCheckpointEntity.prototype, "seq", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "name", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", typeof StepKind === "undefined" ? Object : StepKind)
-], StepCheckpointEntity.prototype, "kind", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "stepId", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _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({
-    type: "integer"
-  }),
-  _ts_metadata("design:type", Number)
-], StepCheckpointEntity.prototype, "attempts", void 0);
-_ts_decorate([
-  Property({
-    type: "string",
-    nullable: true
-  }),
-  _ts_metadata("design:type", String)
-], StepCheckpointEntity.prototype, "workerGroup", void 0);
-_ts_decorate([
-  Property({
-    type: "Date",
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "wakeAt", void 0);
-_ts_decorate([
-  Property({
-    type: "Date",
-    nullable: true
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "enqueuedAt", void 0);
-_ts_decorate([
-  Property({
-    type: "Date"
-  }),
-  _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
-], StepCheckpointEntity.prototype, "startedAt", void 0);
-_ts_decorate([
-  Property({
-    type: "Date"
-  }),
-  _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");
@@ -274,40 +61,8 @@ var RunAttributeEntity = class {
   runId;
   key;
   strValue;
-  // `float`/`double` is portable across SQLite/MySQL/Postgres for numeric range scans.
   numValue;
 };
-_ts_decorate([
-  PrimaryKey({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], RunAttributeEntity.prototype, "runId", void 0);
-_ts_decorate([
-  PrimaryKey({
-    type: "string"
-  }),
-  _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");
@@ -316,29 +71,6 @@ var SignalWaiterEntity = class {
   runId;
   seq;
 };
-_ts_decorate([
-  PrimaryKey({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], SignalWaiterEntity.prototype, "token", void 0);
-_ts_decorate([
-  Property({
-    type: "string"
-  }),
-  _ts_metadata("design:type", String)
-], SignalWaiterEntity.prototype, "runId", void 0);
-_ts_decorate([
-  Property({
-    type: "integer"
-  }),
-  _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");
@@ -347,48 +79,273 @@ var BufferedSignalEntity = class {
   token;
   payload;
 };
-_ts_decorate([
-  PrimaryKey({
-    autoincrement: true,
-    type: "integer"
-  }),
-  _ts_metadata("design:type", Number)
-], BufferedSignalEntity.prototype, "id", void 0);
-_ts_decorate([
-  Property({
-    index: true,
-    type: "string"
-  }),
-  _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
+var DURABLE_TABLE_NAMES = /* @__PURE__ */ new Set([
+  "durable_workflow_runs",
+  "durable_step_checkpoints",
+  "durable_run_attributes",
+  "durable_signal_waiters",
+  "durable_buffered_signals"
+]);
+function targetTable(statement) {
+  const match = statement.match(/^(?:create|alter)\s+table\s+(?:if\s+not\s+exists\s+)?[`"']?([\w$]+)[`"']?/i);
+  return match?.[1]?.toLowerCase();
+}
+__name(targetTable, "targetTable");
 async function ensureMikroOrmDurableSchema(orm) {
-  await orm.schema.update({
+  const sql = await orm.schema.getUpdateSchemaSQL({
     safe: true
   });
+  const ours = sql.split(";").map((statement) => statement.trim()).filter((statement) => {
+    const table = targetTable(statement);
+    return table !== void 0 && DURABLE_TABLE_NAMES.has(table);
+  });
+  if (ours.length === 0) {
+    return;
+  }
+  const connection = orm.em.getConnection();
+  for (const statement of ours) {
+    await connection.execute(statement);
+  }
 }
 __name(ensureMikroOrmDurableSchema, "ensureMikroOrmDurableSchema");
 
@@ -825,6 +782,7 @@ export {
   SignalWaiterEntity,
   StepCheckpointEntity,
   WorkflowRunEntity,
+  durableEntities,
   ensureMikroOrmDurableSchema
 };
 //# sourceMappingURL=index.js.map