使用Nest.js设计RBAC权限系统:分步指南

前言

对于后台管理系统而言,「访问控制」和「个性化界面」是必备功能。

举例:超级管理员能查看所有页面;普通用户只能访问 A 和 B 页面;VIP 用户则能查看 A、B、C、D 四个页面。

这些功能背后依赖的是三大核心概念:

  • 用户(User) :最小单位,如 Alice、Bob、Charlie。
  • 角色(Role) :一个用户可以拥有多个角色。例如 Alice 既是普通用户又是 VIP。
  • 权限(Permission) :角色会关联若干权限。VIP 角色可能拥有「查看、编辑、新增」权限,超级管理员则拥有「查看、编辑、新增、删除」权限。

三者关系可用下图概括。

接下来,我们将用 Nest.js 从零开始搭建一套这样的权限体系。

创建数据库

首先建库。我们使用 MySQL,执行:

js 复制代码
CREATE DATABASE `nest-database` DEFAULT CHARACTER SET utf8 COLLATE utf8_general_ci;

初始化项目

新建 Nest 项目:

js 复制代码
nest new nest-project

安装数据库依赖,主要是 typeormmysql2

js 复制代码
npm install --save @nestjs/typeorm typeorm mysql2

app.module.ts 中配置 TypeORM:

js 复制代码
import { Module } from '@nestjs/common';
import { AppController } from './app.controller';
import { AppService } from './app.service';
import { TypeOrmModule } from '@nestjs/typeorm';

@Module({
  imports: [
    TypeOrmModule.forRoot({
      type: 'mysql',
      host: 'localhost',
      port: 3306,
      username: 'root',
      password: 'password',
      database: 'nest-database',
      synchronize: true,
      logging: true,
      entities: [__dirname + '/**/*.entity{.ts,.js}'],
      poolSize: 10,
      connectorPackage: 'mysql2',
    }),
  ],
  controllers: [AppController],
  providers: [AppService],
})
export class AppModule {}

表结构设计

典型的 RBAC 系统需要 5 张表:

  • user:用户表,存储用户名、密码、邮箱等基本信息。
  • role:角色表,存储角色名称、角色编码等。
  • permission:权限表,存储权限名称、权限编码等。
  • user_role_relation:用户与角色的多对多关系表。
  • role_permission_relation:角色与权限的多对多关系表。

整体关系如下图。

接下来在 Nest 中创建三张实体表并定义关联。

user.entity.ts

js 复制代码
import {
  Column,
  CreateDateColumn,
  Entity,
  JoinTable,
  ManyToMany,
  PrimaryGeneratedColumn,
  UpdateDateColumn,
} from 'typeorm';
import { Role } from './role.entity';

@Entity()
export class User {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ length: 50 })
  username: string;

  @Column({ length: 50 })
  password: string;

  @CreateDateColumn()
  createTime: Date;

  @UpdateDateColumn()
  updateTime: Date;

  @ManyToMany(() => Role)
  @JoinTable({
    name: 'user_role_relation',
    joinColumn: { name: 'userId', referencedColumnName: 'id' },
    inverseJoinColumn: { name: 'roleId', referencedColumnName: 'id' },
  })
  roles: Role[];
}

User 表通过 roles 字段与中间表 user_role_relation 关联:
user.id === userRoleRelation.userIdrole.id === userRoleRelation.roleId

匹配的 Role 记录会自动挂载到 User

role.entity.ts

js 复制代码
import {
  Column,
  CreateDateColumn,
  Entity,
  JoinTable,
  ManyToMany,
  PrimaryGeneratedColumn,
  UpdateDateColumn,
} from 'typeorm';
import { Permission } from './permission.entity';

@Entity()
export class Role {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ length: 20 })
  name: string;

  @CreateDateColumn()
  createTime: Date;

  @UpdateDateColumn()
  updateTime: Date;

  @ManyToMany(() => Permission)
  @JoinTable({
    name: 'role_permission_relation',
    joinColumn: { name: 'roleId', referencedColumnName: 'id' },
    inverseJoinColumn: { name: 'permissionId', referencedColumnName: 'id' },
  })
  permissions: Permission[];
}

Role 表通过 permissions 字段与中间表 role_permission_relation 关联,逻辑同上。

permission.entity.ts

js 复制代码
import {
  Column,
  CreateDateColumn,
  Entity,
  PrimaryGeneratedColumn,
  UpdateDateColumn,
} from 'typeorm';

@Entity()
export class Permission {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ length: 50 })
  name: string;

  @Column({ length: 100, nullable: true })
  desc: string;

  @CreateDateColumn()
  createTime: Date;

  @UpdateDateColumn()
  updateTime: Date;
}

Permission 表仅记录可用权限,不与其他表做直接关联。

数据初始化

下面是一个初始化测试数据的服务:

js 复制代码
async function initData() {
  const user1 = new User();
  user1.username = 'Alice';
  user1.password = 'aaaaaa';

  const user2 = new User();
  user2.username = 'Bob';
  user2.password = 'bbbbbb';

  const user3 = new User();
  user3.username = 'Charlie';
  user3.password = 'cccccc';

  const role1 = new Role();
  role1.name = 'Administrator';

  const role2 = new Role();
  role2.name = 'Regular User';

  const permission1 = new Permission();
  permission1.name = 'Add resource_a';

  const permission2 = new Permission();
  permission2.name = 'Edit resource_a';

  const permission3 = new Permission();
  permission3.name = 'Delete resource_a';

  const permission4 = new Permission();
  permission4.name = 'Query resource_a';

  const permission5 = new Permission();
  permission5.name = 'Add resource_b';

  const permission6 = new Permission();
  permission6.name = 'Edit resource_b';

  const permission7 = new Permission();
  permission7.name = 'Delete resource_b';

  const permission8 = new Permission();
  permission8.name = 'Query resource_b';

  role1.permissions = [
    permission1,
    permission2,
    permission3,
    permission4,
    permission5,
    permission6,
    permission7,
    permission8,
  ];

  role2.permissions = [permission1, permission2, permission3, permission4];

  user1.roles = [role1];
  user2.roles = [role2];

  await this.entityManager.save(Permission, [
    permission1,
    permission2,
    permission3,
    permission4,
    permission5,
    permission6,
    permission7,
    permission8,
  ]);

  await this.entityManager.save(Role, [role1, role2]);
  await this.entityManager.save(User, [user1, user2]);
}

通过浏览器或 Postman 调用该 initData 服务,即可完成数据写入。

原文:dev.to/leapcell/de...

相关推荐
张洪权13 小时前
nest.js websocket 群聊----私聊功能
前端·nestjs
用户71282896844671 天前
NestJS 发布v12版本,看一下有哪些改动
nestjs
YIAN6 天前
NestJS 入门核心梳理:模块化架构、装饰器与依赖注入
后端·nestjs
半个落月8 天前
NestJS 入门实战:从工厂模式到 Todo CRUD,讲透模块化、依赖注入与测试
后端·nestjs
浮生望8 天前
NestJS MVC 实战:模块化架构、CRUD 全流程与标准化错误处理
nestjs
浮生望8 天前
NestJS 架构解密:从工厂模式到装饰器模式的企业级Node.js框架设计
nestjs
烬羽9 天前
Dockerfile 是"做奶茶的配方"?用 todos 全栈项目看懂 Docker 构建与发布
nginx·docker·nestjs
Liora_Yvonne10 天前
从数据库到 Vue 页面:用 LY Fullstack 完成第一个真实全栈业务模块
前端·后端·nestjs
倾颜12 天前
NestJS 核心概念梳理:从 Module、Controller 到 Guard、Interceptor
后端·node.js·nestjs
薛定谔的算法13 天前
NestJS:让 Node.js 后端告别「野路子」
后端·node.js·nestjs