nestjs项目权限的代码实现

在 NestJS 中，可以使用中间件（Middleware）和守卫（Guard）来实现权限控制。以下是一个示例代码实现。

首先，定义一个守卫，用于验证用户是否有权限访问特定的路由。

import { CanActivate, ExecutionContext, Injectable } from '@nestjs/common';
import { Reflector } from '@nestjs/core';

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

  canActivate(context: ExecutionContext): boolean {
    const roles = this.reflector.get<string[]>('roles', context.getHandler());
    if (!roles) {
      return true;
    }
    const request = context.switchToHttp().getRequest();
    const user = request.user;
    return roles.includes(user.role);
  }
}

在守卫中使用了 `Reflector`，用于获取路由上标记的角色信息。在 `canActivate` 方法中，先判断路由是否标记了角色信息，如果没有，则默认允许访问。如果有，则从请求中获取用户信息，判断用户是否有对应的角色权限。

接下来，在控制器中使用 `@UseGuards` 标记需要进行权限验证的路由，并使用 `@Roles` 标记需要的角色信息。

import { Controller, Get, UseGuards } from '@nestjs/common';
import { RolesGuard } from './roles.guard';
import { Roles } from './roles.decorator';

@Controller('users')
export class UsersController {
  @Get()
  @Roles('admin')
  @UseGuards(RolesGuard)
  findAll() {
    return 'This action returns all users';
  }
}

在控制器中，使用 `@Roles` 标记需要的角色信息，将它与 `@UseGuards` 一起使用，表示需要使用 `RolesGuard` 守卫进行权限验证。

最后，定义一个自定义装饰器 `@Roles`，用于标记路由需要的角色信息。

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

export const Roles = (...roles: string[]) => SetMetadata('roles', roles);

在装饰器中，使用 `SetMetadata` 方法，将角色信息存储在路由元数据中。

以上就是 NestJS 中实现权限控制的代码示例。通过使用守卫和装饰器，可以轻松地实现权限控制，提高应用的安全性。