@prisma/streams-server 0.1.1 → 0.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prisma/streams-server",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Bun-only self-hosted Prisma Streams server.",
   "repository": {
     "type": "git",

package/src/runtime/host_runtime.ts

@@ -0,0 +1,5 @@
+export type HostRuntime = "bun" | "node";
+
+export function detectHostRuntime(): HostRuntime {
+  return typeof (globalThis as any).Bun !== "undefined" || Boolean(process.versions?.bun) ? "bun" : "node";
+}

package/src/segment/segmenter_worker.ts

@@ -1,13 +1,16 @@
 import { parentPort, workerData } from "node:worker_threads";
 import type { Config } from "../config.ts";
 import { SqliteDurableStore } from "../db/db.ts";
+import type { HostRuntime } from "../runtime/host_runtime.ts";
+import { setSqliteRuntimeOverride } from "../sqlite/adapter.ts";
 import { Segmenter, type SegmenterHooks, type SegmenterOptions } from "./segmenter.ts";
 import { initConsoleLogging } from "../util/log.ts";
 
 initConsoleLogging();
 
-const data = workerData as { config: Config; opts?: SegmenterOptions };
+const data = workerData as { config: Config; hostRuntime?: HostRuntime; opts?: SegmenterOptions };
 const cfg = data.config;
+setSqliteRuntimeOverride(data.hostRuntime ?? null);
 // The main server process initializes/migrates schema; workers should avoid
 // concurrent migrations on the same sqlite file.
 const db = new SqliteDurableStore(cfg.dbPath, { cacheBytes: cfg.sqliteCacheBytes, skipMigrations: true });

package/src/segment/segmenter_workers.ts

@@ -3,6 +3,7 @@ import { resolve } from "node:path";
 import { fileURLToPath } from "node:url";
 import { Worker } from "node:worker_threads";
 import type { Config } from "../config";
+import { detectHostRuntime } from "../runtime/host_runtime.ts";
 import type { SegmenterHooks, SegmenterOptions } from "./segmenter";
 
 export type SegmenterController = {
@@ -63,6 +64,7 @@ export class SegmenterWorkerPool implements SegmenterController {
     const worker = new Worker(workerSpec, {
       workerData: {
         config: this.config,
+        hostRuntime: detectHostRuntime(),
         opts: this.opts,
       },
       type: "module",

package/src/sqlite/adapter.ts

@@ -1,3 +1,6 @@
+import { createRequire } from "node:module";
+import type { HostRuntime } from "../runtime/host_runtime.ts";
+import { detectHostRuntime } from "../runtime/host_runtime.ts";
 import { dsError } from "../util/ds_error.ts";
 export interface SqliteStatement {
   get(...params: any[]): any;
@@ -149,16 +152,37 @@ class NodeDatabaseAdapter implements SqliteDatabase {
 }
 
 let openImpl: ((path: string) => SqliteDatabase) | null = null;
+let openImplRuntime: HostRuntime | null = null;
+let runtimeOverride: HostRuntime | null = null;
+const require = createRequire(import.meta.url);
 
-if (typeof (globalThis as any).Bun !== "undefined") {
-  const { Database } = await import("bun:sqlite");
-  openImpl = (path: string) => new BunDatabaseAdapter(new Database(path));
-} else {
-  const { DatabaseSync } = await import("node:sqlite");
-  openImpl = (path: string) => new NodeDatabaseAdapter(new DatabaseSync(path));
+function selectedRuntime(): HostRuntime {
+  return runtimeOverride ?? detectHostRuntime();
+}
+
+function buildOpenImpl(runtime: HostRuntime): (path: string) => SqliteDatabase {
+  if (runtime === "bun") {
+    const { Database } = require("bun:sqlite") as { Database: new (path: string) => any };
+    return (path: string) => new BunDatabaseAdapter(new Database(path));
+  }
+  const { DatabaseSync } = require("node:sqlite") as { DatabaseSync: new (path: string) => any };
+  return (path: string) => new NodeDatabaseAdapter(new DatabaseSync(path));
+}
+
+export function setSqliteRuntimeOverride(runtime: HostRuntime | null): void {
+  runtimeOverride = runtime;
+  if (runtimeOverride && openImplRuntime && runtimeOverride !== openImplRuntime) {
+    openImpl = null;
+    openImplRuntime = null;
+  }
 }
 
 export function openSqliteDatabase(path: string): SqliteDatabase {
+  const runtime = selectedRuntime();
+  if (!openImpl || openImplRuntime !== runtime) {
+    openImpl = buildOpenImpl(runtime);
+    openImplRuntime = runtime;
+  }
   if (!openImpl) throw dsError("sqlite adapter not initialized");
   return openImpl(path);
 }

package/src/touch/interpreter_worker.ts

@@ -2,6 +2,8 @@ import { parentPort, workerData } from "node:worker_threads";
 import { Result } from "better-result";
 import type { Config } from "../config.ts";
 import { SqliteDurableStore } from "../db/db.ts";
+import type { HostRuntime } from "../runtime/host_runtime.ts";
+import { setSqliteRuntimeOverride } from "../sqlite/adapter.ts";
 import { initConsoleLogging } from "../util/log.ts";
 import type { ProcessRequest } from "./worker_protocol.ts";
 import { interpretRecordToChanges } from "./engine.ts";
@@ -10,8 +12,11 @@ import { isTouchEnabled } from "./spec.ts";
 
 initConsoleLogging();
 
-const data = workerData as { config: Config };
+const data = workerData as { config: Config; hostRuntime?: HostRuntime };
 const cfg = data.config;
+// Bun worker_threads can miss the Bun globals that the main thread sees.
+// Use the parent host runtime hint before the worker opens SQLite.
+setSqliteRuntimeOverride(data.hostRuntime ?? null);
 // The main server process initializes/migrates schema; workers should avoid
 // concurrent migrations on the same sqlite file.
 const db = new SqliteDurableStore(cfg.dbPath, { cacheBytes: cfg.sqliteCacheBytes, skipMigrations: true });

package/src/touch/worker_pool.ts

@@ -4,6 +4,7 @@ import { fileURLToPath } from "node:url";
 import { Worker } from "node:worker_threads";
 import { Result } from "better-result";
 import type { Config } from "../config";
+import { detectHostRuntime } from "../runtime/host_runtime.ts";
 import type { ProcessRequest, ProcessResult, WorkerMessage } from "./worker_protocol";
 import { dsError } from "../util/ds_error.ts";
 
@@ -123,7 +124,7 @@ export class TouchInterpreterWorkerPool {
     }
 
     const worker = new Worker(workerSpec, {
-      workerData: { config: this.cfg },
+      workerData: { config: this.cfg, hostRuntime: detectHostRuntime() },
       type: "module",
       smol: true,
     } as any);