你能在条件下导入一个 NestJS 模块吗

Question

我正在 NestJS 中创建一个微服务。现在我想使用 RabbitMQ 向另一个服务发送消息。

我的问题是：是否可以基于.env 变量导入 RabbitmqModule？如： USE_BROKER=false。如果这个变量是假的，比不导入模块？

RabbitMQ 是在下面的 GraphQLModule 中导入的。

@Module({
  imports: [
    GraphQLFederationModule.forRoot({
      autoSchemaFile: true,
      context: ({ req }) => ({ req }),
    }),
    DatabaseModule,
    AuthModule,
    RabbitmqModule,
  ],
  providers: [UserResolver, FamilyResolver, AuthResolver],
})
export class GraphQLModule {}

RabbitmqModule：

import { Global, Module } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';
import { RabbitMQModule } from '@golevelup/nestjs-rabbitmq';
import { UserProducer } from './producers/user.producer';

@Global()
@Module({
  imports: [
    RabbitMQModule.forRootAsync(RabbitMQModule, {
      useFactory: async (config: ConfigService) => ({
        exchanges: [
          {
            name: config.get('rabbitMQ.exchange'),
            type: config.get('rabbitMQ.exchangeType'),
          },
        ],
        uri: config.get('rabbitMQ.url'),
        connectionInitOptions: { wait: false },
      }),
      inject: [ConfigService],
    }),
  ],
  providers: [UserProducer],
  exports: [UserProducer],
})
export class RabbitmqModule {}

Answer 1

我认为推荐的方法是使用 NestJS 的 DynamicModule 功能。

这里解释：https://docs.nestjs.com/fundamentals/dynamic-modules

只需在 register 函数中检查您的环境变量并返回您的 Module 对象。 比如：

@Module({})
export class GraphQLModule {
  static register(): DynamicModule {
    const imports = [
      GraphQLFederationModule.forRoot({
        autoSchemaFile: true,
        context: ({ req }) => ({ req }),
      }),
      DatabaseModule,
      AuthModule]
    if (process.env.USE_BROKER) {
      modules.push(RabbitmqModule)
    }
    return {
      imports,
      providers: [UserResolver, FamilyResolver, AuthResolver],
    };
  }
}

Answer 2

嗯，我在一个小巢项目中尝试了一个简单的解决方法，它工作得很好。看看吧：

const mymodules = [TypeOrmModule.forRoot(typeOrmConfig), UsersModule];
if (config.get('importModule')) {
    mymodules.push(PoopModule);
}
@Module({
    imports: mymodules,
    controllers: [AppController],
    providers: [AppService],
})
export class AppModule {}

我在我的 env/config 中创建了一个“importModule”，并用 true 和 false 对其进行了测试。如果为真，我的 Poop 模块会被部署，否则不会部署，只会部署其他模块。

你能在你的项目中尝试同样的方法吗？