开发者的福音：NestJS 拦截器

前端

2023-10-31 20:59:56

NestJS 拦截器：为你的应用程序增添超能力

简介

在 NestJS 的世界里，拦截器是一个强大的工具，可以让你在请求和响应之间发挥你的魔法。它们就像应用程序的守门员，让你在它们进出之前进行各种操作。从记录请求到验证数据，拦截器可以增强你的应用程序，而无需修改其核心代码。

拦截器的工作原理

拦截器是通过装饰器实现的，就像你的应用程序的时尚配件一样。你可以给类或方法穿上这些装饰器，NestJS 就会在它们执行前后的某个时刻向它们打个招呼。

拦截器有两种类型：

全局拦截器： 这些拦截器适用于应用程序的每个请求和响应，就像无处不在的保镖。
局部拦截器： 这些拦截器只适用于特定的请求和响应，就像针对特定任务的专家。

使用拦截器

要使用拦截器，你需要打造一个拦截器类并将其放入你的应用程序中。通过 @Injectable() 装饰器，你可以赋予你的拦截器生命，然后使用 @UseInterceptors() 装饰器，将它们添加到你的控制器或方法上。

例如，我们可以创建一个拦截器来记录每次请求和响应，就像一个小小的日记本：

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 request = context.switchToHttp().getRequest();
    const response = context.switchToHttp().getResponse();

    console.log(`Request: ${request.method} ${request.url}`);
    console.log(`Response: ${response.statusCode} ${response.statusMessage}`);

    return next.handle().pipe(
      tap(() => {
        console.log(`Request: ${request.method} ${request.url}`);
        console.log(`Response: ${response.statusCode} ${response.statusMessage}`);
      }),
    );
  }
}

然后，你可以给你的控制器或方法披上 LoggingInterceptor 这件时髦的外衣，让它在每次调用时记录请求和响应：

import { Controller, Get, UseInterceptors } from '@nestjs/common';
import { LoggingInterceptor } from './logging.interceptor';

@Controller()
export class AppController {
  @Get()
  @UseInterceptors(LoggingInterceptor)
  getHello() {
    return 'Hello, world!';
  }
}