Initial commit

2026-06-24 23:47:55 -07:00
commit d134b480a0
297 changed files with 30726 additions and 0 deletions
@@ -0,0 +1,103 @@
+import knex, { type Knex } from "knex"
+
+import {
+  DB_HEALTH_CHECK_INTERVAL_SECONDS,
+  DB_HEALTH_CHECK_RETRIES,
+  DB_HEALTH_CHECK_START_PERIOD_SECONDS,
+  DB_HEALTH_CHECK_TIMEOUT_SECONDS,
+} from "@/config"
+import logger from "@/utils/logger"
+import sleep from "@/utils/sleep"
+import {
+  isCredentialFailure,
+  isNetworkFailure,
+  isSocketFailure,
+  isMissingDatabaseFailure,
+} from "@/utils/db-error-helpers"
+import { buildKnexConfig } from "@/db/db-migration-client"
+
+function checkHealth(dbMigrationClient: Knex, timeoutSeconds: number) {
+  return new Promise((resolve, reject) => {
+    const timer = setTimeout(() => reject(new Error("Connection timeout")), timeoutSeconds * 1000)
+    dbMigrationClient
+      .raw("SELECT 1")
+      .then(() => {
+        clearTimeout(timer)
+        resolve(null)
+      })
+      .catch(reject)
+  })
+}
+
+export async function waitForDatabase({
+  intervalSeconds = DB_HEALTH_CHECK_INTERVAL_SECONDS,
+  timeoutSeconds = DB_HEALTH_CHECK_TIMEOUT_SECONDS,
+  retries = DB_HEALTH_CHECK_RETRIES,
+  startPeriodSeconds = DB_HEALTH_CHECK_START_PERIOD_SECONDS,
+}: {
+  intervalSeconds?: number
+  timeoutSeconds?: number
+  retries?: number
+  startPeriodSeconds?: number
+} = {}): Promise<void> {
+  await sleep(startPeriodSeconds)
+
+  logger.info("Attempting direct to database connection...")
+  const databaseConfig = buildKnexConfig()
+
+  let dbMigrationClient = knex(databaseConfig)
+  let isDatabaseSocketReady = false
+
+  for (let i = 0; i < retries; i++) {
+    try {
+      await checkHealth(dbMigrationClient, timeoutSeconds)
+      logger.info("Database connection successful.")
+      return
+    } catch (error) {
+      if (isSocketFailure(error)) {
+        logger.info(`Database socket is not ready, retrying... ${error}`, { error })
+        await sleep(intervalSeconds)
+      } else if (isNetworkFailure(error)) {
+        logger.info(`Network error, retrying... ${error}`, { error })
+        await sleep(intervalSeconds)
+      } else if (isCredentialFailure(error)) {
+        if (isDatabaseSocketReady) {
+          logger.error(`Database connection failed due to invalid credentials: ${error}`, { error })
+          throw error
+        } else {
+          logger.info(
+            "Falling back to database server-level connection (database might not exist)..."
+          )
+          const serverLevelConfig = buildKnexConfig({ connection: { database: "" } })
+          dbMigrationClient = knex(serverLevelConfig)
+          i -= 1
+          isDatabaseSocketReady = true
+          continue
+        }
+      } else if (isMissingDatabaseFailure(error)) {
+        if (isDatabaseSocketReady) {
+          logger.error(`Database connection failed because database does not exist): ${error}`, {
+            error,
+          })
+          throw error
+        } else {
+          logger.info(
+            "Falling back to default postgres connection (after database does not exist failure)..."
+          )
+          const serverLevelConfig = buildKnexConfig({ connection: { database: "postgres" } })
+          dbMigrationClient = knex(serverLevelConfig)
+          i -= 1
+          isDatabaseSocketReady = true
+          continue
+        }
+      } else {
+        logger.error(`Unknown database connection error: ${error}`, { error })
+        //throw error
+      }
+    }
+  }
+
+  throw new Error(`Failed to connect to the database due to timeout.`)
+}
+
+export default waitForDatabase