migrator.ts — drizzle-orm Source File

Architecture documentation for migrator.ts, a typescript file in the drizzle-orm codebase. 3 imports, 0 dependents.

File typescript DrizzleORM RelationalQuery 3 imports 1 functions

Entity Profile

DrizzleORM→ RelationalQuery→ migrator.ts — drizzle-orm Source File

Dependency Diagram

graph LR
  2a844237_8d8e_95f7_04ad_b2253da28464["migrator.ts"]
  89e1a0a3_2876_d680_cdf6_8f2b181caa6d["driver.ts"]
  2a844237_8d8e_95f7_04ad_b2253da28464 --> 89e1a0a3_2876_d680_cdf6_8f2b181caa6d
  e5e6906d_5758_78e9_01f9_fe38ab11030b["migrator.ts"]
  2a844237_8d8e_95f7_04ad_b2253da28464 --> e5e6906d_5758_78e9_01f9_fe38ab11030b
  be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd["sql.ts"]
  2a844237_8d8e_95f7_04ad_b2253da28464 --> be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd
  style 2a844237_8d8e_95f7_04ad_b2253da28464 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

import type { MigrationConfig } from '~/migrator.ts';
import { readMigrationFiles } from '~/migrator.ts';
import { sql } from '~/sql/sql.ts';
import type { DrizzleD1Database } from './driver.ts';

export async function migrate<TSchema extends Record<string, unknown>>(
	db: DrizzleD1Database<TSchema>,
	config: MigrationConfig,
) {
	const migrations = readMigrationFiles(config);
	const migrationsTable = config.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
		)
	`;
	await db.session.run(migrationTableCreate);

	const dbMigrations = await db.values<[number, string, string]>(
		sql`SELECT id, hash, created_at FROM ${sql.identifier(migrationsTable)} ORDER BY created_at DESC LIMIT 1`,
	);

	const lastDbMigration = dbMigrations[0] ?? undefined;

	const statementToBatch = [];

	for (const migration of migrations) {
		if (!lastDbMigration || Number(lastDbMigration[2])! < migration.folderMillis) {
			for (const stmt of migration.sql) {
				statementToBatch.push(db.run(sql.raw(stmt)));
			}

			statementToBatch.push(
				db.run(
					sql`INSERT INTO ${sql.identifier(migrationsTable)} ("hash", "created_at") VALUES(${
						sql.raw(`'${migration.hash}'`)
					}, ${sql.raw(`${migration.folderMillis}`)})`,
				),
			);
		}
	}

	if (statementToBatch.length > 0) {
		await db.session.batch(statementToBatch);
	}
}