feat: add programmatic migration runner for better PostgreSQL support

This commit is contained in:
MythEclipse
2026-05-14 16:21:02 +07:00
parent b580430eb6
commit 9889d20edd
2 changed files with 49 additions and 0 deletions

View File

@@ -14,6 +14,7 @@
"test": "vitest run", "test": "vitest run",
"db:generate": "drizzle-kit generate", "db:generate": "drizzle-kit generate",
"db:migrate": "drizzle-kit migrate", "db:migrate": "drizzle-kit migrate",
"db:migrate:programmatic": "tsx src/database/migrate.ts",
"db:studio": "drizzle-kit studio" "db:studio": "drizzle-kit studio"
}, },
"dependencies": { "dependencies": {

48
src/database/migrate.ts Normal file
View File

@@ -0,0 +1,48 @@
import "dotenv/config";
import { migrate } from "drizzle-orm/node-postgres/migrator";
import { migrate as migrateSqlite } from "drizzle-orm/better-sqlite3/migrator";
import { initializeDatabase } from "./drizzle";
import { config } from "../config";
import { createChildLogger } from "../logger";
import Database from "better-sqlite3";
const logger = createChildLogger("migrate");
export async function runMigrations(): Promise<void> {
try {
logger.info("Starting database migrations");
if (config.DATABASE_TYPE === "postgres") {
logger.info("Running PostgreSQL migrations");
const db = await initializeDatabase();
await migrate(db as any, { migrationsFolder: "./drizzle/migrations" });
logger.info("PostgreSQL migrations completed successfully");
} else {
logger.info("Running SQLite migrations");
const sqlite = new Database(".muxer-queue.db");
sqlite.pragma("journal_mode = WAL");
const db = require("drizzle-orm/better-sqlite3").drizzle(sqlite);
migrateSqlite(db, { migrationsFolder: "./drizzle/migrations" });
logger.info("SQLite migrations completed successfully");
}
} catch (error) {
logger.error(
{ error: error instanceof Error ? error.message : String(error) },
"Migration failed"
);
throw error;
}
}
// Run migrations if called directly
if (require.main === module) {
runMigrations()
.then(() => {
logger.info("Migrations completed");
process.exit(0);
})
.catch((error) => {
logger.error({ error }, "Migration failed");
process.exit(1);
});
}