mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-21 10:13:01 +00:00
306 lines
6.9 KiB
Go
306 lines
6.9 KiB
Go
package service
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/go-admin-team/go-admin-core/sdk/pkg"
|
|
"go-admin/app/admin/models/system"
|
|
"go-admin/app/admin/service/dto"
|
|
cDto "go-admin/common/dto"
|
|
"go-admin/common/service"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type SysMenu struct {
|
|
service.Service
|
|
}
|
|
|
|
// GetSysMenuPage 获取SysMenu列表
|
|
func (e *SysMenu) GetSysMenuPage(c *dto.SysMenuSearch) (*[]system.SysMenu, error) {
|
|
var m = make([]system.SysMenu, 0)
|
|
var err error
|
|
var menu = make([]system.SysMenu, 0)
|
|
err = e.getSysMenuPage(c, &menu)
|
|
for i := 0; i < len(menu); i++ {
|
|
if menu[i].ParentId != 0 {
|
|
continue
|
|
}
|
|
menusInfo := menuCall(&menu, menu[i])
|
|
|
|
m = append(m, menusInfo)
|
|
}
|
|
return &m, err
|
|
}
|
|
|
|
// getSysMenuPage 菜单列表
|
|
func (e *SysMenu) getSysMenuPage(c *dto.SysMenuSearch, list *[]system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
err = e.Orm.Model(&data).
|
|
Scopes(
|
|
cDto.OrderDest("sort", false),
|
|
cDto.MakeCondition(c.GetNeedSearch()),
|
|
).
|
|
Find(list).Error
|
|
if err != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetSysMenu 获取SysMenu对象
|
|
func (e *SysMenu) GetSysMenu(d *dto.SysMenuById, model *system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
db := e.Orm.Model(&data).
|
|
First(model, d.GetId())
|
|
err = db.Error
|
|
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
|
|
err = errors.New("查看对象不存在或无权查看")
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
if db.Error != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// InsertSysMenu 创建SysMenu对象
|
|
func (e *SysMenu) InsertSysMenu(model *system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
err = e.Orm.Model(&data).
|
|
Create(model).Error
|
|
if err != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (e *SysMenu) initPaths(menu *system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
parentMenu := new(system.SysMenu)
|
|
if menu.ParentId != 0 {
|
|
e.Orm.Model(&data).First(parentMenu, menu.ParentId)
|
|
if parentMenu.Paths == "" {
|
|
err = errors.New("父级paths异常，请尝试对当前节点父级菜单进行更新操作！")
|
|
return err
|
|
}
|
|
menu.Paths = parentMenu.Paths + "/" + pkg.IntToString(menu.MenuId)
|
|
} else {
|
|
menu.Paths = "/0/" + pkg.IntToString(menu.MenuId)
|
|
}
|
|
e.Orm.Model(&data).Where("menu_id = ?", menu.MenuId).Update("paths", menu.Paths)
|
|
return err
|
|
}
|
|
|
|
// UpdateSysMenu 修改SysMenu对象
|
|
func (e *SysMenu) UpdateSysMenu(c *system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
db := e.Orm.Model(&data).
|
|
Where(c.GetId()).Updates(c)
|
|
if db.Error != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
if db.RowsAffected == 0 {
|
|
return errors.New("无权更新该数据")
|
|
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RemoveSysMenu 删除SysMenu
|
|
func (e *SysMenu) RemoveSysMenu(d *dto.SysMenuById) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
db := e.Orm.Model(&data).
|
|
Where(d.GetId()).Delete(&data)
|
|
if db.Error != nil {
|
|
err = db.Error
|
|
e.Log.Errorf("Delete error: %s", err)
|
|
return err
|
|
}
|
|
if db.RowsAffected == 0 {
|
|
err = errors.New("无权删除该数据")
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetSysMenuList 获取菜单数据
|
|
func (e *SysMenu) GetSysMenuList(c *dto.SysMenuSearch, list *[]system.SysMenu) error {
|
|
var err error
|
|
var data system.SysMenu
|
|
|
|
err = e.Orm.Model(&data).
|
|
Scopes(
|
|
cDto.MakeCondition(c.GetNeedSearch()),
|
|
).
|
|
Find(list).Error
|
|
if err != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SetSysMenuTree 设置菜单数据
|
|
func (e *SysMenu) SetSysMenuLabel() (m []dto.MenuLabel, err error) {
|
|
var list []system.SysMenu
|
|
err = e.GetSysMenuList(&dto.SysMenuSearch{}, &list)
|
|
|
|
m = make([]dto.MenuLabel, 0)
|
|
for i := 0; i < len(list); i++ {
|
|
if list[i].ParentId != 0 {
|
|
continue
|
|
}
|
|
e := dto.MenuLabel{}
|
|
e.Id = list[i].MenuId
|
|
e.Label = list[i].Title
|
|
deptsInfo := menuLabelCall(&list, e)
|
|
|
|
m = append(m, deptsInfo)
|
|
}
|
|
return
|
|
}
|
|
|
|
// 左侧菜单
|
|
func (e *SysMenu) GetSysMenuByRoleName(roleName string) (Menus []system.SysMenu, err error) {
|
|
var table *gorm.DB
|
|
var data system.SysMenu
|
|
|
|
if roleName == "admin" {
|
|
table = e.Orm.Model(&data).Select("sys_menu.*")
|
|
table = table.Where(" menu_type in ('M','C')")
|
|
} else {
|
|
table = e.Orm.Model(&data).Select("sys_menu.*").Joins("left join sys_role_menu on sys_role_menu.menu_id=sys_menu.menu_id")
|
|
table = table.Where("sys_role_menu.role_name=? and menu_type in ('M','C')", roleName)
|
|
}
|
|
|
|
err = table.Order("sort").Find(&Menus).Error
|
|
|
|
if err != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
// menuLabelCall 递归构造组织数据
|
|
func menuLabelCall(elist *[]system.SysMenu, dept dto.MenuLabel) dto.MenuLabel {
|
|
list := *elist
|
|
|
|
min := make([]dto.MenuLabel, 0)
|
|
for j := 0; j < len(list); j++ {
|
|
|
|
if dept.Id != list[j].ParentId {
|
|
continue
|
|
}
|
|
mi := dto.MenuLabel{}
|
|
mi.Id = list[j].MenuId
|
|
mi.Label = list[j].Title
|
|
mi.Children = []dto.MenuLabel{}
|
|
if list[j].MenuType != "F" {
|
|
ms := menuLabelCall(elist, mi)
|
|
min = append(min, ms)
|
|
} else {
|
|
min = append(min, mi)
|
|
}
|
|
}
|
|
if len(min) > 0 {
|
|
dept.Children = min
|
|
} else {
|
|
dept.Children = nil
|
|
}
|
|
return dept
|
|
}
|
|
|
|
func menuCall(menulist *[]system.SysMenu, menu system.SysMenu) system.SysMenu {
|
|
list := *menulist
|
|
|
|
min := make([]system.SysMenu, 0)
|
|
for j := 0; j < len(list); j++ {
|
|
|
|
if menu.MenuId != list[j].ParentId {
|
|
continue
|
|
}
|
|
mi := system.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.Action = list[j].Action
|
|
mi.Permission = list[j].Permission
|
|
mi.ParentId = list[j].ParentId
|
|
mi.NoCache = list[j].NoCache
|
|
mi.Breadcrumb = list[j].Breadcrumb
|
|
mi.Component = list[j].Component
|
|
mi.Sort = list[j].Sort
|
|
mi.Visible = list[j].Visible
|
|
mi.CreatedAt = list[j].CreatedAt
|
|
mi.Children = []system.SysMenu{}
|
|
|
|
if mi.MenuType != "F" {
|
|
ms := menuCall(menulist, mi)
|
|
min = append(min, ms)
|
|
|
|
} else {
|
|
min = append(min, mi)
|
|
}
|
|
|
|
}
|
|
menu.Children = min
|
|
return menu
|
|
}
|
|
|
|
func (e *SysMenu) SetMenuRole(roleName string) (m []system.SysMenu, err error) {
|
|
|
|
menus, err := e.getByRoleName(roleName)
|
|
|
|
m = make([]system.SysMenu, 0)
|
|
for i := 0; i < len(menus); i++ {
|
|
if menus[i].ParentId != 0 {
|
|
continue
|
|
}
|
|
menusInfo := menuCall(&menus, menus[i])
|
|
|
|
m = append(m, menusInfo)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (e *SysMenu) getByRoleName(roleName string) (Menus []system.SysMenu, err error) {
|
|
var data system.SysMenu
|
|
|
|
var table *gorm.DB
|
|
if roleName == "admin" {
|
|
table = e.Orm.Model(&data).Select("sys_menu.*")
|
|
table = table.Where(" menu_type in ('M','C')")
|
|
} else {
|
|
table = e.Orm.Model(&data).Select("sys_menu.*").Joins("left join sys_role_menu on sys_role_menu.menu_id=sys_menu.menu_id")
|
|
table = table.Where("sys_role_menu.role_name=? and menu_type in ('M','C')", roleName)
|
|
}
|
|
err = table.Scopes(
|
|
cDto.OrderDest("sort", false),
|
|
).Find(&Menus).Error
|
|
|
|
if err != nil {
|
|
e.Log.Errorf("db error:%s", err)
|
|
return
|
|
}
|
|
return
|
|
}
|