282 lines
8.3 KiB
Go
282 lines
8.3 KiB
Go
package logic
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
|
||
"easygoadmin/api/v1"
|
||
"easygoadmin/internal/consts"
|
||
"easygoadmin/internal/dao"
|
||
"easygoadmin/internal/service"
|
||
"easygoadmin/utility/cache"
|
||
)
|
||
|
||
// ============================== 角色业务实现 ==============================
|
||
|
||
// sRole 角色业务实现
|
||
// 实现 service.IRole 接口，接口定义见 api/v1/role.go
|
||
type sRole struct{}
|
||
|
||
func init() {
|
||
service.RegisterRole(&sRole{})
|
||
}
|
||
|
||
// roleSerializeMaps 枚举字段到字典编码的映射
|
||
var roleSerializeMaps = map[string]string{
|
||
"data_scope": "role_data_scope",
|
||
}
|
||
|
||
// roleSpec 角色模块声明
|
||
//
|
||
// 编码唯一 + 删除守卫 + 关联清理
|
||
var roleSpec = ModuleSpec[v1.RoleItem, v1.RoleListItem, v1.RoleDetailRes]{
|
||
Model: dao.Role.Ctx,
|
||
LikeFields: []string{"name", "code"},
|
||
OrderBy: "sort, id",
|
||
SerializeMaps: roleSerializeMaps,
|
||
UniqueFields: []UniqueRule{{Field: "code"}},
|
||
FieldLabels: map[string]string{"code": "角色编码"}, // 唯一提示文案
|
||
ToData: roleToData,
|
||
AfterUpdate: roleClearCache, // 身份缓存失效
|
||
BeforeDelete: roleDeleteGuard, // 存在性 + 关联用户禁止
|
||
AfterDelete: roleCleanRelations, // 清理 role_menu
|
||
AfterBatchDelete: func(ctx context.Context, ids []int64) error {
|
||
return roleCleanRelationsBatch(ctx, ids)
|
||
},
|
||
AfterPageList: roleFillUserCount, // 行内用户数批量补全
|
||
}
|
||
|
||
// roleToData 全量表单语义转换
|
||
//
|
||
// 修改请求不包含编码，因编码不允许变更；tenant_id 由 hooks 自动填充；
|
||
// 数据范围缺省归一为"全部"（1）
|
||
func roleToData(req any) map[string]any {
|
||
switch r := req.(type) {
|
||
case *v1.RoleAddReq:
|
||
scope := r.DataScope
|
||
if scope <= 0 {
|
||
scope = 1
|
||
}
|
||
return map[string]any{"name": r.Name, "code": r.Code, "data_scope": scope, "sort": r.Sort, "note": r.Note}
|
||
case *v1.RoleUpdateReq:
|
||
scope := r.DataScope
|
||
if scope <= 0 {
|
||
scope = 1
|
||
}
|
||
return map[string]any{"name": r.Name, "data_scope": scope, "sort": r.Sort, "note": r.Note}
|
||
}
|
||
panic("roleToData: 不支持的入参类型")
|
||
}
|
||
|
||
// roleFillUserCount 分页行用户数批量补全
|
||
// 依据 user_role 关联统计
|
||
func roleFillUserCount(ctx context.Context, list []*v1.RoleItem, req any) error {
|
||
counts, err := roleUserCounts(ctx)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for _, item := range list {
|
||
item.UserCount = counts[item.Id]
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// roleClearCache 角色变更后清理身份缓存
|
||
// 失效该角色下所有用户的身份缓存
|
||
func roleClearCache(ctx context.Context, id int64, req any) error {
|
||
roleClearIdentityCache(ctx, id)
|
||
return nil
|
||
}
|
||
|
||
// roleDeleteGuard 删除守卫
|
||
// 校验记录存在性，存在关联用户时禁止删除
|
||
func roleDeleteGuard(ctx context.Context, id int64) error {
|
||
if n, err := dao.Role.Ctx(ctx).Where("id", id).Count(); err != nil {
|
||
return err
|
||
} else if n == 0 {
|
||
return gerror.NewCode(consts.CodeRecordNotFound)
|
||
}
|
||
if n, err := dao.UserRole.Ctx(ctx).Where("role_id", id).Count(); err != nil {
|
||
return err
|
||
} else if n > 0 {
|
||
return gerror.NewCode(consts.CodeInUse)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// roleCleanRelations 单条删除后清理角色菜单关联
|
||
func roleCleanRelations(ctx context.Context, id int64) error {
|
||
_, err := dao.RoleMenu.Ctx(ctx).Where("role_id", id).Delete()
|
||
return err
|
||
}
|
||
|
||
// roleCleanRelationsBatch 批量删除后清理角色菜单关联
|
||
func roleCleanRelationsBatch(ctx context.Context, ids []int64) error {
|
||
_, err := dao.RoleMenu.Ctx(ctx).WhereIn("role_id", ids).Delete()
|
||
return err
|
||
}
|
||
|
||
// ---------- 标准方法：3行委托 ----------
|
||
|
||
// Page 角色分页
|
||
// 分页返回角色列表，并补充关联用户数
|
||
func (s *sRole) Page(ctx context.Context, req *v1.RolePageReq) (res *v1.RolePageRes, err error) {
|
||
list, total, err := roleSpec.Page(ctx, req, req.GetPageNo(), req.GetPageSize())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return v1.NewPageData(req.GetPageNo(), req.GetPageSize(), total, derefSlice(list)), nil
|
||
}
|
||
|
||
// List 角色全量列表
|
||
// 按关键词返回角色精简列表
|
||
func (s *sRole) List(ctx context.Context, req *v1.RoleListReq) (res v1.RoleListRes, err error) {
|
||
list, err := roleSpec.List(ctx, req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return derefSlice(list), nil
|
||
}
|
||
|
||
// Detail 角色详情
|
||
// 按角色ID查询详情
|
||
func (s *sRole) Detail(ctx context.Context, req *v1.RoleDetailReq) (res *v1.RoleDetailRes, err error) {
|
||
return roleSpec.Detail(ctx, req.Id, req)
|
||
}
|
||
|
||
// Add 新增角色
|
||
// 校验角色编码唯一后新增，返回新记录ID
|
||
func (s *sRole) Add(ctx context.Context, req *v1.RoleAddReq) (res *v1.IdRes, err error) {
|
||
id, err := roleSpec.Add(ctx, req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &v1.IdRes{Id: id}, nil
|
||
}
|
||
|
||
// Update 修改角色
|
||
// 按角色ID更新记录，编码不允许变更
|
||
func (s *sRole) Update(ctx context.Context, req *v1.RoleUpdateReq) (res *v1.RoleUpdateRes, err error) {
|
||
if err = roleSpec.Update(ctx, req.Id, req); err != nil {
|
||
return nil, err
|
||
}
|
||
return &v1.RoleUpdateRes{}, nil
|
||
}
|
||
|
||
// Delete 删除角色
|
||
// 按角色ID删除单条记录，存在关联用户时禁止
|
||
func (s *sRole) Delete(ctx context.Context, req *v1.RoleDeleteReq) (res *v1.RoleDeleteRes, err error) {
|
||
if err = roleSpec.Delete(ctx, req.Id); err != nil {
|
||
return nil, err
|
||
}
|
||
return &v1.RoleDeleteRes{}, nil
|
||
}
|
||
|
||
// BatchDelete 批量删除角色
|
||
// 按角色ID列表批量删除记录
|
||
func (s *sRole) BatchDelete(ctx context.Context, req *v1.RoleBatchDeleteReq) (res *v1.RoleBatchDeleteRes, err error) {
|
||
if err = roleSpec.BatchDelete(ctx, req.Ids); err != nil {
|
||
return nil, err
|
||
}
|
||
return &v1.RoleBatchDeleteRes{}, nil
|
||
}
|
||
|
||
// ---------- 非标准方法（手写实现） ----------
|
||
|
||
// MenuList 角色菜单勾选列表
|
||
// 返回菜单平铺列表并附带勾选状态，供权限配置使用
|
||
func (s *sRole) MenuList(ctx context.Context, req *v1.RoleMenuListReq) (res v1.RoleMenuListRes, err error) {
|
||
roleMenus, err := dao.RoleMenu.Ctx(ctx).Where("role_id", req.RoleId).All()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
checkedSet := make(map[int64]bool, len(roleMenus))
|
||
for _, r := range roleMenus {
|
||
checkedSet[r["menu_id"].Int64()] = true
|
||
}
|
||
rows, err := dao.Menu.Ctx(ctx).Order("sort, id").All()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
list := make([]v1.RoleMenuItem, 0, len(rows))
|
||
for _, r := range rows {
|
||
list = append(list, v1.RoleMenuItem{
|
||
Id: r["id"].Int64(),
|
||
ParentId: r["parent_id"].Int(),
|
||
Name: r["name"].String(),
|
||
Icon: r["icon"].String(),
|
||
Icon2: r["icon2"].String(),
|
||
Path: r["path"].String(),
|
||
Component: r["component"].String(),
|
||
Permission: r["permission"].String(),
|
||
Target: r["target"].Int(),
|
||
Type: r["type"].Int(),
|
||
Status: r["status"].Int(),
|
||
Hide: r["hide"].Int(),
|
||
Sort: r["sort"].Int(),
|
||
Checked: checkedSet[r["id"].Int64()],
|
||
})
|
||
}
|
||
return list, nil
|
||
}
|
||
|
||
// SavePermission 保存角色菜单权限
|
||
//
|
||
// 全量替换角色菜单关联，并失效相关用户的身份缓存
|
||
func (s *sRole) SavePermission(ctx context.Context, req *v1.RoleSavePermissionReq) (res *v1.RoleSavePermissionRes, err error) {
|
||
if _, err = dao.RoleMenu.Ctx(ctx).Where("role_id", req.RoleId).Delete(); err != nil {
|
||
return nil, err
|
||
}
|
||
for _, mid := range req.MenuIds {
|
||
if mid <= 0 {
|
||
continue
|
||
}
|
||
if _, err = dao.RoleMenu.Ctx(ctx).Data(map[string]any{
|
||
"role_id": req.RoleId,
|
||
"menu_id": mid,
|
||
}).Insert(); err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
s.clearRoleIdentityCache(ctx, req.RoleId)
|
||
return &v1.RoleSavePermissionRes{}, nil
|
||
}
|
||
|
||
// ---------- 私有辅助 ----------
|
||
|
||
// roleUserCounts 统计角色ID到用户数的映射
|
||
func roleUserCounts(ctx context.Context) (map[int64]int, error) {
|
||
result := make(map[int64]int)
|
||
rows, err := dao.UserRole.Ctx(ctx).All()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
for _, r := range rows {
|
||
result[r["role_id"].Int64()]++
|
||
}
|
||
return result, nil
|
||
}
|
||
|
||
// roleClearIdentityCache 失效角色下所有用户的身份缓存
|
||
func roleClearIdentityCache(ctx context.Context, roleId int64) {
|
||
urRows, err := dao.UserRole.Ctx(ctx).Where("role_id", roleId).All()
|
||
if err != nil {
|
||
return
|
||
}
|
||
for _, r := range urRows {
|
||
uid := int(r["user_id"].Int64())
|
||
cache.Del(ctx,
|
||
consts.CacheKeyUserIdentity(uid),
|
||
consts.CacheKeyUserPerms(uid),
|
||
consts.CacheKeyUserRoles(uid),
|
||
)
|
||
}
|
||
}
|
||
|
||
// clearRoleIdentityCache 身份缓存失效（方法版）
|
||
// 供 SavePermission 调用
|
||
func (s *sRole) clearRoleIdentityCache(ctx context.Context, roleId int64) {
|
||
roleClearIdentityCache(ctx, roleId)
|
||
}
|