42 lines
1.4 KiB
TypeScript
42 lines
1.4 KiB
TypeScript
import type { Env } from './env'
|
|
|
|
import pg from 'pg'
|
|
|
|
import { useLogger } from '@guiiai/logg'
|
|
import { migrations } from '@proj-airi/drizzle-migration'
|
|
import { migrate } from '@proj-airi/drizzle-orm-browser-migrator/pg'
|
|
import { drizzle } from 'drizzle-orm/node-postgres'
|
|
|
|
import * as fullSchema from '../schemas'
|
|
|
|
const logger = useLogger('db')
|
|
|
|
export type Database = ReturnType<typeof createDrizzle>['db']
|
|
|
|
type DrizzleEnv = Pick<Env, 'DATABASE_URL' | 'DB_POOL_MAX' | 'DB_POOL_IDLE_TIMEOUT_MS' | 'DB_POOL_CONNECTION_TIMEOUT_MS' | 'DB_POOL_KEEPALIVE_INITIAL_DELAY_MS'>
|
|
|
|
// NOTICE: pg is imported statically here. The OTEL instrumentation hooks are
|
|
// registered via --import ./instrumentation.ts (preload) which runs before
|
|
// tsx loads application modules, allowing require-in-the-middle to patch pg.
|
|
export function createDrizzle(env: DrizzleEnv) {
|
|
const pool = new pg.Pool({
|
|
connectionString: env.DATABASE_URL,
|
|
max: env.DB_POOL_MAX,
|
|
idleTimeoutMillis: env.DB_POOL_IDLE_TIMEOUT_MS,
|
|
connectionTimeoutMillis: env.DB_POOL_CONNECTION_TIMEOUT_MS,
|
|
keepAlive: true,
|
|
keepAliveInitialDelayMillis: env.DB_POOL_KEEPALIVE_INITIAL_DELAY_MS,
|
|
})
|
|
|
|
pool.on('error', (err) => {
|
|
logger.withError(err).error('Unexpected pool error on idle client')
|
|
})
|
|
|
|
const db = drizzle(pool, { schema: fullSchema })
|
|
return { db, pool }
|
|
}
|
|
|
|
export function migrateDatabase(db: Database) {
|
|
return migrate(db, migrations)
|
|
}
|