NestJS 中间件的秘密大揭秘

NestJS 中间件：处理 HTTP 请求和响应的强大工具

什么是 NestJS 中间件？

NestJS 中间件是 NestJS 中的一种特殊函数，它在路由处理函数执行之前或之后执行，用于处理 HTTP 请求和响应。这些中间件就像拦截器，可以对请求进行预处理或对响应进行后处理。

中间件的应用场景

NestJS 中间件的应用场景广泛，包括：

• 身份验证： 在路由处理之前验证用户身份，确保访问权限。
• 日志记录： 记录 HTTP 请求和响应信息，方便故障排除和调试。
• 异常处理： 捕获并处理路由处理过程中的异常，保证应用程序稳定运行。
• 缓存： 从缓存中获取数据，提高应用程序性能。

中间件的运行原理

NestJS 中间件的运行原理如下：

1. HTTP 请求到达时，NestJS 根据路由配置找到对应的中间件并按序执行。
2. 中间件可以处理请求和响应，如验证参数或格式化数据。
3. 中间件执行成功，则继续执行路由处理函数。
4. 中间件执行失败，则终止路由处理并返回错误响应。

自定义 NestJS 中间件

NestJS 提供两种自定义中间件的方式：

• 使用 @Injectable() 装饰器： 创建一个中间件类，并实现 use() 方法。
• 使用 @Middleware() 装饰器： 直接创建一个中间件函数。

自定义中间件示例

以下是一个自定义 NestJS 中间件的示例：

// 使用 @Injectable() 装饰器
@Injectable()
export class AuthMiddleware implements NestMiddleware {
  use(req: Request, res: Response, next: Function) {
    // 在这里进行身份验证
    // ...
    
    // 验证成功，调用 next() 继续执行路由处理函数
    next();
  }
}

// 使用 @Middleware() 装饰器
@Middleware()
export function authMiddleware(req: Request, res: Response, next: Function) {
  // 在这里进行身份验证
  // ...

  // 验证成功，调用 next() 继续执行路由处理函数
  next();
}

在路由上使用中间件

在路由处理函数上使用 @UseMiddleware() 装饰器应用自定义中间件：

@Controller('users')
export class UsersController {
  @UseMiddleware(AuthMiddleware)
  @Get()
  getUsers() {
    // 在这里获取用户数据
    // ...
  }
}

总结

NestJS 中间件是处理 HTTP 请求和响应的强大工具，可以实现广泛的功能。通过自定义中间件，你可以轻松地为你的 NestJS 应用程序添加各种功能，如身份验证、日志记录、异常处理和缓存等。

常见问题解答

1. 中间件是否可以在所有路由上使用？
   是的，你可以将中间件应用于单个路由或一组路由。

2. 中间件可以访问路由处理函数的请求和响应吗？
   是的，中间件可以访问 HTTP 请求和响应对象。

3. 中间件可以在路由处理函数之前和之后执行吗？
   是的，你可以控制中间件执行的顺序。

4. NestJS 中有哪些内置的中间件？
   NestJS 提供了许多内置中间件，如 bodyParser() 和 cookieParser()。

5. 如何使用自定义中间件代替内置中间件？
   你可以通过将自定义中间件放在内置中间件之前来覆盖它们。