NestJS拆分配置Throttler Guard后GraphQL模块仍异常的问题求助

问题场景 #

我有一个包含REST模块和GraphQL模块的Nest.js应用，二者均导入至App.module.ts中。为保护整个应用，我使用了Nest的Throttler Guard。由于已知GraphQL无法兼容常规ThrottlerGuard，我创建了GqlThrottlerGuard并在GraphQL模块中配置，同时在REST模块中配置原生ThrottlerGuard。

GraphQL模块代码 #

@Module({
  imports: [
    GraphQLModule.forRoot<ApolloDriverConfig>({
      driver: ApolloDriver,
      autoSchemaFile: true
    }),
    ThrottlerModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (config: ConfigService) => ({
        ttl: config.get('security.throttle.ttl'),
        limit: config.get('security.throttle.limit'),
      }),
    }),
  ],
  providers: [
    {
      provide: APP_GUARD,
      useClass: GqlThrottlerGuard,
    },
  ],
})
export class GraphModule { }

REST模块代码 #

@Module({
  imports: [
    ThrottlerModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (config: ConfigService) => ({
        ttl: config.get('security.throttle.ttl'),
        limit: config.get('security.throttle.limit'),
      }),
    }),
  ],
  controllers: [RestController],
  providers: [
    {
      provide: APP_GUARD,
      useClass: ThrottlerGuard,
    },
  ],
})
export class RestModule implements NestModule {}

AppModule代码 #

@Module({
  imports: [
    RestModule,
    GraphModule,
  ],
})
export class AppModule { }

尽管已拆分配置，GraphModule仍出现GitHub issue #574中的错误，请问这是否符合预期？该如何解决？

问题分析 #

这种情况不符合预期，问题根源在于两个核心特性冲突：

1. ThrottlerModule的全局特性：ThrottlerModule.forRootAsync是全局模块，在多个子模块重复导入会触发重复初始化，导致配置紊乱或依赖冲突。
2. APP_GUARD的全局作用域：APP_GUARD是全局守卫令牌，两个模块都注册各自守卫时，后加载的守卫会覆盖先加载的，导致GraphQL请求被错误应用REST的ThrottlerGuard，触发兼容性问题。

解决方案 #

方案1：统一全局配置+动态守卫工厂 #

将Throttler的全局配置移至AppModule，同时创建工厂守卫根据请求类型自动切换限流逻辑。

步骤1：重构AppModule

@Module({
  imports: [
    ConfigModule.forRoot({ isGlobal: true }), // 确保ConfigModule全局可用
    ThrottlerModule.forRootAsync({
      inject: [ConfigService],
      useFactory: (config: ConfigService) => ({
        ttl: config.get('security.throttle.ttl'),
        limit: config.get('security.throttle.limit'),
      }),
    }),
    GraphQLModule.forRoot<ApolloDriverConfig>({
      driver: ApolloDriver,
      autoSchemaFile: true
    }),
    RestModule,
    GraphModule,
  ],
  providers: [
    ThrottlerGuard,
    GqlThrottlerGuard,
    {
      provide: APP_GUARD,
      useClass: ThrottlerGuardFactory,
    },
  ],
})
export class AppModule { }

步骤2：创建守卫工厂

import { Injectable, CanActivate, ExecutionContext, ContextType } from '@nestjs/common';
import { ThrottlerGuard } from '@nestjs/throttler';
import { GqlThrottlerGuard } from './gql-throttler.guard'; // 导入你的自定义GraphQL守卫

@Injectable()
export class ThrottlerGuardFactory implements CanActivate {
  constructor(
    private readonly restThrottler: ThrottlerGuard,
    private readonly gqlThrottler: GqlThrottlerGuard,
  ) {}

  canActivate(context: ExecutionContext): boolean | Promise<boolean> {
    // 判断当前请求类型
    const isGraphQL = context.getType<ContextType | 'graphql'>() === 'graphql';
    return isGraphQL 
      ? this.gqlThrottler.canActivate(context) 
      : this.restThrottler.canActivate(context);
  }
}

步骤3：简化子模块

移除RestModule和GraphModule中的ThrottlerModule导入以及APP_GUARD注册：

RestModule

@Module({
  controllers: [RestController],
})
export class RestModule implements NestModule {}

GraphModule

@Module({
  // 若GraphQL配置无需模块级别隔离，可移至AppModule，此处留空或保留其他模块专属providers
})
export class GraphModule { }

方案2：模块级别手动绑定守卫 #

如果不需要全局限流，可在REST控制器和GraphQL Resolver上分别手动绑定对应守卫：

• REST控制器：

@Controller('rest')
@UseGuards(ThrottlerGuard)
export class RestController {
  // ...
}

• GraphQL Resolver：

@Resolver()
@UseGuards(GqlThrottlerGuard)
export class DemoResolver {
  // ...
}

此方案无需注册APP_GUARD，只需在AppModule全局导入一次ThrottlerModule即可。

内容的提问来源于stack exchange，提问作者G. Hruschka