如何在不重启NestJS服务的情况下运行时修改模块配置值

直接使用MongooseModule.forRoot()的静态配置方式无法在运行时修改连接地址，因为模块初始化阶段就已经固定了连接参数。要实现运行时动态切换，需要手动管理Mongoose连接实例，具体步骤如下：

1. 实现动态连接管理服务 #

创建一个自定义服务来处理Mongoose的连接、断开和重连逻辑：

import { Injectable, OnModuleInit } from '@nestjs/common';
import * as mongoose from 'mongoose';

@Injectable()
export class MongooseConnectionService implements OnModuleInit {
  private connection: mongoose.Connection;
  private currentUri: string = 'mongodb://localhost/nest'; // 默认连接地址

  async onModuleInit() {
    // 模块初始化时建立初始连接
    await this.connect();
  }

  async connect(uri?: string) {
    const targetUri = uri || this.currentUri;
    
    // 若已有连接，先断开旧连接
    if (this.connection) {
      await this.connection.close();
    }

    // 建立新连接并保存实例
    const mongooseInstance = await mongoose.connect(targetUri);
    this.connection = mongooseInstance.connection;
    this.currentUri = targetUri;
    console.log(`MongoDB连接已切换至: ${targetUri}`);
  }

  // 提供连接实例供其他模块使用
  getConnection(): mongoose.Connection {
    return this.connection;
  }
}

2. 重构AppModule，移除静态配置 #

替换原有的MongooseModule.forRoot()，改用自定义服务管理连接：

import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import { MongooseConnectionService } from './mongoose-connection.service';
import { ConfigController } from './config.controller';

@Module({
  providers: [MongooseConnectionService],
  exports: [MongooseConnectionService],
  controllers: [ConfigController],
  // 若有使用Mongoose的Feature模块，需通过连接服务指定连接实例
  // 示例: MongooseModule.forFeature([{ name: 'User', schema: UserSchema }], MongooseConnectionService)
})
export class AppModule {}

3. 创建API接口触发配置修改 #

新增控制器提供接口，调用连接服务完成运行时切换：

import { Controller, Post, Body, HttpException, HttpStatus } from '@nestjs/common';
import { MongooseConnectionService } from './mongoose-connection.service';

@Controller('api')
export class ConfigController {
  constructor(private readonly mongooseConnService: MongooseConnectionService) {}

  @Post('changeMongoURL')
  async changeMongoURL(@Body() body: { uri: string }) {
    if (!body.uri) {
      throw new HttpException('MongoDB地址不能为空', HttpStatus.BAD_REQUEST);
    }
    
    try {
      await this.mongooseConnService.connect(body.uri);
      return { success: true, message: `已成功切换至新MongoDB地址: ${body.uri}` };
    } catch (error) {
      throw new HttpException(`连接失败: ${error.message}`, HttpStatus.INTERNAL_SERVER_ERROR);
    }
  }
}

关键注意点 #

• 切换连接时会中断当前未完成的数据库操作，建议在业务低峰期执行，或针对关键操作添加事务保障。
• 需做好错误处理，避免因无效地址导致服务失去数据库连接。
• 若需管理多数据库连接，可扩展服务维护连接实例映射表。

内容的提问来源于stack exchange，提问作者Saif Islam