@effect-agent/storage-sqlite 0.1.0-beta.9 → 0.1.0-beta.90

package/src/SqliteSubscriptionStore.ts

@@ -0,0 +1,55 @@
+import { Effect, Layer, Schema } from "effect";
+import {
+  makeSqlSubscriptionStore,
+  SqlSubscriptionTransaction,
+} from "effect-agent/sql-subscription-store";
+import { SourcePartition, SubscriptionError, SubscriptionStore } from "effect-agent/subscription";
+import * as SqlClientService from "effect/unstable/sql/SqlClient";
+
+import { initializeSqliteJournal } from "./internal/sqlite-journal.ts";
+import type { SqliteStorageConfig } from "./SqliteStorageConfig.ts";
+import type { SqliteStorageFailpoint } from "./SqliteStorageFailpoint.ts";
+import type { SqliteStorageInitializationError } from "./SqliteThreadStore.ts";
+
+const transactionLayer = Layer.effect(
+  SqlSubscriptionTransaction,
+  Effect.gen(function* () {
+    const sql = yield* SqlClientService.SqlClient;
+
+    return SqlSubscriptionTransaction.of({
+      run: (body) =>
+        sql
+          .withTransaction(body)
+          .pipe(
+            Effect.catchTag("SqlError", () =>
+              Effect.fail(SubscriptionError.make({ reason: "storage", code: "transaction" })),
+            ),
+          ),
+    });
+  }),
+);
+
+const makeSubscriptionStore = Effect.fn("SqliteSubscriptionStore.make")(function* (
+  owned: SourcePartition,
+) {
+  const partition = yield* Schema.decodeEffect(SourcePartition)(owned).pipe(
+    Effect.mapError(() => SubscriptionError.make({ reason: "validation", code: "partition" })),
+  );
+
+  yield* initializeSqliteJournal();
+
+  return yield* makeSqlSubscriptionStore(partition, {
+    maxStoredJsonLength: 16 * 1024 * 1024,
+  });
+});
+
+export const subscriptionStoreLayer = (
+  partition: SourcePartition,
+): Layer.Layer<
+  SubscriptionStore,
+  SqliteStorageInitializationError | SubscriptionError,
+  SqliteStorageConfig | SqliteStorageFailpoint | SqlClientService.SqlClient
+> =>
+  Layer.effect(SubscriptionStore, makeSubscriptionStore(partition)).pipe(
+    Layer.provide(transactionLayer),
+  );