mirror of
https://github.com/simstudioai/sim.git
synced 2026-10-16 06:39:56 +08:00
Theodore Li
0613cebfab
* feat(db): resolve DATABASE_URL per role (DATABASE_URL_<ROLE> with fallback) * fix(db): pin realtime process to SIM_DB_ROLE=realtime so both pools share the role Without it, the realtime process left SIM_DB_ROLE unset: the shared @sim/db client defaulted role to 'web' (web pool profile + DATABASE_URL_WEB) while socketDb used 'realtime', so the two pools diverged after cutover. Set it at the process level (bootstrap + dev/start scripts), mirroring DB_APP_NAME, so the shared client and socketDb both resolve the realtime profile and URL.
74 lines
2.4 KiB
TypeScript
74 lines
2.4 KiB
TypeScript
import { drizzle } from 'drizzle-orm/postgres-js'
|
|
import postgres from 'postgres'
|
|
import { resolveDbUrl } from './connection-url'
|
|
import * as schema from './schema'
|
|
import { instrumentPoolClient } from './tx-tripwire'
|
|
|
|
/**
|
|
* Per-role pool profiles. Starting numbers — validate against real per-role
|
|
* process counts (PgBouncer transaction mode, max_connections=200).
|
|
*/
|
|
export const DB_POOL_PROFILES = {
|
|
web: { primaryMax: 10, replicaMax: 4, appName: 'sim-app' },
|
|
// 5, not 3 — one run can need 3+ simultaneous connections (parallel queries +
|
|
// overlapping logging writes); 3 risks intra-run deadlock.
|
|
trigger: { primaryMax: 5, replicaMax: 2, appName: 'sim-trigger' },
|
|
realtime: { primaryMax: 5, replicaMax: 3, appName: 'sim-realtime' },
|
|
} as const
|
|
|
|
type DbRole = keyof typeof DB_POOL_PROFILES
|
|
|
|
const roleEnv = process.env.SIM_DB_ROLE?.trim()
|
|
if (roleEnv && !Object.hasOwn(DB_POOL_PROFILES, roleEnv)) {
|
|
throw new Error(
|
|
`Invalid SIM_DB_ROLE '${roleEnv}' — expected one of ${Object.keys(DB_POOL_PROFILES).join(', ')} (or unset for web)`
|
|
)
|
|
}
|
|
const role = (roleEnv as DbRole) || 'web'
|
|
const profile = DB_POOL_PROFILES[role]
|
|
|
|
const connectionString = resolveDbUrl('DATABASE_URL', role)
|
|
if (!connectionString) {
|
|
throw new Error('Missing DATABASE_URL environment variable')
|
|
}
|
|
|
|
const poolOptions = {
|
|
prepare: false,
|
|
idle_timeout: 20,
|
|
connect_timeout: 30,
|
|
onnotice: () => {},
|
|
connection: { application_name: process.env.DB_APP_NAME ?? profile.appName },
|
|
}
|
|
|
|
const postgresClient = instrumentPoolClient(
|
|
postgres(connectionString, { ...poolOptions, max: profile.primaryMax }),
|
|
'db'
|
|
)
|
|
|
|
export const db = drizzle(postgresClient, { schema })
|
|
|
|
/**
|
|
* Opt-in read-replica client for reads that tolerate bounded staleness and have
|
|
* no read-your-writes dependency (logs, exports, dashboard aggregations). Never
|
|
* for auth, workflow state, or billing enforcement. Falls back to the primary
|
|
* when `DATABASE_REPLICA_URL` is unset, so call sites never branch.
|
|
*/
|
|
const replicaUrl = resolveDbUrl('DATABASE_REPLICA_URL', role)
|
|
if (replicaUrl && !/^postgres(ql)?:\/\//.test(replicaUrl)) {
|
|
throw new Error(
|
|
'DATABASE_REPLICA_URL is set but is not a postgres:// DSN — fix the URL or unset the variable'
|
|
)
|
|
}
|
|
|
|
export const dbReplica: typeof db = replicaUrl
|
|
? drizzle(
|
|
instrumentPoolClient(
|
|
postgres(replicaUrl, { ...poolOptions, max: profile.replicaMax }),
|
|
'dbReplica'
|
|
),
|
|
{
|
|
schema,
|
|
}
|
|
)
|
|
: db
|