UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

61 lines 1.92 kB
import { sql } from "../sql/index.js"; function readMigrationFiles({ journal, migrations }) { const migrationQueries = []; for (const journalEntry of journal.entries) { const query = migrations[`m${journalEntry.idx.toString().padStart(4, "0")}`]; if (!query) { throw new Error(`Missing migration: ${journalEntry.tag}`); } try { const result = query.split("--> statement-breakpoint").map((it) => { return it; }); migrationQueries.push({ sql: result, bps: journalEntry.breakpoints, folderMillis: journalEntry.when, hash: "" }); } catch { throw new Error(`Failed to parse migration: ${journalEntry.tag}`); } } return migrationQueries; } async function migrate(db, config) { const migrations = readMigrationFiles(config); db.transaction((tx) => { try { const migrationsTable = "__drizzle_migrations"; const migrationTableCreate = sql` CREATE TABLE IF NOT EXISTS ${sql.identifier(migrationsTable)} ( id SERIAL PRIMARY KEY, hash text NOT NULL, created_at numeric ) `; db.run(migrationTableCreate); const dbMigrations = db.values( sql`SELECT id, hash, created_at FROM ${sql.identifier(migrationsTable)} ORDER BY created_at DESC LIMIT 1` ); const lastDbMigration = dbMigrations[0] ?? void 0; for (const migration of migrations) { if (!lastDbMigration || Number(lastDbMigration[2]) < migration.folderMillis) { for (const stmt of migration.sql) { db.run(sql.raw(stmt)); } db.run( sql`INSERT INTO ${sql.identifier(migrationsTable)} ("hash", "created_at") VALUES(${migration.hash}, ${migration.folderMillis})` ); } } } catch (error) { tx.rollback(); throw error; } }); } export { migrate }; //# sourceMappingURL=migrator.js.map