Make database type schema agnostic

package.json

@@ -3,7 +3,7 @@
   "name": "mysql-queue",
   "description": "A lite job queue for Node.js",
   "author": "lilac <hi@rabain.com>",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

src/db/index.ts

@@ -4,19 +4,19 @@ import mysql from "mysql2/promise";
 import path from "node:path";
 import * as schema from "./schema";
 
-export type Database = MySql2Database<typeof schema>;
+export type Database = MySql2Database<Record<string, unknown>>;
 
 export const affectedRows = (rawResult: MySqlRawQueryResult) => {
     return rawResult[0].affectedRows
 };
 
-export async function connect(url: string) {
-    const connection = await mysql.createConnection(url);
+export function connect(url: string) {
+    const connection = mysql.createPool(url);
     const db = drizzle(connection, { schema, mode: 'default' });
     return db;
 }
 
-export function migrateDB(db: MySql2Database<Record<string, unknown>>) {
+export function migrateDB(db: Database) {
     return migrate(db, {
         migrationsFolder: path.join(import.meta.dirname, '../drizzle')
     });

src/test.ts

@@ -5,7 +5,7 @@ const defaultUrl = env['DATABASE_URL'] ?? 'mysql://root:root@localhost:3306/queu
 
 
 async function prepareDB(url?: string) {
-  const db = await connect(url ?? defaultUrl);
+  const db = connect(url ?? defaultUrl);
   await migrateDB(db);
   return db;
 }