@reventlessdev/reventless-local 3.0.0-alpha.113

package/src/adapter/DcbEventLog/DcbEventLogStorage_Sqlite.res.mjs

@@ -0,0 +1,299 @@
+// Generated by ReScript, PLEASE EDIT WITH CARE
+
+import * as S from "sury/src/S.res.mjs";
+import * as Stdlib_Int from "@rescript/runtime/lib/es6/Stdlib_Int.js";
+import * as Stdlib_Option from "@rescript/runtime/lib/es6/Stdlib_Option.js";
+import * as Effect from "effect/Effect";
+import * as Stream from "effect/Stream";
+import * as Pulumi from "@pulumi/pulumi";
+import * as Stdlib_JsError from "@rescript/runtime/lib/es6/Stdlib_JsError.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 Message$ReventlessCore from "@reventlessdev/reventless-core/src/Message.res.mjs";
+import * as SqliteDriver$ReventlessLocal from "../SqliteDriver.res.mjs";
+
+function ensureSchema(db) {
+  SqliteDriver$ReventlessLocal.exec(db, "CREATE TABLE IF NOT EXISTS dcb_event (log_name TEXT NOT NULL, position INTEGER NOT NULL, event_type TEXT NOT NULL, data TEXT NOT NULL, meta TEXT NOT NULL, recorded_at TEXT NOT NULL, PRIMARY KEY (log_name, position))");
+  SqliteDriver$ReventlessLocal.exec(db, "CREATE TABLE IF NOT EXISTS dcb_tag (log_name TEXT NOT NULL, position INTEGER NOT NULL, tag_key TEXT NOT NULL, tag_value TEXT NOT NULL)");
+  SqliteDriver$ReventlessLocal.exec(db, "CREATE INDEX IF NOT EXISTS dcb_tag_by_kv ON dcb_tag(log_name, tag_key, tag_value, position)");
+}
+
+function posToInt(pos) {
+  return Stdlib_Option.getOr(Stdlib_Int.fromString(pos, undefined), 0);
+}
+
+function parseDataPayload(s) {
+  try {
+    return JSON.parse(s);
+  } catch (exn) {
+    return null;
+  }
+}
+
+function countAll(db) {
+  ensureSchema(db);
+  let row = SqliteDriver$ReventlessLocal.get(SqliteDriver$ReventlessLocal.prepare(db, "SELECT COUNT(*) AS c FROM dcb_event"), []);
+  if (row === undefined) {
+    return 0;
+  }
+  let match = row["c"];
+  if (typeof match === "number") {
+    return match | 0;
+  } else {
+    return 0;
+  }
+}
+
+function coercePosition(v) {
+  switch (typeof v) {
+    case "string" :
+      return posToInt(v);
+    case "number" :
+      return v | 0;
+    default:
+      return 0;
+  }
+}
+
+function sqlEscape(s) {
+  return s.replaceAll("'", "''");
+}
+
+function buildClauseSql(logName, clause, after) {
+  let parts = {
+    contents: ["log_name = ?"]
+  };
+  let params = {
+    contents: [logName]
+  };
+  if (after !== undefined) {
+    parts.contents = parts.contents.concat(["position > ?"]);
+    params.contents = params.contents.concat([after]);
+  }
+  let types = clause.eventTypes;
+  if (types !== undefined && types.length !== 0) {
+    let placeholders = types.map(param => "?").join(",");
+    parts.contents = parts.contents.concat([`event_type IN (` + placeholders + `)`]);
+    types.forEach(t => {
+      params.contents = params.contents.concat([t]);
+    });
+  }
+  let tags = clause.tags;
+  if (tags !== undefined && tags.length !== 0) {
+    tags.forEach(tag => {
+      parts.contents = parts.contents.concat([`EXISTS (SELECT 1 FROM dcb_tag t WHERE t.log_name = ? AND t.position = dcb_event.position AND t.tag_key = ? AND t.tag_value = ?)`]);
+      params.contents = params.contents.concat([
+        logName,
+        tag.key,
+        tag.value
+      ]);
+    });
+  }
+  return [
+    "(" + parts.contents.join(" AND ") + ")",
+    params.contents
+  ];
+}
+
+function buildQuerySql(logName, query, after) {
+  if (query.length === 0) {
+    let baseWhere = after !== undefined ? "log_name = ? AND position > ?" : "log_name = ?";
+    let baseParams = after !== undefined ? [
+        logName,
+        after
+      ] : [logName];
+    return [
+      `SELECT position, event_type, data, meta, recorded_at FROM dcb_event WHERE ` + baseWhere + ` ORDER BY position ASC`,
+      baseParams
+    ];
+  }
+  let clauses = query.map(c => buildClauseSql(logName, c, after));
+  let combinedWhere = clauses.map(param => param[0]).join(" OR ");
+  let combinedParams = clauses.flatMap(param => param[1]);
+  return [
+    `SELECT position, event_type, data, meta, recorded_at FROM dcb_event WHERE ` + combinedWhere + ` ORDER BY position ASC`,
+    combinedParams
+  ];
+}
+
+function makeStorage(db, bus, publishToTopic, name, param, param$1, param$2) {
+  ensureSchema(db);
+  let insertEventStmt = SqliteDriver$ReventlessLocal.prepare(db, "INSERT INTO dcb_event(log_name, position, event_type, data, meta, recorded_at) VALUES(?,?,?,?,?,?)");
+  let insertTagStmt = SqliteDriver$ReventlessLocal.prepare(db, "INSERT INTO dcb_tag(log_name, position, tag_key, tag_value) VALUES(?,?,?,?)");
+  let maxPositionStmt = SqliteDriver$ReventlessLocal.prepare(db, "SELECT COALESCE(MAX(position), 0) AS m FROM dcb_event WHERE log_name = ?");
+  SqliteDriver$ReventlessLocal.prepare(db, "SELECT MAX(position) AS m FROM dcb_event WHERE log_name = ?");
+  let tagsForPositionsStmt = SqliteDriver$ReventlessLocal.prepare(db, "SELECT position, tag_key, tag_value FROM dcb_tag WHERE log_name = ? AND position = ?");
+  let currentMaxPosition = () => {
+    let row = SqliteDriver$ReventlessLocal.get(maxPositionStmt, [name]);
+    if (row === undefined) {
+      return 0;
+    }
+    let v = row["m"];
+    if (v !== undefined) {
+      return coercePosition(v);
+    } else {
+      return 0;
+    }
+  };
+  let tagsForPosition = pos => SqliteDriver$ReventlessLocal.all(tagsForPositionsStmt, [
+    name,
+    pos
+  ]).map(row => {
+    let match = row["tag_key"];
+    let key = typeof match === "string" ? match : "";
+    let match$1 = row["tag_value"];
+    let value = typeof match$1 === "string" ? match$1 : "";
+    return {
+      key: key,
+      value: value
+    };
+  });
+  let runQuery = (query, after) => {
+    let match = buildQuerySql(name, query, after);
+    let stmt = SqliteDriver$ReventlessLocal.prepare(db, match[0]);
+    return SqliteDriver$ReventlessLocal.all(stmt, match[1]).map(row => {
+      let v = row["position"];
+      let position = v !== undefined ? coercePosition(v) : 0;
+      let match = row["event_type"];
+      let eventType = typeof match === "string" ? match : "";
+      let match$1 = row["data"];
+      let data = typeof match$1 === "string" ? parseDataPayload(match$1) : null;
+      let match$2 = row["meta"];
+      let meta;
+      if (typeof match$2 === "string") {
+        let exit = 0;
+        let metaJson;
+        try {
+          metaJson = JSON.parse(match$2);
+          exit = 1;
+        } catch (exn) {
+          meta = Stdlib_JsError.throwWithMessage("invalid meta JSON in dcb_event row");
+        }
+        if (exit === 1) {
+          meta = S.parseJsonOrThrow(metaJson, Message$Reventless.metaSchema);
+        }
+      } else {
+        meta = Stdlib_JsError.throwWithMessage("missing meta column in dcb_event row");
+      }
+      let match$3 = row["recorded_at"];
+      let recordedAt = typeof match$3 === "string" ? match$3 : "";
+      return {
+        position: position.toString(),
+        eventType: eventType,
+        data: data,
+        tags: tagsForPosition(position),
+        meta: meta,
+        recordedAt: recordedAt
+      };
+    });
+  };
+  let read = async (query, after) => {
+    let afterInt = Stdlib_Option.map(after, posToInt);
+    let events = runQuery(query, afterInt);
+    let head = currentMaxPosition();
+    if (head > 0) {
+      return {
+        events: events,
+        headPosition: head.toString()
+      };
+    } else {
+      return {
+        events: events
+      };
+    }
+  };
+  let append = async (newEvents, condition) => {
+    let result = {
+      contents: {
+        TAG: "Ok",
+        _0: ""
+      }
+    };
+    try {
+      SqliteDriver$ReventlessLocal.transaction(db, () => {
+        if (condition !== undefined) {
+          let afterInt = Stdlib_Option.map(condition.after, posToInt);
+          let conflicting = runQuery(condition.query, afterInt);
+          if (conflicting.length !== 0) {
+            throw {
+              RE_EXN_ID: "Failure",
+              _1: "conflict: condition check failed",
+              Error: new Error()
+            };
+          }
+        }
+        let mutable_ = {
+          contents: currentMaxPosition()
+        };
+        let lastPos = {
+          contents: mutable_.contents
+        };
+        newEvents.forEach(event => {
+          mutable_.contents = mutable_.contents + 1 | 0;
+          let pos = mutable_.contents;
+          let metaJson = S.reverseConvertToJsonOrThrow(event.meta, Message$Reventless.metaSchema);
+          let recordedAt = Message$ReventlessCore.nowAsISOString();
+          SqliteDriver$ReventlessLocal.run(insertEventStmt, [
+            name,
+            pos,
+            event.eventType,
+            JSON.stringify(event.data),
+            JSON.stringify(metaJson),
+            recordedAt
+          ]);
+          event.tags.forEach(tag => SqliteDriver$ReventlessLocal.run(insertTagStmt, [
+            name,
+            pos,
+            tag.key,
+            tag.value
+          ]));
+          lastPos.contents = pos;
+        });
+        result.contents = {
+          TAG: "Ok",
+          _0: lastPos.contents.toString()
+        };
+      });
+      return result.contents;
+    } catch (raw_msg) {
+      let msg = Primitive_exceptions.internalToException(raw_msg);
+      if (msg.RE_EXN_ID === "Failure") {
+        return {
+          TAG: "Error",
+          _0: msg._1
+        };
+      } else {
+        return {
+          TAG: "Error",
+          _0: "conflict"
+        };
+      }
+    }
+  };
+  let readStream = (query, after, param) => Stream.flatMap(Stream.fromEffect(Effect.map(Effect.promise(() => read(query, after)), result => result.events)), arr => Stream.fromIterable(arr));
+  return [
+    name,
+    read,
+    {
+      resources: [],
+      operations: Pulumi.output({
+        read: read,
+        append: append,
+        readStream: readStream
+      })
+    }
+  ];
+}
+
+export {
+  ensureSchema,
+  posToInt,
+  parseDataPayload,
+  countAll,
+  coercePosition,
+  sqlEscape,
+  buildClauseSql,
+  buildQuerySql,
+  makeStorage,
+}
+/* S Not a pure module */