从入门到精通！Nest.js中rbac权限控制的全面指南

2023-04-17 23:43:44

在 Nest.js 中使用 RBAC 实现权限控制：全面指南

RBAC 权限控制简介

在构建应用程序时，权限控制至关重要。RBAC（基于角色的访问控制）是一种广泛使用的权限控制模型，它通过将用户分配给不同的角色并为每个角色分配权限来实现。通过这种方式，我们可以轻松控制用户对应用程序不同功能的访问。

在 Nest.js 中使用 RBAC

安装依赖项

首先，安装 nestjs-rbac 模块：

npm install nestjs-rbac

配置模块

在 app.module.ts 中配置模块：

import { Module } from '@nestjs/common';
import { RbacModule } from 'nestjs-rbac';

@Module({
  imports: [RbacModule.forRoot()],
})
export class AppModule {}

定义角色和权限

在 rbac.service.ts 中定义角色和权限：

import { Injectable } from '@nestjs/common';
import { RbacService } from 'nestjs-rbac';

@Injectable()
export class RbacService {
  constructor(private readonly rbacService: RbacService) {}

  async createRoles() {
    await this.rbacService.createRole({ name: 'admin' });
    await this.rbacService.createRole({ name: 'user' });
  }

  async createPermissions() {
    await this.rbacService.createPermission({ name: 'create_user' });
    await this.rbacService.createPermission({ name: 'update_user' });
    await this.rbacService.createPermission({ name: 'delete_user' });
  }
}

关联角色和权限

关联角色和权限：

import { Injectable } from '@nestjs/common';
import { RbacService } from 'nestjs-rbac';

@Injectable()
export class RbacService {
  // ...

  async assignPermissionsToRoles() {
    await this.rbacService.assignPermissionsToRole({
      roleName: 'admin',
      permissionNames: ['create_user', 'update_user', 'delete_user'],
    });

    await this.rbacService.assignPermissionsToRole({
      roleName: 'user',
      permissionNames: ['create_user'],
    });
  }
}

使用 RBAC 进行授权

在控制器中使用 RBAC：

import { Controller, Get, UseGuards } from '@nestjs/common';
import { AppService } from './app.service';
import { RbacGuard } from 'nestjs-rbac';

@Controller()
export class AppController {
  constructor(private readonly appService: AppService) {}

  @Get()
  @UseGuards(RbacGuard)
  getHello(): string {
    return this.appService.getHello();
  }
}