mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-27 03:39:37 +08:00
项目目录优化，任务模块后端代码
This commit is contained in:
PandaGoAdmin
249
apps/system/services/menu.go
Normal file
249
apps/system/services/menu.go
Normal file
@@ -0,0 +1,249 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"pandax/apps/system/entity"
|
||||
"pandax/base/biz"
|
||||
"pandax/base/global"
|
||||
)
|
||||
|
||||
type (
|
||||
SysMenuModel interface {
|
||||
Insert(data entity.SysMenu) *entity.SysMenu
|
||||
FindOne(menuId int64) *entity.SysMenu
|
||||
FindListPage(page, pageSize int, data entity.SysMenu) (*[]entity.SysMenu, int64)
|
||||
FindList(data entity.SysMenu) *[]entity.SysMenu
|
||||
Update(data entity.SysMenu) *entity.SysMenu
|
||||
Delete(menuId []int64)
|
||||
SelectMenu(data entity.SysMenu) *[]entity.SysMenu
|
||||
SelectMenuLable(data entity.SysMenu) *[]entity.MenuLable
|
||||
SelectMenuRole(roleName string) *[]entity.SysMenu
|
||||
GetMenuRole(data entity.MenuRole) *[]entity.MenuRole
|
||||
}
|
||||
|
||||
sysMenuModelImpl struct {
|
||||
table string
|
||||
}
|
||||
)
|
||||
|
||||
var SysMenuModelDao SysMenuModel = &sysMenuModelImpl{
|
||||
table: `sys_menus`,
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) Insert(data entity.SysMenu) *entity.SysMenu {
|
||||
err := global.Db.Table(m.table).Create(&data).Error
|
||||
biz.ErrIsNil(err, "添加菜单失败")
|
||||
//m.InitPaths(&data)
|
||||
return &data
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) FindOne(menuId int64) *entity.SysMenu {
|
||||
resData := new(entity.SysMenu)
|
||||
err := global.Db.Table(m.table).Where("`menu_id` = ?", menuId).First(resData).Error
|
||||
biz.ErrIsNil(err, "查询菜单失败")
|
||||
return resData
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) FindListPage(page, pageSize int, data entity.SysMenu) (*[]entity.SysMenu, int64) {
|
||||
list := make([]entity.SysMenu, 0)
|
||||
var total int64 = 0
|
||||
|
||||
offset := pageSize * (page - 1)
|
||||
db := global.Db.Table(m.table)
|
||||
// 此处填写 where参数判断
|
||||
db.Where("delete_time IS NULL")
|
||||
err := db.Count(&total).Error
|
||||
err = db.Limit(pageSize).Offset(offset).Find(&list).Error
|
||||
biz.ErrIsNil(err, "查询分页菜单失败")
|
||||
return &list, total
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) FindList(data entity.SysMenu) *[]entity.SysMenu {
|
||||
list := make([]entity.SysMenu, 0)
|
||||
|
||||
db := global.Db.Table(m.table)
|
||||
// 此处填写 where参数判断
|
||||
if data.MenuName != "" {
|
||||
db = db.Where("menu_name like ?", "%"+data.MenuName+"%")
|
||||
}
|
||||
if data.Path != "" {
|
||||
db = db.Where("path = ?", data.Path)
|
||||
}
|
||||
if data.MenuType != "" {
|
||||
db = db.Where("menu_type = ?", data.MenuType)
|
||||
}
|
||||
if data.Title != "" {
|
||||
db = db.Where("title like ?", "%"+data.Title+"%")
|
||||
}
|
||||
if data.Status != "" {
|
||||
db = db.Where("status = ?", data.Status)
|
||||
}
|
||||
db.Where("delete_time IS NULL")
|
||||
err := db.Order("sort").Find(&list).Error
|
||||
biz.ErrIsNil(err, "查询菜单列表失败")
|
||||
return &list
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) Update(data entity.SysMenu) *entity.SysMenu {
|
||||
err := global.Db.Table(m.table).Updates(&data).Error
|
||||
biz.ErrIsNil(err, "修改菜单失败")
|
||||
//m.InitPaths(&data)
|
||||
return &data
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) Delete(menuIds []int64) {
|
||||
err := global.Db.Table(m.table).Delete(&entity.SysMenu{}, "`menu_id` in (?)", menuIds).Error
|
||||
biz.ErrIsNil(err, "修改菜单失败")
|
||||
return
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) SelectMenu(data entity.SysMenu) *[]entity.SysMenu {
|
||||
menuList := m.FindList(data)
|
||||
|
||||
redData := make([]entity.SysMenu, 0)
|
||||
ml := *menuList
|
||||
for i := 0; i < len(ml); i++ {
|
||||
if ml[i].ParentId != 0 {
|
||||
continue
|
||||
}
|
||||
menusInfo := DiguiMenu(menuList, ml[i])
|
||||
redData = append(redData, menusInfo)
|
||||
}
|
||||
return &redData
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) SelectMenuLable(data entity.SysMenu) *[]entity.MenuLable {
|
||||
menuList := m.FindList(data)
|
||||
|
||||
redData := make([]entity.MenuLable, 0)
|
||||
ml := *menuList
|
||||
for i := 0; i < len(ml); i++ {
|
||||
if ml[i].ParentId != 0 {
|
||||
continue
|
||||
}
|
||||
e := entity.MenuLable{}
|
||||
e.MenuId = ml[i].MenuId
|
||||
e.MenuName = ml[i].MenuName
|
||||
menusInfo := DiguiMenuLable(menuList, e)
|
||||
|
||||
redData = append(redData, menusInfo)
|
||||
}
|
||||
return &redData
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) GetMenuByRoleKey(roleKey string) *[]entity.SysMenu {
|
||||
menus := make([]entity.SysMenu, 0)
|
||||
db := global.Db.Table(m.table).Select("sys_menus.*").Joins("left join sys_role_menus on sys_role_menus.menu_id=sys_menus.menu_id")
|
||||
db = db.Where("sys_role_menus.role_name=? and menu_type in ('M','C')", roleKey)
|
||||
db.Where("sys_menus.delete_time IS NULL")
|
||||
err := db.Order("sort").Find(&menus).Error
|
||||
biz.ErrIsNil(err, "通过角色名查询菜单失败")
|
||||
return &menus
|
||||
}
|
||||
|
||||
func (m *sysMenuModelImpl) SelectMenuRole(roleKey string) *[]entity.SysMenu {
|
||||
redData := make([]entity.SysMenu, 0)
|
||||
|
||||
menulist := m.GetMenuByRoleKey(roleKey)
|
||||
menuList := *menulist
|
||||
redData = make([]entity.SysMenu, 0)
|
||||
for i := 0; i < len(menuList); i++ {
|
||||
if menuList[i].ParentId != 0 {
|
||||
continue
|
||||
}
|
||||
menusInfo := DiguiMenu(&menuList, menuList[i])
|
||||
|
||||
redData = append(redData, menusInfo)
|
||||
}
|
||||
return &redData
|
||||
}
|
||||
func (m *sysMenuModelImpl) GetMenuRole(data entity.MenuRole) *[]entity.MenuRole {
|
||||
menus := make([]entity.MenuRole, 0)
|
||||
|
||||
db := global.Db.Table(m.table)
|
||||
if data.MenuName != "" {
|
||||
db = db.Where("menu_name = ?", data.MenuName)
|
||||
}
|
||||
db.Where("delete_time IS NULL")
|
||||
biz.ErrIsNil(db.Order("sort").Find(&menus).Error, "查询角色菜单失败")
|
||||
return &menus
|
||||
}
|
||||
|
||||
//func (m *sysMenuModelImpl) InitPaths(menu *entity.SysMenu) {
|
||||
// parentMenu := new(entity.SysMenu)
|
||||
// if int(menu.ParentId) != 0 {
|
||||
// parentMenu = m.FindOne(menu.ParentId)
|
||||
// biz.IsTrue(parentMenu.Paths != "", "父级paths异常，请尝试对当前节点父级菜单进行更新操作！")
|
||||
// menu.Paths = parentMenu.Paths + "/" + kgo.KConv.Int2Str(menu.MenuId)
|
||||
// } else {
|
||||
// menu.Paths = "/0/" + kgo.KConv.Int2Str(menu.MenuId)
|
||||
// }
|
||||
// global.Db.Table(m.table).Where("menu_id = ?", menu.MenuId).Update("paths", menu.Paths)
|
||||
// return
|
||||
//}
|
||||
|
||||
func DiguiMenu(menulist *[]entity.SysMenu, menu entity.SysMenu) entity.SysMenu {
|
||||
list := *menulist
|
||||
|
||||
min := make([]entity.SysMenu, 0)
|
||||
for j := 0; j < len(list); j++ {
|
||||
|
||||
if menu.MenuId != list[j].ParentId {
|
||||
continue
|
||||
}
|
||||
mi := entity.SysMenu{}
|
||||
mi.MenuId = list[j].MenuId
|
||||
mi.MenuName = list[j].MenuName
|
||||
mi.Title = list[j].Title
|
||||
mi.Icon = list[j].Icon
|
||||
mi.Path = list[j].Path
|
||||
mi.MenuType = list[j].MenuType
|
||||
mi.IsKeepAlive = list[j].IsKeepAlive
|
||||
mi.Permission = list[j].Permission
|
||||
mi.ParentId = list[j].ParentId
|
||||
mi.IsAffix = list[j].IsAffix
|
||||
mi.IsLink = list[j].IsLink
|
||||
mi.Component = list[j].Component
|
||||
mi.Sort = list[j].Sort
|
||||
mi.Status = list[j].Status
|
||||
mi.IsHide = list[j].IsHide
|
||||
mi.CreatedAt = list[j].CreatedAt
|
||||
mi.UpdatedAt = list[j].UpdatedAt
|
||||
mi.Children = []entity.SysMenu{}
|
||||
|
||||
if mi.MenuType != "F" {
|
||||
ms := DiguiMenu(menulist, mi)
|
||||
min = append(min, ms)
|
||||
|
||||
} else {
|
||||
min = append(min, mi)
|
||||
}
|
||||
|
||||
}
|
||||
menu.Children = min
|
||||
return menu
|
||||
}
|
||||
|
||||
func DiguiMenuLable(menulist *[]entity.SysMenu, menu entity.MenuLable) entity.MenuLable {
|
||||
list := *menulist
|
||||
|
||||
min := make([]entity.MenuLable, 0)
|
||||
for j := 0; j < len(list); j++ {
|
||||
|
||||
if menu.MenuId != list[j].ParentId {
|
||||
continue
|
||||
}
|
||||
mi := entity.MenuLable{}
|
||||
mi.MenuId = list[j].MenuId
|
||||
mi.MenuName = list[j].MenuName
|
||||
mi.Children = []entity.MenuLable{}
|
||||
if list[j].MenuType != "F" {
|
||||
ms := DiguiMenuLable(menulist, mi)
|
||||
min = append(min, ms)
|
||||
} else {
|
||||
min = append(min, mi)
|
||||
}
|
||||
|
||||
}
|
||||
menu.Children = min
|
||||
return menu
|
||||
}
|
||||
Reference in New Issue
Block a user