kenneth/projectx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v2_2

Browse Source
This commit is contained in:
kenneth
2025-03-31 11:59:42 +08:00
parent 963e1e005e
commit 6fb06c456c
52 changed files with 2244 additions and 753 deletions
Download Patch File Download Diff File Expand all files Collapse all files

165
internal/erpserver/biz/v1/system/menu.go
View File

@@ -9,9 +9,9 @@ import (
"management/internal/db/model/dto"
db "management/internal/db/sqlc"
"management/internal/erpserver/model/view"
"management/internal/global/keys"
"management/internal/pkg/redis"
"management/internal/pkg/session"
)
type MenuBiz interface {
@@ -19,34 +19,87 @@ type MenuBiz interface {
}
type MenuExpansion interface {
Create(ctx context.Context, arg *db.CreateSysMenuParams) (*db.SysMenu, error)
Update(ctx context.Context, arg *db.UpdateSysMenuParams) (*db.SysMenu, error)
Get(ctx context.Context, id int32) (*db.SysMenu, error)
GetSysMenuByUrl(ctx context.Context, url string) (*db.SysMenu, error)
AllMenusCache(ctx context.Context) ([]*db.SysMenu, error)
ListMenuTree(ctx context.Context) ([]*db.SysMenuDto, error)
ListOwnerMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error)
SetOwnerListMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error)
RecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error)
SetRecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error)
MapOwnerMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error)
SetOwnerMapMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error)
RefreshMenus(ctx context.Context) error
Tree(ctx context.Context, id int32) ([]*view.LayuiTree, error)
XmSelect(ctx context.Context, id int32) ([]*view.XmSelectTree, error)
}
type menuBiz struct {
store db.Store
redis redis.IRedis
session session.ISession
store db.Store
redis redis.IRedis
}
var _ MenuBiz = (*menuBiz)(nil)
func NewMenu(store db.Store, redis redis.IRedis, session session.ISession) *menuBiz {
func NewMenu(store db.Store, redis redis.IRedis) *menuBiz {
return &menuBiz{
store: store,
redis: redis,
session: session,
store: store,
redis: redis,
}
}
func (b *menuBiz) Create(ctx context.Context, arg *db.CreateSysMenuParams) (*db.SysMenu, error) {
return b.store.CreateSysMenu(ctx, arg)
}
func (b *menuBiz) Update(ctx context.Context, arg *db.UpdateSysMenuParams) (*db.SysMenu, error) {
return b.store.UpdateSysMenu(ctx, arg)
}
func (b *menuBiz) Get(ctx context.Context, id int32) (*db.SysMenu, error) {
return b.store.GetSysMenu(ctx, id)
}
func (b *menuBiz) GetSysMenuByUrl(ctx context.Context, url string) (*db.SysMenu, error) {
return b.store.GetSysMenuByUrl(ctx, url)
}
func (b *menuBiz) AllMenusCache(ctx context.Context) ([]*db.SysMenu, error) {
key := keys.GetManageKey(ctx, keys.AllMenus)
bs, err := b.redis.GetBytes(ctx, key)
if err == nil {
var res []*db.SysMenu
if err := json.Unmarshal(bs, &res); err == nil {
return res, nil
}
}
all, err := b.store.AllSysMenu(ctx)
if err != nil {
return nil, err
}
bs, err = json.Marshal(all)
if err != nil {
return nil, err
}
_ = redis.Set(ctx, key, bs, time.Hour*6)
return all, nil
}
func (b *menuBiz) ListMenuTree(ctx context.Context) ([]*db.SysMenuDto, error) {
all, err := b.AllMenusCache(ctx)
if err != nil {
return nil, err
}
return toTreeSysMenu(0, all), nil
}
func (b *menuBiz) ListOwnerMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error) {
// 判断redis是否存储
key := keys.GetManageKey(ctx, keys.OwnerMenus, roleID)
@@ -187,6 +240,40 @@ func (b *menuBiz) SetOwnerMapMenuByRoleID(ctx context.Context, roleID int32) (ma
return result, nil
}
func (b *menuBiz) RefreshMenus(ctx context.Context) error {
all, err := b.store.AllSysMenu(ctx)
if err != nil {
return err
}
bs, err := json.Marshal(all)
if err != nil {
return err
}
key := keys.GetManageKey(ctx, keys.AllMenus)
err = redis.Set(ctx, key, bs, time.Hour*6)
return err
}
func (b *menuBiz) Tree(ctx context.Context, id int32) ([]*view.LayuiTree, error) {
all, err := b.AllMenusCache(ctx)
if err != nil {
return nil, err
}
return b.toTree(id, all), nil
}
func (b *menuBiz) XmSelect(ctx context.Context, id int32) ([]*view.XmSelectTree, error) {
all, err := b.AllMenusCache(ctx)
if err != nil {
return nil, err
}
return b.toXmSelectTree(id, all), nil
}
func (b *menuBiz) ownerMenusByRoleID(ctx context.Context, roleID int32) ([]*db.SysMenu, error) {
// 判断当前用户是否有vip角色
role, err := b.store.GetSysRole(ctx, roleID)
@@ -214,6 +301,68 @@ func (b *menuBiz) ownerMenusByRoleID(ctx context.Context, roleID int32) ([]*db.S
return menus, nil
}
func (b *menuBiz) toTree(parentId int32, data []*db.SysMenu) []*view.LayuiTree {
var res []*view.LayuiTree
for _, v := range data {
if v.ParentID == parentId {
item := view.LayuiTree{}
item.ID = strconv.FormatInt(int64(v.ID), 10)
item.Title = v.DisplayName
item.Children = b.toTree(v.ID, data)
if v.ParentID == 0 {
item.Spread = true
}
res = append(res, &item)
}
}
return res
}
func (b *menuBiz) toXmSelectTree(parentId int32, data []*db.SysMenu) []*view.XmSelectTree {
var res []*view.XmSelectTree
for _, v := range data {
if v.ParentID == parentId {
item := view.XmSelectTree{
Name: v.DisplayName,
Value: strconv.FormatInt(int64(v.ID), 10),
Children: b.toXmSelectTree(v.ID, data),
}
res = append(res, &item)
}
}
return res
}
func toTreeSysMenu(parentId int32, data []*db.SysMenu) []*db.SysMenuDto {
var res []*db.SysMenuDto
for _, v := range data {
if v.ParentID == parentId {
item := db.SysMenuDto{}
item.ID = v.ID
item.Name = v.Name
item.DisplayName = v.DisplayName
item.Url = v.Url
item.Type = v.Type
item.ParentID = v.ParentID
item.ParentPath = v.ParentPath
item.Avatar = v.Avatar
item.Style = v.Style
item.Visible = v.Visible
item.IsList = v.IsList
item.Status = v.Status
item.Sort = v.Sort
item.CreatedAt = v.CreatedAt
item.UpdatedAt = v.UpdatedAt
item.Children = toTreeSysMenu(v.ID, data)
res = append(res, &item)
}
}
return res
}
func convertToMenuUIDto(data []*db.RecursiveSysMenusRow) []*db.SysMenu {
var res []*db.SysMenu