feat: disable migrations flag

Author: luoling8192

apps/server/src/app.ts

@@ -78,7 +78,10 @@ async function bootstrap() {
   const config = await initConfig(flags)
 
   try {
-    await initDrizzle(logger, config, { isDatabaseDebugMode: flags.isDatabaseDebugMode })
+    await initDrizzle(logger, config, {
+      isDatabaseDebugMode: flags.isDatabaseDebugMode,
+      disableMigrations: flags.disableMigrations,
+     })
     logger.log('Database initialized successfully')
   }
   catch (error) {

packages/common/src/flags.ts

@@ -10,6 +10,7 @@ export interface RuntimeFlags {
 
   isDebugMode: boolean
   isDatabaseDebugMode: boolean
+  disableMigrations: boolean
 
   dbProvider?: DatabaseType
   dbUrl?: string
@@ -38,6 +39,7 @@ export interface RuntimeFlags {
 const DEFAULT_FLAGS: RuntimeFlags = {
   logLevel: LoggerLevel.Verbose,
   logFormat: LoggerFormat.Pretty,
+  disableMigrations: false,
 
   isDebugMode: false,
   isDatabaseDebugMode: false,
@@ -103,6 +105,7 @@ export function parseEnvFlags(env: Record<string, string | undefined>): RuntimeF
     }
   }
 
+  result.disableMigrations = readBooleanEnv('DISABLE_MIGRATIONS', env)
   result.isDatabaseDebugMode = readBooleanEnv('DATABASE_DEBUG', env)
 
   assignIfPresent(result, 'dbProvider', readEnvValue('DATABASE_TYPE', env))

packages/core/src/db/index.ts

@@ -32,6 +32,7 @@ export async function initDrizzle(
     dbPath?: string
     debuggerWebSocketUrl?: string
     isDatabaseDebugMode?: boolean
+    disableMigrations?: boolean
   },
 ) {
   logger.log('Initializing database...')
@@ -49,6 +50,7 @@ export async function initDrizzle(
       const { initPgDrizzle } = await import('./pg')
       dbInstance = await initPgDrizzle(logger, config, {
         isDatabaseDebugMode: options?.isDatabaseDebugMode,
+        disableMigrations: options?.disableMigrations,
       })
       break
     }
@@ -58,12 +60,14 @@ export async function initDrizzle(
         const { initPgliteDrizzleInBrowser } = await import('./pglite.browser')
         dbInstance = await initPgliteDrizzleInBrowser(logger, {
           isDatabaseDebugMode: options?.isDatabaseDebugMode,
+          disableMigrations: options?.disableMigrations,
         })
       }
       else {
         const { initPgliteDrizzleInNode } = await import('./pglite')
         dbInstance = await initPgliteDrizzleInNode(logger, config, options?.dbPath, {
           isDatabaseDebugMode: options?.isDatabaseDebugMode,
+          disableMigrations: options?.disableMigrations,
         })
       }
       break

packages/core/src/db/pg.ts

@@ -27,6 +27,7 @@ export async function initPgDrizzle(
   config: Config,
   options: {
     isDatabaseDebugMode?: boolean
+    disableMigrations?: boolean
   } = {},
 ) {
   logger.log('Initializing postgres drizzle...')
@@ -50,7 +51,9 @@ export async function initPgDrizzle(
     logger.log('Database connection established successfully')
 
     // Migrate database
-    await applyMigrations(logger, db)
+    if (!options.disableMigrations) {
+      await applyMigrations(logger, db)
+    }
   }
   catch (error) {
     logger.withError(error).error('Failed to connect to database')

packages/core/src/db/pglite.browser.ts

@@ -31,6 +31,7 @@ export async function initPgliteDrizzleInBrowser(
   options: {
     debuggerWebSocketUrl?: string
     isDatabaseDebugMode?: boolean
+    disableMigrations?: boolean
   } = {},
 ) {
   logger.log('Initializing pglite drizzle in browser...')
@@ -73,7 +74,9 @@ export async function initPgliteDrizzleInBrowser(
       logger.log('Vector extension enabled successfully')
 
       // Migrate database
-      await applyMigrations(logger, db)
+      if (!options.disableMigrations) {
+        await applyMigrations(logger, db)
+      }
     }
     catch (error) {
       logger.withError(error).error('Failed to connect to database')

packages/core/src/db/pglite.ts

@@ -32,6 +32,7 @@ export async function initPgliteDrizzleInNode(
   dbPath?: string,
   options: {
     isDatabaseDebugMode?: boolean
+    disableMigrations?: boolean
   } = {},
 ) {
   logger.log('Initializing pglite drizzle...')
@@ -58,7 +59,9 @@ export async function initPgliteDrizzleInNode(
       logger.log('Vector extension enabled successfully')
 
       // Migrate database
-      await applyMigrations(logger, db)
+      if (!options.disableMigrations) {
+        await applyMigrations(logger, db)
+      }
     }
     catch (error) {
       logger.withError(error).error('Failed to connect to database')