feat: 添加角色类型常量及用户主角色获取方法

This commit is contained in:
unclesam committed 2026-07-08 08:19:51 +08:00
1 parent c5977ce354
commit 20ad8115da
14 files changed
+123 -64

No files matched your search

+18
View File
@@ -0,0 +1,18 @@
package ent
// GetPrimaryRole 获取用户主角色(权限最高的角色)
func (u *User) GetPrimaryRole() *Role {
// 如果没有预加载角色,或者角色列表为空,直接返回 nil
if u.Edges.Roles == nil || len(u.Edges.Roles) == 0 {
return nil
}
primary := u.Edges.Roles[0]
for _, r := range u.Edges.Roles {
if r.Code < primary.Code {
primary = r
}
}
return primary
}