DDRCompanion/lib/db/client.ts
2024-05-15 22:37:00 -05:00

35 lines
987 B
TypeScript

import type { BindParams, Database, QueryExecResult } from "sql.js";
import { RpcProvider } from "worker-rpc";
import DbWorker from "./db.worker?worker";
const worker = new DbWorker();
const rpcProvider = new RpcProvider((message, transfer) =>
worker.postMessage(message, transfer),
);
worker.onmessage = (e) => rpcProvider.dispatch(e.data);
interface AsyncDatabase {
exec(sql: string, params?: BindParams): Promise<QueryExecResult[]>;
run(sql: string, params?: BindParams): Promise<Database>;
}
export let dbClient: AsyncDatabase;
export async function getDbClient() {
await new Promise<void>((resolve) => {
rpcProvider.registerSignalHandler("ready", () => {
resolve();
});
});
dbClient = {
async run(s: string, p?: BindParams): Promise<Database> {
return await rpcProvider.rpc("run", { s, p });
},
async exec(s: string, p?: BindParams): Promise<QueryExecResult[]> {
return await rpcProvider.rpc("exec", { s, p });
},
};
}