签入脚本
This commit is contained in:
小蚂蚁云
@@ -0,0 +1,281 @@
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user