首頁 >web前端 >js教程 >如何在 NestJS 中建立和使用裝飾器和中間件

如何在 NestJS 中建立和使用裝飾器和中間件

Mary-Kate Olsen原創: 2025-01-13 06:05:43252瀏覽

裝飾器和中間件是 NestJS 的基礎——這些工具可以讓你的生活變得非常簡單，也可以讓你有點不知所措，這取決於你如何處理它們。

今天，讓我們逐步建立用於使用者驗證的自訂裝飾器和中間件，同時保持簡單明了。拿起你的咖啡，讓我們開始吧！

1. 介面

首先，讓我們為使用者物件定義一個介面。

這將確保類型安全並讓我們的 IDE 保持快樂（誰不喜歡快樂的 IDE？）。

export interface IUser {
  id: string;
  name: string;
  primaryEmail: string;
  phoneNumber: string | null;
  countryCode: string | null;
  dob: Date | null;
  createdAt: Date;
  updatedAt?: Date;
  deletedAt?: Date | null;
}

2. 建立自訂裝飾器

自訂裝飾器就像 NestJS 應用程式中的酷孩子。

在這裡，我們正在製作一個從請求物件中獲取用戶元資料的方法。

import { createParamDecorator, ExecutionContext } from '@nestjs/common';
import { IUser } from '../interface/user.interface';

export const UserMetadata = createParamDecorator(
  (_data: unknown, ctx: ExecutionContext) => {
    const request = ctx.switchToHttp().getRequest();
    return request.user as IUser;
  },
);

就是這樣！現在可以使用此裝飾器直接在控制器方法中提取使用者資訊。

How to Create and Use a Decorator and Middleware in NestJS

3. 建立Auth中間件

現在，讓我們建立一個 AuthGuard 來像虛擬保鑣一樣保護我們的端點。

import {
  CanActivate,
  ExecutionContext,
  ForbiddenException,
  Injectable
} from '@nestjs/common';
import { Reflector } from '@nestjs/core';
import { verify } from 'jsonwebtoken';
import { Observable } from 'rxjs';
import { IS_PUBLIC_KEY } from '../constant/core';
import { IUser } from '../interface/user.interface';

@Injectable()
export class AuthGuard implements CanActivate {
  constructor(
    private reflector: Reflector,
  ) { }

  canActivate(
    context: ExecutionContext,
  ): boolean | Promise<boolean> | Observable<boolean> {
    const isPublic = this.reflector.getAllAndOverride<boolean>(IS_PUBLIC_KEY, [
      context.getHandler(),
      context.getClass(),
    ]);
    if (isPublic) {
      return true;
    }

    const request = context.switchToHttp().getRequest();
    const headers = request.headers;

    const token = (headers['authorization'] || '').split(' ')[1];
    if (!token) {
      throw new ForbiddenException('Not Authenticated');
    }

    const jwtOpts = {
      expiresIn: '1h', // Replace with env vars in real use
      audience: 'your-audience',
      algorithm: 'HS256',
      issuer: 'your-issuer',
    };

    try {
      const decoded = verify(token, "my-jwt-secret-token", {
        audience: jwtOpts.audience,
        issuer: jwtOpts.issuer,
      }) as { user: IUser };

      request.user = decoded.user;
      return true;
    } catch (err) {
      throw new ForbiddenException('Session Expired or Invalid');
    }
  }
}

4. 新增公共和內部路由的元數據

有些路由應該是公共的（例如登入），有些可能是內部的。

讓我們為此添加兩個簡單的裝飾器。

import { SetMetadata } from '@nestjs/common';

export const IS_PUBLIC_KEY = 'isPublic';
export const IS_INTERNAL = 'isInternal';

export const Public = () => SetMetadata(IS_PUBLIC_KEY, true);
export const Internal = () => SetMetadata(IS_INTERNAL, true);

5. 在控制器中使用它們

最後，以下是您如何在控制器中使用所有這些內容。

import { Controller, Get, UseGuards } from '@nestjs/common';
import { UserMetadata } from '../decorators/user.decorator';
import { AuthGuard } from '../guards/auth.guard';
import { Public } from '../decorators/public.decorator';

@Controller('users')
export class UserController {

  @Public()
  @Get('login')
  login() {
    return { message: 'Login endpoint (public)' };
  }

  @UseGuards(AuthGuard)
  @Get('profile')
  getProfile(@UserMetadata() user: IUser) {
    return {
      message: 'User Profile',
      user,
    };
  }
}