48 lines
2.0 KiB
TypeScript
48 lines
2.0 KiB
TypeScript
import { pool } from "./index.js"
|
|
import { applySqlMigrations } from "./migrate.js"
|
|
import { dropExpiredPartitions, ensurePartitionsAround } from "./partitions.js"
|
|
import { importSqliteToPostgres, shouldImportSqlite, sqliteFileLooksPresent } from "./sqlite-import.js"
|
|
import { env } from "../config.js"
|
|
import { mcpStateDirectory } from "../modules/mcp/security.js"
|
|
import { assertDatabaseMaintenanceStartup } from "../services/database-maintenance-startup.js"
|
|
import { reconcileDatabaseRestoreAudits } from "../services/database-restore-audit.js"
|
|
|
|
const ETL_LOCK = 8723101
|
|
|
|
export async function initDatabase(): Promise<void> {
|
|
await assertDatabaseMaintenanceStartup(mcpStateDirectory(), env.DATABASE_URL)
|
|
await pool.query("SELECT 1")
|
|
await applySqlMigrations(pool)
|
|
await ensurePartitionsAround(pool)
|
|
const client = await pool.connect()
|
|
try {
|
|
await client.query("SELECT pg_advisory_lock($1)", [ETL_LOCK])
|
|
if (await shouldImportSqlite(pool, env.DATABASE_PATH)) {
|
|
console.log(`SQLite → PostgreSQL: импорт ${env.DATABASE_PATH}`)
|
|
const report = await importSqliteToPostgres(pool, env.DATABASE_PATH, { strict: true })
|
|
console.log(
|
|
`SQLite → PostgreSQL: готово за ${report.durationMs}ms, таблиц ${Object.keys(report.tables).length}`,
|
|
)
|
|
} else {
|
|
const marker = await pool.query<{ sqlite_imported_at: string | null }>(
|
|
`SELECT sqlite_imported_at FROM data_migration WHERE id = 1`,
|
|
)
|
|
if (!marker.rows[0]?.sqlite_imported_at && !sqliteFileLooksPresent(env.DATABASE_PATH)) {
|
|
console.warn(
|
|
`SQLite → PostgreSQL: файл ${env.DATABASE_PATH} не найден, база после wipe остаётся пустой (defaults settings)`,
|
|
)
|
|
}
|
|
}
|
|
} finally {
|
|
try {
|
|
await client.query("SELECT pg_advisory_unlock($1)", [ETL_LOCK])
|
|
} catch {
|
|
/* ignore */
|
|
}
|
|
client.release()
|
|
}
|
|
await dropExpiredPartitions(pool)
|
|
await ensurePartitionsAround(pool)
|
|
await reconcileDatabaseRestoreAudits(pool, mcpStateDirectory())
|
|
}
|