@reventlessdev/reventless-postgres 3.0.0-alpha.6 → 3.0.0-alpha.7

package/lib/bs/src/DcbEventLogStorage_Postgres.res.mjs

@@ -1,369 +1,52 @@
 // Generated by ReScript, PLEASE EDIT WITH CARE
 
-import * as S from "sury/src/S.res.mjs";
-import * as Stream from "@reventlessdev/rescript-effect/src/Stream.res.mjs";
-import * as Stdlib_Int from "@rescript/runtime/lib/es6/Stdlib_Int.js";
-import * as Stdlib_Array from "@rescript/runtime/lib/es6/Stdlib_Array.js";
-import * as Stdlib_JsExn from "@rescript/runtime/lib/es6/Stdlib_JsExn.js";
-import * as Stdlib_Option from "@rescript/runtime/lib/es6/Stdlib_Option.js";
-import * as Effect from "effect/Effect";
 import * as Pulumi from "@pulumi/pulumi";
-import * as Stdlib_JsError from "@rescript/runtime/lib/es6/Stdlib_JsError.js";
-import * as Primitive_option from "@rescript/runtime/lib/es6/Primitive_option.js";
-import * as Message$Reventless from "@reventlessdev/reventless-spec/src/types/Message.res.mjs";
-import * as Primitive_exceptions from "@rescript/runtime/lib/es6/Primitive_exceptions.js";
-import * as PgDriver$ReventlessPostgres from "./PgDriver.res.mjs";
+import * as DcbEventLogStorage_Postgres_Ops$ReventlessPostgres from "./DcbEventLogStorage_Postgres_Ops.res.mjs";
 
-function strategyParam(s) {
-  if (s === "AdvisoryLocks") {
-    return "advisory";
-  } else {
-    return "rows";
-  }
+function makeStorage(pool, name, indexes, partitionTag, crossPartitionTagKeys, opts, lockStrategyOpt, onAppendedOpt) {
+  let lockStrategy = lockStrategyOpt !== undefined ? lockStrategyOpt : "AdvisoryLocks";
+  let onAppended = onAppendedOpt !== undefined ? onAppendedOpt : DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.noTracking;
+  let match = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.makeOps(pool, name, indexes, partitionTag, crossPartitionTagKeys, opts, lockStrategy, onAppended);
+  let ops = match[1];
+  return [
+    match[0],
+    ops,
+    {
+      resources: [],
+      operations: Pulumi.output(ops)
+    }
+  ];
 }
 
-function noTracking(param) {
-  
-}
+let strategyParam = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.strategyParam;
 
-let selectColumns = "\n  lpad(transaction_id::text, 20, '0') || ':' || lpad(position::text, 20, '0') AS cursor,\n  event_type,\n  tags,\n  data,\n  meta,\n  to_char(recorded_at AT TIME ZONE 'UTC', 'YYYY-MM-DD\"T\"HH24:MI:SS.MS\"Z\"') AS recorded_at\n";
+let noTracking = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.noTracking;
 
-function decodeTag(s) {
-  let i = s.indexOf("=");
-  if (i !== -1) {
-    return {
-      key: s.slice(0, i),
-      value: s.slice(i + 1 | 0, s.length)
-    };
-  } else {
-    return {
-      key: s,
-      value: ""
-    };
-  }
-}
+let selectColumns = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.selectColumns;
 
-function decodeTags(v) {
-  if (v !== undefined) {
-    if (Array.isArray(v)) {
-      return Stdlib_Array.filterMap(v, x => {
-        if (typeof x === "string") {
-          return decodeTag(x);
-        }
-      });
-    } else {
-      return [];
-    }
-  } else {
-    return [];
-  }
-}
+let decodeTag = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.decodeTag;
 
-function rowToEvent(row) {
-  let str = key => {
-    let match = row[key];
-    if (typeof match === "string") {
-      return match;
-    } else {
-      return "";
-    }
-  };
-  let metaJson = row["meta"];
-  let meta = metaJson !== undefined ? S.parseJsonOrThrow(metaJson, Message$Reventless.metaSchema) : Stdlib_JsError.throwWithMessage("missing meta column in dcb_event row");
-  return {
-    position: str("cursor"),
-    eventType: str("event_type"),
-    data: Stdlib_Option.getOr(row["data"], null),
-    tags: decodeTags(row["tags"]),
-    meta: meta,
-    recordedAt: str("recorded_at")
-  };
-}
+let decodeTags = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.decodeTags;
 
-function mkBuilder() {
-  return {
-    params: []
-  };
-}
+let rowToEvent = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.rowToEvent;
 
-function param(b, v) {
-  b.params.push(v);
-  return "$" + b.params.length.toString();
-}
+let mkBuilder = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.mkBuilder;
 
-function buildReadWhere(b, name, query, after, applyFence) {
-  let parts = [`log_name = ` + param(b, name)];
-  if (applyFence) {
-    parts.push("transaction_id < pg_snapshot_xmin(pg_current_snapshot())");
-  }
-  if (after !== undefined) {
-    let txP = param(b, after[0]);
-    let posP = param(b, after[1]);
-    parts.push(`(transaction_id > ` + txP + `::xid8 OR (transaction_id = ` + txP + `::xid8 AND position > ` + posP + `::bigint))`);
-  }
-  if (query.length !== 0) {
-    let clauseSqls = query.map(clause => {
-      let clauseParts = [];
-      let types = clause.eventTypes;
-      if (types !== undefined && types.length !== 0) {
-        let ph = types.map(t => param(b, t)).join(",");
-        clauseParts.push(`event_type IN (` + ph + `)`);
-      }
-      let tags = clause.tags;
-      if (tags !== undefined && tags.length !== 0) {
-        let arr = tags.map(t => param(b, t.key + "=" + t.value)).join(",");
-        clauseParts.push(`tags @> ARRAY[` + arr + `]::text[]`);
-      }
-      if (clauseParts.length === 0) {
-        return "TRUE";
-      } else {
-        return "(" + clauseParts.join(" AND ") + ")";
-      }
-    });
-    parts.push("(" + clauseSqls.join(" OR ") + ")");
-  }
-  return parts.join(" AND ");
-}
+let param = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.param;
 
-function stripZeros(s) {
-  let t = s.replace(/^0+/, "");
-  if (t === "") {
-    return "0";
-  } else {
-    return t;
-  }
-}
+let buildReadWhere = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.buildReadWhere;
 
-function decodeCursor(pos) {
-  let match = pos.split(":");
-  if (match.length !== 2) {
-    return;
-  }
-  let tx = match[0];
-  let p = match[1];
-  return [
-    stripZeros(tx),
-    stripZeros(p)
-  ];
-}
+let stripZeros = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.stripZeros;
 
-function isConflict(exn) {
-  let msg = Stdlib_Option.getOr(Stdlib_Option.flatMap(Stdlib_JsExn.fromException(exn), Stdlib_JsExn.message), "");
-  if (msg.includes("dcb_conflict") || msg.includes("duplicate key")) {
-    return true;
-  } else {
-    return msg.includes("23505");
-  }
-}
+let decodeCursor = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.decodeCursor;
 
-async function countAll(pool) {
-  let row = await PgDriver$ReventlessPostgres.queryOne(pool, "SELECT COUNT(*)::bigint AS c FROM dcb_event", []);
-  if (row === undefined) {
-    return 0;
-  }
-  let match = row["c"];
-  if (match === undefined) {
-    return 0;
-  }
-  switch (typeof match) {
-    case "string" :
-      return Stdlib_Option.getOr(Stdlib_Int.fromString(match, undefined), 0);
-    case "number" :
-      return match | 0;
-    default:
-      return 0;
-  }
-}
+let pageSize = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.pageSize;
 
-function makeStorage(pool, name, param, param$1, param$2, param$3, lockStrategyOpt, onAppendedOpt) {
-  let lockStrategy = lockStrategyOpt !== undefined ? lockStrategyOpt : "AdvisoryLocks";
-  let onAppended = onAppendedOpt !== undefined ? onAppendedOpt : noTracking;
-  let currentHead = async () => {
-    let row = await PgDriver$ReventlessPostgres.queryOne(pool, "SELECT lpad(transaction_id::text, 20, '0') || ':' || lpad(position::text, 20, '0') AS head\n         FROM dcb_event\n        WHERE log_name = $1 AND transaction_id < pg_snapshot_xmin(pg_current_snapshot())\n        ORDER BY transaction_id DESC, position DESC\n        LIMIT 1", [name]);
-    if (row === undefined) {
-      return;
-    }
-    let match = row["head"];
-    if (typeof match === "string") {
-      return match;
-    }
-  };
-  let runQuery = async (query, after) => {
-    let b = {
-      params: []
-    };
-    let where = buildReadWhere(b, name, query, after, true);
-    let sql = `SELECT ` + selectColumns + ` FROM dcb_event WHERE ` + where + ` ORDER BY transaction_id ASC, position ASC`;
-    return (await PgDriver$ReventlessPostgres.query(pool, sql, b.params)).map(rowToEvent);
-  };
-  let read = async (query, after) => {
-    let afterCols = Stdlib_Option.flatMap(after, decodeCursor);
-    let events = await runQuery(query, afterCols);
-    let head = await currentHead();
-    if (head !== undefined) {
-      return {
-        events: events,
-        headPosition: head
-      };
-    } else {
-      return {
-        events: events
-      };
-    }
-  };
-  let append = async (newEvents, condition) => {
-    let eventsJson = newEvents.map(ev => (Object.fromEntries([
-      [
-        "event_type",
-        ev.eventType
-      ],
-      [
-        "tags",
-        ev.tags.map(t => t.key + "=" + t.value)
-      ],
-      [
-        "data",
-        ev.data
-      ],
-      [
-        "meta",
-        S.reverseConvertToJsonOrThrow(ev.meta, Message$Reventless.metaSchema)
-      ]
-    ])));
-    let conditionParam;
-    if (condition !== undefined) {
-      let queryJson = condition.query.map(clause => {
-        let types = clause.eventTypes;
-        let tags = clause.tags;
-        return Object.fromEntries([
-          [
-            "eventTypes",
-            types !== undefined ? types.map(prim => prim) : null
-          ],
-          [
-            "tags",
-            tags !== undefined ? tags.map(t => t.key + "=" + t.value) : null
-          ]
-        ]);
-      });
-      let fields = [[
-          "query",
-          queryJson
-        ]];
-      let a = condition.after;
-      if (a !== undefined) {
-        fields.push([
-          "after",
-          a
-        ]);
-      }
-      conditionParam = JSON.stringify(Object.fromEntries(fields));
-    } else {
-      conditionParam = null;
-    }
-    try {
-      let rows = await PgDriver$ReventlessPostgres.query(pool, "SELECT dcb_append($1, $2::jsonb, $3::jsonb, $4) AS pos", [
-        name,
-        JSON.stringify(eventsJson),
-        conditionParam,
-        strategyParam(lockStrategy)
-      ]);
-      let match = Stdlib_Option.flatMap(rows[0], r => r["pos"]);
-      if (match === undefined) {
-        return {
-          TAG: "Error",
-          _0: {
-            TAG: "StorageFailure",
-            _0: "dcb_append returned no position"
-          }
-        };
-      }
-      if (typeof match !== "string") {
-        return {
-          TAG: "Error",
-          _0: {
-            TAG: "StorageFailure",
-            _0: "dcb_append returned no position"
-          }
-        };
-      }
-      onAppended(Stdlib_Array.filterMap(newEvents, ev => {
-        let match$1 = decodeCursor(match);
-        if (match$1 !== undefined) {
-          return Stdlib_Option.map(Stdlib_Int.fromString(match$1[1], undefined), p => [
-            ev.meta.msgId,
-            p
-          ]);
-        }
-      }));
-      return {
-        TAG: "Ok",
-        _0: match
-      };
-    } catch (raw_exn) {
-      let exn = Primitive_exceptions.internalToException(raw_exn);
-      if (isConflict(exn)) {
-        return {
-          TAG: "Error",
-          _0: "Conflict"
-        };
-      } else {
-        return {
-          TAG: "Error",
-          _0: {
-            TAG: "StorageFailure",
-            _0: Stdlib_Option.getOr(Stdlib_Option.flatMap(Stdlib_JsExn.fromException(exn), Stdlib_JsExn.message), "storage error")
-          }
-        };
-      }
-    }
-  };
-  let readStream = (query, after, param) => Stream.paginateEffect(Stdlib_Option.flatMap(after, decodeCursor), cursor => Effect.map(Effect.promise(() => {
-    let b = {
-      params: []
-    };
-    let where = buildReadWhere(b, name, query, cursor, true);
-    let sql = `SELECT ` + selectColumns + `, transaction_id::text AS tx_raw, position::text AS pos_raw FROM dcb_event WHERE ` + where + ` ORDER BY transaction_id ASC, position ASC LIMIT ` + (500).toString();
-    return PgDriver$ReventlessPostgres.query(pool, sql, b.params);
-  }), rows => {
-    let events = rows.map(rowToEvent);
-    let next;
-    if (rows.length < 500) {
-      next = undefined;
-    } else {
-      let lastRow = rows[rows.length - 1 | 0];
-      if (lastRow !== undefined) {
-        let match = lastRow["tx_raw"];
-        let match$1 = lastRow["pos_raw"];
-        next = typeof match === "string" && typeof match$1 === "string" ? Primitive_option.some([
-            match,
-            match$1
-          ]) : undefined;
-      } else {
-        next = undefined;
-      }
-    }
-    return [
-      events,
-      next
-    ];
-  }));
-  let ops = {
-    read: read,
-    append: append,
-    readStream: readStream
-  };
-  return [
-    name,
-    ops,
-    {
-      resources: [],
-      operations: Pulumi.output(ops)
-    }
-  ];
-}
+let isConflict = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.isConflict;
+
+let countAll = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.countAll;
 
-let pageSize = 500;
+let makeOps = DcbEventLogStorage_Postgres_Ops$ReventlessPostgres.makeOps;
 
 export {
   strategyParam,
@@ -380,6 +63,7 @@ export {
   pageSize,
   isConflict,
   countAll,
+  makeOps,
   makeStorage,
 }
-/* S Not a pure module */
+/* @pulumi/pulumi Not a pure module */