深入探讨 NestJS 中间件的高级用法：从全局请求处理到 AOP 编程

NestJS 中间件的高级用法：从全局请求处理到 AOP 编程

NestJS 作为一个强大的 Node.js 框架，以其模块化、可扩展性和面向切面编程（AOP）的支持而闻名。在实际开发中，NestJS 中间件是一个核心功能，它允许开发者在请求和响应之间插入逻辑，从而实现诸如日志记录、身份验证、错误处理等功能。本文将深入探讨 NestJS 中间件的高级用法，包括全局请求处理、自定义请求/响应修改以及 AOP 编程的实际应用场景。

1. 全局请求处理

在 NestJS 中，全局中间件可以应用于所有传入的请求。这在处理如跨域资源共享（CORS）、请求日志记录或全局错误捕获时非常有用。以下是一个简单的全局中间件实现示例：

import { Injectable, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express';
 
@Injectable()
export class GlobalMiddleware implements NestMiddleware {
  use(req: Request, res: Response, next: NextFunction) {
    console.log('Global Middleware: Request received');
    next();
  }
}

在 main.ts 中，可以通过 app.use() 方法将其注册为全局中间件：

import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module';
import { GlobalMiddleware } from './middleware/global.middleware';
 
async function bootstrap() {
  const app = await NestFactory.create(AppModule);
  app.use(new GlobalMiddleware().use);
  await app.listen(3000);
}
bootstrap();

全局中间件的主要优点是其简单性和一致性，但它也带来了潜在的性能开销，因此在使用时需谨慎评估。

2. 自定义请求/响应修改

NestJS 中间件不仅可以用于全局请求处理，还可以针对特定路由或控制器进行自定义请求/响应修改。例如，你可以在特定路由中修改请求头或响应体。以下是一个自定义中间件的示例：

import { Injectable, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express';
 
@Injectable()
export class CustomHeaderMiddleware implements NestMiddleware {
  use(req: Request, res: Response, next: NextFunction) {
    req.headers['x-custom-header'] = 'Custom Value';
    next();
  }
}

在控制器中使用该中间件：

import { Controller, Get, UseMiddleware } from '@nestjs/common';
import { CustomHeaderMiddleware } from './middleware/custom-header.middleware';
 
@Controller('test')
export class TestController {
  @Get()
  @UseMiddleware(CustomHeaderMiddleware)
  getTest() {
    return 'Test with custom header';
  }
}

通过这种方式，你可以灵活地为特定路由或控制器添加自定义逻辑，而不会影响其他部分。

3. AOP 编程在 NestJS 中的应用

AOP（面向切面编程）是一种编程范式，旨在将横切关注点（如日志记录、事务管理等）从业务逻辑中分离出来。NestJS 通过装饰器和拦截器提供了强大的 AOP 支持。以下是一个简单的拦截器示例，用于记录方法执行时间：

import { Injectable, NestInterceptor, ExecutionContext, CallHandler } from '@nestjs/common';
import { Observable } from 'rxjs';
import { tap } from 'rxjs/operators';
 
@Injectable()
export class LoggingInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    const now = Date.now();
    return next
      .handle()
      .pipe(
        tap(() => console.log(`Method executed in ${Date.now() - now}ms`)),
      );
  }
}

在控制器中使用该拦截器：

import { Controller, Get, UseInterceptors } from '@nestjs/common';
import { LoggingInterceptor } from './interceptors/logging.interceptor';
 
@Controller('test')
@UseInterceptors(LoggingInterceptor)
export class TestController {
  @Get()
  getTest() {
    return 'Test with logging';
  }
}

通过 AOP 编程，你可以将非核心业务逻辑（如日志记录、性能监控等）从业务代码中分离出来，使代码更加清晰和可维护。

4. 应用价值与潜在风险

NestJS 中间件和 AOP 编程为开发者提供了强大的工具，但在实际应用中需注意以下两点：

• 性能开销：中间件和拦截器会在每个请求或方法调用时执行，因此可能带来额外的性能开销。在复杂系统中，建议对中间件和拦截器的使用进行性能优化。
• 过度使用：虽然中间件和 AOP 功能强大，但过度使用可能导致代码逻辑复杂化。建议在必要场景下使用，并确保其与业务逻辑的清晰分离。

总之，NestJS 中间件和 AOP 编程是提升代码质量和可维护性的有效工具，但需结合实际场景灵活使用。