method to get db from db wrapper

vlcn-io · Dec 1, 2023 · 5fa2543 · 5fa2543
1 parent 5f8858f
commit 5fa2543
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/packages/ws-litefs/src/internal/LiteFSDB.ts b/packages/ws-litefs/src/internal/LiteFSDB.ts
@@ -2,6 +2,7 @@ import { Change } from "@vlcn.io/ws-common";
 import { IDB } from "@vlcn.io/ws-server";
 import { PrimaryConnection } from "./PrimaryConnection.js";
 import logger from "../logger.js";
+import type BetterSqlite3 from "better-sqlite3";
 
 export class LiteFSDB implements IDB {
   readonly #proxied;
@@ -28,6 +29,25 @@ export class LiteFSDB implements IDB {
     return this.#proxied.schemaVersion;
   }
 
+  getDB(): BetterSqlite3.Database {
+    throw new Error(
+      "get db is unsafe under litefs deployments as litefs requires write forwarding to the primary"
+    );
+  }
+
+  // read<T>(sql: string, params: unknown[]): T[] {
+  //   return this.#proxied.read(sql, params);
+  // }
+
+  // async write(sql: string, params: unknown[]): Promise<void> {
+  //   if (!this.#primaryConnection.isPrimary()) {
+  //     logger.info("Proxying write to primary");
+  //     await this.#primaryConnection.writeOnPrimary(this.#room, sql, params);
+  //   } else {
+  //     await this.#proxied.write(sql, params);
+  //   }
+  // }
+
   getLastSeen(site: Uint8Array): [bigint, number] {
     return this.#proxied.getLastSeen(site);
   }

diff --git a/packages/ws-litefs/src/internal/PrimaryConnection.ts b/packages/ws-litefs/src/internal/PrimaryConnection.ts
@@ -118,6 +118,10 @@ export class PrimaryConnection {
     });
   }
 
+  writeOnPrimary(room: string, sql: string, params: unknown[]): Promise<void> {
+    throw new Error("not implemented");
+  }
+
   applyChangesOnPrimary(
     room: string,
     changes: readonly Change[],

diff --git a/packages/ws-server/src/DB.ts b/packages/ws-server/src/DB.ts
@@ -1,4 +1,5 @@
 import Database from "better-sqlite3";
+import type BetterSqlite3 from "better-sqlite3";
 import { Config } from "./config.js";
 import path from "node:path";
 import fs from "node:fs";
@@ -13,6 +14,13 @@ export interface IDB {
   readonly schemaName: string;
   readonly schemaVersion: bigint;
 
+  /**
+   * This will work for single-node SQLite setups.
+   * Does not work under Turso.
+   * Will work for reads under LiteFS but not writes.
+   */
+  getDB(): BetterSqlite3.Database;
+
   getLastSeen(site: Uint8Array): [bigint, number];
   applyChangesetAndSetLastSeen(
     changes: readonly Change[],
@@ -208,6 +216,10 @@ export default class DB implements IDB {
     return this.#schemaVersion;
   }
 
+  getDB(): BetterSqlite3.Database {
+    return this.#db;
+  }
+
   getLastSeen(site: Uint8Array): [bigint, number] {
     const result = this.#getLastSeenStmt.raw(true).get(site) as
       | [bigint, bigint]