如何在NestJS后端中条件性使用app.use()配置中间件？

我正尝试为NestJS应用添加helmet中间件，同时需要集成graphqlUploadExpress。请问如何通过usesUpload条件，正确实现仅使用helmet或同时使用helmet+graphqlUploadExpress的配置？以下是我的代码示例：

import { NestFactory } from '@nestjs/core'
import graphqlUploadExpress from 'graphql-upload/graphqlUploadExpress.js'
import helmet from 'helmet'

const upload = graphqlUploadExpress()

export const main = async (
  AppModule: unknown,
  usesUpload = false
) => {
  const app = await NestFactory.create(AppModule, options)
  app.use(usesUpload ? helm : helm, upload) // <-- I think, this is not correct
  await app.listen(3000)
}

正确实现方式 #

你的代码存在两处问题：一是拼写错误（helm应为helmet），二是条件判断逻辑失效，当前写法无论usesUpload是否为true，都会同时加载两个中间件，不符合需求。

下面提供两种简洁的实现方案：

方案一：分支判断注册中间件 #

先固定注册Helmet，再根据usesUpload的值决定是否添加上传中间件：

import { NestFactory } from '@nestjs/core'
import graphqlUploadExpress from 'graphql-upload/graphqlUploadExpress.js'
import helmet from 'helmet'

const upload = graphqlUploadExpress()

export const main = async (
  AppModule: unknown,
  usesUpload = false
) => {
  const app = await NestFactory.create(AppModule); // 若有自定义配置可补充options参数
  // 必加Helmet中间件
  app.use(helmet());
  // 按需添加上传中间件
  if (usesUpload) {
    app.use(upload);
  }
  await app.listen(3000)
}

方案二：动态构建中间件数组 #

把需要加载的中间件存入数组，再批量传递给app.use：

import { NestFactory } from '@nestjs/core'
import graphqlUploadExpress from 'graphql-upload/graphqlUploadExpress.js'
import helmet from 'helmet'

const upload = graphqlUploadExpress()

export const main = async (
  AppModule: unknown,
  usesUpload = false
) => {
  const app = await NestFactory.create(AppModule);
  // 初始化基础中间件数组
  const middlewares = [helmet()];
  // 按需追加上传中间件
  usesUpload && middlewares.push(upload);
  // 批量注册
  app.use(...middlewares);
  await app.listen(3000)
}

注意：

• 确保helmet()是调用后的实例（NestJS也支持直接传函数，但调用后逻辑更明确）
• 原代码中的options变量未定义，需根据实际需求补充配置或移除该参数

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