NestJS拦截器：实用案例揭秘

2023-09-12 07:06:56

面向切面编程（AOP）是NestJS框架中一个强大的工具，允许我们以非侵入方式增强和修改应用程序的行为。拦截器是AOP的一种具体实现，它允许我们在特定事件发生时拦截和处理请求和响应。

在本文中，我们将深入了解NestJS拦截器，并通过一些实际示例来说明它们的强大功能。

日志记录

拦截器的一个常见用例是日志记录。我们可以使用拦截器在请求和响应期间记录重要信息，帮助我们调试和分析应用程序。以下是一个简单的日志拦截器的示例：

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 method = request.method;
    const url = request.url;
    const startTime = Date.now();

    return next.handle().pipe(
      tap(() => {
        const elapsedTime = Date.now() - startTime;
        console.log(`${method} ${url} ${elapsedTime}ms`);
      }),
    );
  }
}

异常处理

拦截器还可以用于处理应用程序中的异常。我们可以使用拦截器在发生异常时捕获和处理它，并返回自定义响应或将错误信息记录到数据库。以下是一个简单的异常处理拦截器的示例：

import { Injectable, NestInterceptor, ExecutionContext, CallHandler } from '@nestjs/common';
import { Observable, throwError } from 'rxjs';
import { catchError } from 'rxjs/operators';

@Injectable()
export class ExceptionInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    return next.handle().pipe(
      catchError((err) => {
        console.error(err);
        return throwError(err);
      }),
    );
  }
}

身份验证和授权

拦截器还可以用于实现身份验证和授权逻辑。我们可以使用拦截器检查请求中是否存在有效的令牌，并根据用户的权限授予或拒绝对特定端点的访问。以下是一个简单的身份验证拦截器的示例：

import { Injectable, NestInterceptor, ExecutionContext, CallHandler } from '@nestjs/common';
import { Observable } from 'rxjs';
import { UnauthorizedException } from '@nestjs/common';

@Injectable()
export class AuthInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    const request = context.switchToHttp().getRequest();
    const token = request.headers['authorization'];

    if (!token) {
      throw new UnauthorizedException();
    }

    // 验证令牌的逻辑

    return next.handle();
  }
}