Blizzard
127 lines
3.7 KiB
Go
127 lines
3.7 KiB
Go
package system
|
|
|
|
import (
|
|
"errors"
|
|
"sundynix-go/global"
|
|
"sundynix-go/model/system"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type MenuService struct{}
|
|
|
|
var MenuServiceApp = new(MenuService)
|
|
|
|
func (s *MenuService) SaveMenu(menu system.Menu) error {
|
|
//1.根据code和name查询是否存在重名
|
|
if err := global.DB.Where("code = ? or name = ?", menu.Code, menu.Name).First(&system.Menu{}).Error; err == nil {
|
|
return errors.New("菜单已存在")
|
|
}
|
|
return global.DB.Create(&menu).Error
|
|
}
|
|
|
|
func (s *MenuService) UpdateMenu(menu *system.Menu) (err error) {
|
|
var sysMenu system.Menu
|
|
menuMap := map[string]interface{}{
|
|
"Category": menu.Category,
|
|
"Name": menu.Name,
|
|
"Title": menu.Title,
|
|
"Code": menu.Code,
|
|
"Permission": menu.Permission,
|
|
"Locale": menu.Locale,
|
|
"Icon": menu.Icon,
|
|
"Sort": menu.Sort,
|
|
}
|
|
err = global.DB.Where("id = ?", menu.Id).First(&sysMenu).Error
|
|
if err != nil {
|
|
global.Logger.Debug(err.Error())
|
|
return errors.New("查询菜单失败")
|
|
}
|
|
err = global.DB.Model(&sysMenu).Updates(menuMap).Error
|
|
return err
|
|
}
|
|
|
|
func (s *MenuService) DeleteMenu(id string) (err error) {
|
|
err = global.DB.First(&system.Menu{}, "parent_id = ?", id).Error
|
|
if err == nil {
|
|
return errors.New("请先删除子菜单")
|
|
}
|
|
var menu system.Menu
|
|
err = global.DB.Where("id = ?", id).First(&menu).Error
|
|
if err != nil {
|
|
return errors.New("菜单记录不存在")
|
|
}
|
|
// 同步删除menu表和role-menu表数据
|
|
return global.DB.Transaction(func(tx *gorm.DB) error {
|
|
if err = tx.Where("id = ?", id).Delete(&system.Menu{}).Error; err != nil {
|
|
return err
|
|
}
|
|
if err = tx.Where("menu_id = ?", id).Delete(&system.RoleMenu{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (s *MenuService) GetMenuById(id string) (menu system.Menu, err error) {
|
|
var m system.Menu
|
|
err = global.DB.Where("id = ?", id).First(&m).Error
|
|
return m, err
|
|
}
|
|
|
|
func (s *MenuService) GetAllMenuTree(category int, parentId string) (menus []*system.Menu, err error) {
|
|
//1,先根据category和parentId获取所有菜单 category默认为0,parentId默认为0
|
|
//2.讲查询出的列表构建为树结构
|
|
var menuList []*system.Menu
|
|
db := global.DB.Model(&system.Menu{})
|
|
if category != 0 {
|
|
db.Where("category = ?", category)
|
|
}
|
|
if parentId != "0" {
|
|
db.Where("parent_id = ?", parentId)
|
|
}
|
|
err = db.Order("sort asc").Find(&menuList).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
tree := buildMenuTree(menuList)
|
|
return tree, nil
|
|
}
|
|
|
|
func (s *MenuService) GetUserRoutes(userId string) (menus []*system.Menu, err error) {
|
|
//1.根据userId 查询角色 根据角色查询菜单 去重
|
|
//2.构建树结构
|
|
var roleIds []string
|
|
err = global.DB.Model(&system.UserRole{}).Where("user_id = ?", userId).Pluck("role_id", &roleIds).Error
|
|
var menuIds []string
|
|
err = global.DB.Model(&system.RoleMenu{}).Where("role_id in ?", roleIds).Pluck("menu_id", &menuIds).Error
|
|
var menuList []*system.Menu
|
|
err = global.DB.Model(&system.Menu{}).Where("id in ?", menuIds).Order("sort asc").Find(&menuList).Error
|
|
return buildMenuTree(menuList), nil
|
|
}
|
|
|
|
func buildMenuTree(list []*system.Menu) []*system.Menu {
|
|
//1.定义一个map
|
|
menuMap := make(map[string]*system.Menu)
|
|
for _, item := range list {
|
|
menuMap[item.Id] = item
|
|
}
|
|
//构建树结构
|
|
var treeList []*system.Menu
|
|
for _, item := range list {
|
|
if item.ParentId == "0" {
|
|
// 如果没有父节点，直接添加到树中
|
|
treeList = append(treeList, item)
|
|
} else {
|
|
if parent, exists := menuMap[item.ParentId]; exists {
|
|
// 如果有父节点，将当前节点添加到父节点的Children中
|
|
parent.Children = append(parent.Children, item)
|
|
} else {
|
|
// 如果没有父节点，将当前节点添加到树中
|
|
treeList = append(treeList, item)
|
|
}
|
|
}
|
|
}
|
|
return treeList
|
|
}
|