EasyGoAdmin_GoFrame_EleVue/internal/logic/menu.go

package logic

import (
	"context"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/errors/gerror"

	"easygoadmin/api/v1"
	"easygoadmin/internal/consts"
	"easygoadmin/internal/dao"
	"easygoadmin/internal/service"
	"easygoadmin/utility/cache"
)

// ============================== 菜单业务实现 ==============================

// sMenu 菜单业务实现
// 实现 service.IMenu 接口，接口定义见 api/v1/menu.go
type sMenu struct{}

func init() {
	service.RegisterMenu(&sMenu{})
}

// menuSerializeMaps 枚举字段到字典编码的映射
// {field}Text 由基座自动填充
var menuSerializeMaps = map[string]string{
	"type":   "menu_type",
	"status": "menu_status",
}

// menuSpec 菜单模块声明
//
// 平铺列表 + 上级循环校验 + 删除守卫，并连带清理 role_menu 与缓存；
// 树形结构由前端 buildTree 构建
var menuSpec = ModuleSpec[v1.MenuItem, v1.MenuItem, v1.MenuDetailRes]{
	Model:         dao.Menu.Ctx,
	LikeFields:    []string{"name", "path"},
	OrderBy:       "sort, id",
	SerializeMaps: menuSerializeMaps,
	Filter:        menuFilter, // Type/Status 为 *int 指针（nil=不筛，指向0也筛），EqFields 表达不了
	ToData:        menuToData,
	BeforeUpdate:  menuParentGuard,    // 禁止挂到自身/后代下
	BeforeDelete:  menuDeleteGuard,    // 存在性 + 有子菜单禁止
	AfterDelete:   menuCleanRelations, // 清理 role_menu + 失效受影响用户缓存
	AfterBatchDelete: func(ctx context.Context, ids []int64) error {
		return menuCleanRelationsBatch(ctx, ids)
	},
}

// menuFilter 复杂筛选
// Type 与 Status 为指针字段，nil 时跳过，指向 0 时同样生效
func menuFilter(ctx context.Context, m *gdb.Model, req any) *gdb.Model {
	switch r := req.(type) {
	case *v1.MenuPageReq:
		if r.Type != nil {
			m = m.Where("type", *r.Type)
		}
		if r.Status != nil {
			m = m.Where("status", *r.Status)
		}
	case *v1.MenuListReq:
		if r.Type != nil {
			m = m.Where("type", *r.Type)
		}
		if r.Status != nil {
			m = m.Where("status", *r.Status)
		}
	}
	return m
}

// menuToData 全量表单语义转换
// 新增与修改均提交全部字段；修改时 icon/icon2/note 未传值则不落键，保留库中原值
func menuToData(req any) map[string]any {
	switch r := req.(type) {
	case *v1.MenuAddReq:
		return map[string]any{
			"parent_id": r.ParentId, "name": r.Name, "icon": r.Icon, "icon2": r.Icon2, "path": r.Path,
			"component": r.Component, "permission": r.Permission, "target": r.Target,
			"type": r.Type, "status": r.Status, "hide": r.Hide, "sort": r.Sort, "note": r.Note,
		}
	case *v1.MenuUpdateReq:
		data := map[string]any{
			"parent_id": r.ParentId, "name": r.Name, "path": r.Path,
			"component": r.Component, "permission": r.Permission, "target": r.Target,
			"type": r.Type, "status": r.Status, "hide": r.Hide, "sort": r.Sort,
		}
		if r.Icon != nil {
			data["icon"] = *r.Icon
		}
		if r.Icon2 != nil {
			data["icon2"] = *r.Icon2
		}
		if r.Note != nil {
			data["note"] = *r.Note
		}
		return data
	}
	panic("menuToData: 不支持的入参类型")
}

// menuParentGuard 上级循环校验
// 禁止将上级设置为自身或自己的后代节点
func menuParentGuard(ctx context.Context, data map[string]any, id int64, req any) error {
	r := mustAs[*v1.MenuUpdateReq](req)
	if r.ParentId <= 0 {
		return nil
	}
	if int64(r.ParentId) == id {
		return gerror.NewCodef(consts.CodeParamError, "上级不能选择自己")
	}
	descIds, err := menuDescendantIds(ctx, id)
	if err != nil {
		return err
	}
	for _, did := range descIds {
		if did == int64(r.ParentId) {
			return gerror.NewCodef(consts.CodeParamError, "上级不能选择自己的下级")
		}
	}
	return nil
}

// menuDeleteGuard 删除守卫
// 校验记录存在性，存在子菜单时禁止删除
func menuDeleteGuard(ctx context.Context, id int64) error {
	if n, err := dao.Menu.Ctx(ctx).Where("id", id).Count(); err != nil {
		return err
	} else if n == 0 {
		return gerror.NewCode(consts.CodeRecordNotFound)
	}
	if n, err := dao.Menu.Ctx(ctx).Where("parent_id", id).Count(); err != nil {
		return err
	} else if n > 0 {
		return gerror.NewCode(consts.CodeHasChildren)
	}
	return nil
}

// menuCleanRelations 单条删除后清理
// 清理 role_menu 关联并失效受影响用户的缓存
func menuCleanRelations(ctx context.Context, id int64) error {
	return menuCleanRelationsBatch(ctx, []int64{id})
}

// menuCleanRelationsBatch 批量删除后清理
// 清理 role_menu 关联并失效受影响用户的缓存
func menuCleanRelationsBatch(ctx context.Context, ids []int64) error {
	roleMenus, err := dao.RoleMenu.Ctx(ctx).WhereIn("menu_id", ids).All()
	if err != nil {
		return err
	}
	if _, err = dao.RoleMenu.Ctx(ctx).WhereIn("menu_id", ids).Delete(); err != nil {
		return err
	}
	if len(roleMenus) == 0 {
		return nil
	}
	roleIds := make([]int64, 0, len(roleMenus))
	for _, rm := range roleMenus {
		roleIds = append(roleIds, rm["role_id"].Int64())
	}
	urRows, err := dao.UserRole.Ctx(ctx).WhereIn("role_id", roleIds).All()
	if err != nil {
		return err
	}
	for _, ur := range urRows {
		uid := int(ur["user_id"].Int64())
		cache.Del(ctx,
			consts.CacheKeyUserIdentity(uid),
			consts.CacheKeyUserPerms(uid),
			consts.CacheKeyUserRoles(uid),
		)
	}
	return nil
}

// menuDescendantIds 递归获取全部后代ID
// 采用广度优先遍历，返回指定菜单下所有层级的后代节点ID
func menuDescendantIds(ctx context.Context, menuId int64) ([]int64, error) {
	result := make([]int64, 0)
	queue := []int64{menuId}
	for len(queue) > 0 {
		current := queue[0]
		queue = queue[1:]
		rows, err := dao.Menu.Ctx(ctx).Where("parent_id", current).Fields("id").All()
		if err != nil {
			return nil, err
		}
		for _, r := range rows {
			idv := r["id"].Int64()
			result = append(result, idv)
			queue = append(queue, idv)
		}
	}
	return result, nil
}

// ---------- 标准方法：3行委托 ----------

// Page 菜单分页（平铺）
// 按查询条件分页返回平铺的菜单列表
func (s *sMenu) Page(ctx context.Context, req *v1.MenuPageReq) (res *v1.MenuPageRes, err error) {
	list, total, err := menuSpec.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 菜单平铺列表
// data 直接为数组，树形结构由前端 buildTree 构建
func (s *sMenu) List(ctx context.Context, req *v1.MenuListReq) (res v1.MenuListRes, err error) {
	list, err := menuSpec.List(ctx, req)
	if err != nil {
		return nil, err
	}
	return list, nil
}

// Detail 菜单详情
// 按菜单ID查询详情
func (s *sMenu) Detail(ctx context.Context, req *v1.MenuDetailReq) (res *v1.MenuDetailRes, err error) {
	return menuSpec.Detail(ctx, req.Id, req)
}

// Add 新增菜单
// 新增成功后返回新记录ID
func (s *sMenu) Add(ctx context.Context, req *v1.MenuAddReq) (res *v1.IdRes, err error) {
	id, err := menuSpec.Add(ctx, req)
	if err != nil {
		return nil, err
	}
	return &v1.IdRes{Id: id}, nil
}

// Update 修改菜单
// 按菜单ID更新记录
func (s *sMenu) Update(ctx context.Context, req *v1.MenuUpdateReq) (res *v1.MenuUpdateRes, err error) {
	if err = menuSpec.Update(ctx, req.Id, req); err != nil {
		return nil, err
	}
	return &v1.MenuUpdateRes{}, nil
}

// Delete 删除菜单
// 按菜单ID删除单条记录，并清理关联关系与缓存
func (s *sMenu) Delete(ctx context.Context, req *v1.MenuDeleteReq) (res *v1.MenuDeleteRes, err error) {
	if err = menuSpec.Delete(ctx, req.Id); err != nil {
		return nil, err
	}
	return &v1.MenuDeleteRes{}, nil
}

// BatchDelete 批量删除菜单
// 按菜单ID列表批量删除记录，并清理关联关系与缓存
func (s *sMenu) BatchDelete(ctx context.Context, req *v1.MenuBatchDeleteReq) (res *v1.MenuBatchDeleteRes, err error) {
	if err = menuSpec.BatchDelete(ctx, req.Ids); err != nil {
		return nil, err
	}
	return &v1.MenuBatchDeleteRes{}, nil
}