From f77a41a75237b8c816e0c9ca7217dfacc32cc7d0 Mon Sep 17 00:00:00 2001
From: MohamedBassem <me@mbassem.com>
Date: Sun, 14 Jul 2024 20:42:06 +0000
Subject: chore: Add a new sqlite based queue package

---
 packages/queue/db.ts | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 packages/queue/db.ts

(limited to 'packages/queue/db.ts')

diff --git a/packages/queue/db.ts b/packages/queue/db.ts
new file mode 100644
index 00000000..f1412fef
--- /dev/null
+++ b/packages/queue/db.ts
@@ -0,0 +1,19 @@
+import path from "node:path";
+import Database from "better-sqlite3";
+import { BetterSQLite3Database, drizzle } from "drizzle-orm/better-sqlite3";
+import { migrate } from "drizzle-orm/better-sqlite3/migrator";
+
+import * as schema from "./schema";
+
+export function buildDBClient(dbPath: string, runMigrations = false) {
+  const sqlite = new Database(dbPath);
+  const db = drizzle(sqlite, { schema });
+  if (runMigrations) {
+    migrateDB(db);
+  }
+  return db;
+}
+
+export function migrateDB(db: BetterSQLite3Database<typeof schema>) {
+  migrate(db, { migrationsFolder: path.join(__dirname, "drizzle") });
+}
-- 
cgit v1.2.3-70-g09d2