diff --git a/src/migrator.ts b/src/migrator.ts index 1a591eb119..37fdb88756 100644 --- a/src/migrator.ts +++ b/src/migrator.ts @@ -6,7 +6,7 @@ import type { IUnleashConfig } from './lib/types/option.js'; import { secondsToMilliseconds } from 'date-fns'; import path from 'path'; import { fileURLToPath } from 'node:url'; -import { getDBPasswordResolver } from './lib/db/aws-iam.js'; +import { getDBPassword } from './lib/db/aws-iam.js'; log.setLogLevel('error'); const __filename = fileURLToPath(import.meta.url); @@ -23,10 +23,11 @@ export async function migrateDb( { db }: Pick, stopAt?: string, ): Promise { + const password = await getDBPassword(db); return noDatabaseUrl(async () => { const custom = { ...db, - password: getDBPasswordResolver(db), + password, connectionTimeoutMillis: secondsToMilliseconds(10), }; @@ -45,10 +46,11 @@ export async function migrateDb( export async function requiresMigration({ db, }: Pick): Promise { + const password = await getDBPassword(db); return noDatabaseUrl(async () => { const custom = { ...db, - password: getDBPasswordResolver(db), + password, connectionTimeoutMillis: secondsToMilliseconds(10), }; @@ -67,10 +69,11 @@ export async function requiresMigration({ // This exists to ease testing export async function resetDb({ db }: IUnleashConfig): Promise { + const password = await getDBPassword(db); return noDatabaseUrl(async () => { const custom = { ...db, - password: getDBPasswordResolver(db), + password, connectionTimeoutMillis: secondsToMilliseconds(10), };