323 lines
12 KiB
Go
323 lines
12 KiB
Go
// +======================================================================
|
||
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
||
// +----------------------------------------------------------------------
|
||
// | 开源协议: Apache-2.0
|
||
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
||
// +----------------------------------------------------------------------
|
||
// | 官方网站: https://www.easygoadmin.vip
|
||
// +----------------------------------------------------------------------
|
||
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
||
// +======================================================================
|
||
// | 版权和免责声明:
|
||
// | 一、知识产权声明
|
||
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
||
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
||
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
||
// | 二、使用限制与禁止条款
|
||
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
||
// | 1. 任何违反中华人民共和国法律法规的用途；
|
||
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
||
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
||
// | 三、责任豁免
|
||
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
||
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
||
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
||
// | 四、违约与法律责任
|
||
// | 如发现任何违反本声明的行为，本团队有权：
|
||
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
||
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
||
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
||
// | 五、许可范围说明
|
||
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
||
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
||
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
||
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
||
// | 成的一切经济损失及商誉损害。
|
||
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
||
// | 许可协议》附件为准。
|
||
// +======================================================================
|
||
|
||
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)
|
||
}
|