RBAC核心结构应建Role、Permission、User三者通过关联表解耦,避免用户直连权限;用结构体+显式RolePermission关联表支持层级与动态更新,并配联合索引防N+1。
RBAC核心结构怎么建才不踩坑
Go里实现RBAC,关键不是堆代码,而是把Role、Permission、User三者关系建对。别一上来就写中间件——先想清楚:权限是绑定到角色还是直接绑用户?推荐角色继承链(如 admin → editor → viewer),避免“用户-权限”直连,否则后期删权限时要遍历所有用户。
常见错误是用 map[string][]string 存角色权限,看似简单,但无法表达层级和动态更新。实际建议用结构体 + 显式关联表:
type Role struct {
ID uint `gorm:"primaryKey"`
Name string `gorm:"uniqueIndex"`
}
type Permission struct {
ID uint `gorm:"primaryKey"`
Code string `gorm:"uniqueIndex"` // 如 "post:read", "user:delete"
}
type RolePermission struct {
RoleID uint `gorm:"primaryKey;column:role_id"`
PermissionID uint `gorm:"primaryKey;column:permission_id"`
}
这样查权限时能走数据库 JOIN,也方便加缓存层。
如何在HTTP handler里快速校验权限
别在每个 handler 里手写 if !hasPerm(u, "post:write")——封装成 Gin/Echo 的中间件最稳。核心逻辑是:从 context 取当前 userID → 查其所有角色 → 查角色对应的所有 Permission.Code → 做字符串匹配或前缀匹配(如 "post:*" 匹配 "post:create")。
实操注意点:
- 权限检查必须放在 JWT 解析之后,且依赖可靠的
userID来源,别信前端传的 user_id - 用
sync.Map缓存roleID → []string{perm1, perm2},避免每次请求都查库 - 支持通配符时,用
strings.HasPrefix比正则快得多,别上regexp - 若用 Gin,中间件返回
ctx.AbortWithStatusJSON(403, ...),别忘了return
GORM 关联查询权限时为什么总 N+1
查一个用户的所有权限,容易写出这种低效代码:user.Roles[i].Permissions 循环触发 N 次 SQL。GORM 默认不会预加载嵌套关系。