|
@@ -0,0 +1,669 @@
|
|
|
+package service
|
|
|
+
|
|
|
+import (
|
|
|
+ "cold-logistics/common/actions"
|
|
|
+ dto2 "cold-logistics/common/dto"
|
|
|
+ "cold-logistics/common/global"
|
|
|
+ cModel "cold-logistics/common/model"
|
|
|
+ "encoding/json"
|
|
|
+ "fmt"
|
|
|
+ "gogs.baozhida.cn/zoie/OAuth-core/service"
|
|
|
+ "strings"
|
|
|
+
|
|
|
+ "github.com/pkg/errors"
|
|
|
+ "gorm.io/gorm"
|
|
|
+
|
|
|
+ "cold-logistics/app/admin/model"
|
|
|
+ "cold-logistics/app/admin/service/dto"
|
|
|
+ "gogs.baozhida.cn/zoie/OAuth-core/pkg"
|
|
|
+)
|
|
|
+
|
|
|
+type SysMenu struct {
|
|
|
+ service.Service
|
|
|
+}
|
|
|
+
|
|
|
+// GetPage 获取SysMenu列表
|
|
|
+func (e *SysMenu) GetPage(c *dto.SysMenuGetPageReq, menus *[]model.SysMenu) *SysMenu {
|
|
|
+ var menu = make([]model.SysMenu, 0)
|
|
|
+ err := e.getPage(c, &menu).Error
|
|
|
+ if err != nil {
|
|
|
+ _ = e.AddError(global.GetFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ for i := 0; i < len(menu); i++ {
|
|
|
+ if menu[i].ParentId != 0 {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ menusInfo := SysMenuCall(&menu, menu[i])
|
|
|
+ *menus = append(*menus, menusInfo)
|
|
|
+ }
|
|
|
+
|
|
|
+ if len(*menus) == 0 {
|
|
|
+ parentMap, flag := getSysMenuParent(&menu)
|
|
|
+ if flag {
|
|
|
+ for i := 0; i < len(menu); i++ {
|
|
|
+ if parentMap[menu[i].ParentId] {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ menusInfo := SysMenuCall(&menu, menu[i])
|
|
|
+ *menus = append(*menus, menusInfo)
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ for i := 0; i < len(menu); i++ {
|
|
|
+ menusInfo := SysMenuCall(&menu, menu[i])
|
|
|
+ *menus = append(*menus, menusInfo)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+// 获取搜索出来的菜单的父菜单
|
|
|
+func getSysMenuParent(menuList *[]model.SysMenu) (map[int]bool, bool) {
|
|
|
+ list := *menuList
|
|
|
+ var flag = false
|
|
|
+ var parentMap = map[int]bool{}
|
|
|
+ for j := 0; j < len(list); j++ {
|
|
|
+ parentMap[list[j].ParentId] = false
|
|
|
+ }
|
|
|
+
|
|
|
+ for j := 0; j < len(list); j++ {
|
|
|
+ if _, ok := parentMap[list[j].Id]; !ok {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ parentMap[list[j].Id] = true
|
|
|
+ flag = true
|
|
|
+ }
|
|
|
+ return parentMap, flag
|
|
|
+}
|
|
|
+
|
|
|
+// getPage 菜单分页列表
|
|
|
+func (e *SysMenu) getPage(c *dto.SysMenuGetPageReq, list *[]model.SysMenu) *SysMenu {
|
|
|
+ var err error
|
|
|
+ var data model.SysMenu
|
|
|
+ var tempList []model.SysMenu
|
|
|
+ err = e.Orm.Model(&data).
|
|
|
+ Scopes(dto2.MakeCondition(c.GetNeedSearch())).
|
|
|
+ Order("id").
|
|
|
+ Find(&tempList).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("getSysMenuPage error: %s", err)
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ // 获取菜单及其子菜单下ID
|
|
|
+ for _, menu := range tempList {
|
|
|
+ var menuList = make([]model.SysMenu, 0)
|
|
|
+ err = e.Orm.Where("menu_path like ?", menu.MenuPath+"%").Order(fmt.Sprintf("sort %s", c.SortOrder)).Order("id").Find(&menuList).Error
|
|
|
+ *list = append(*list, menuList...)
|
|
|
+ }
|
|
|
+
|
|
|
+ *list = DeduplicateSysMenu(*list)
|
|
|
+
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+// Get 获取SysMenu对象
|
|
|
+func (e *SysMenu) Get(d *dto.SysMenuGetReq, menuModel *model.SysMenu) *SysMenu {
|
|
|
+ err := e.Orm.First(menuModel, d.GetId()).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
+ err = global.GetNotFoundOrNoPermissionErr
|
|
|
+ } else {
|
|
|
+ err = global.GetFailedErr
|
|
|
+ }
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+// Insert 创建SysMenu对象
|
|
|
+func (e *SysMenu) Insert(c *dto.SysMenuInsertReq) *SysMenu {
|
|
|
+ var err error
|
|
|
+ var data model.SysMenu
|
|
|
+
|
|
|
+ tx := e.Orm.Begin()
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ } else {
|
|
|
+ tx.Commit()
|
|
|
+ }
|
|
|
+ }()
|
|
|
+
|
|
|
+ c.Generate(&data)
|
|
|
+ err = tx.Create(&data).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(global.CreateFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ c.Id = data.Id
|
|
|
+
|
|
|
+ menuPath, err := e.initPaths(&data)
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ err = tx.Model(&data).Where("id = ?", data.Id).Update("menu_path", menuPath).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(global.CreateFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ // 添加菜单后删除角色菜单缓存,防止系统菜单与服务菜单不同步
|
|
|
+ err = model.DeleteAllRoleMenuCache()
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("删除角色菜单缓存失败 %s", err)
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+func (e *SysMenu) initPaths(menu *model.SysMenu) (menuPath string, err error) {
|
|
|
+ var data model.SysMenu
|
|
|
+ parentMenu := new(model.SysMenu)
|
|
|
+ if menu.ParentId != 0 {
|
|
|
+ e.Orm.Model(&data).First(parentMenu, menu.ParentId)
|
|
|
+ if parentMenu.MenuPath == "" {
|
|
|
+ err = errors.New("父级menu_path异常,请尝试对当前节点父级菜单进行更新操作!")
|
|
|
+ return menuPath, err
|
|
|
+ }
|
|
|
+ menuPath = parentMenu.MenuPath + "/" + pkg.IntToString(menu.Id)
|
|
|
+ } else {
|
|
|
+ menuPath = "/0/" + pkg.IntToString(menu.Id)
|
|
|
+ }
|
|
|
+ return menuPath, nil
|
|
|
+}
|
|
|
+
|
|
|
+// Update 修改SysMenu对象
|
|
|
+func (e *SysMenu) Update(c *dto.SysMenuUpdateReq) *SysMenu {
|
|
|
+ var err error
|
|
|
+ tx := e.Orm.Debug().Begin()
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ } else {
|
|
|
+ tx.Commit()
|
|
|
+ }
|
|
|
+ }()
|
|
|
+ var menuModel = model.SysMenu{}
|
|
|
+ err = tx.First(&menuModel, c.GetId()).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
+ err = global.UpdateNotFoundOrNoPermissionErr
|
|
|
+ } else {
|
|
|
+ err = global.UpdateFailedErr
|
|
|
+ }
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ oldPath := menuModel.MenuPath
|
|
|
+ oldParentId := menuModel.ParentId
|
|
|
+ c.Generate(&menuModel)
|
|
|
+
|
|
|
+ // 替换旧的path
|
|
|
+ if c.ParentId != oldParentId {
|
|
|
+ menuPath, err := e.initPaths(&menuModel)
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ menuModel.MenuPath = menuPath
|
|
|
+ }
|
|
|
+ // 更新菜单信息
|
|
|
+ err = tx.Model(&menuModel).Save(&menuModel).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(global.UpdateFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ // 替换旧的path
|
|
|
+ if oldPath != menuModel.MenuPath {
|
|
|
+ var menuList []model.SysMenu
|
|
|
+ tx.Where("menu_path like ?", oldPath+"%").Find(&menuList)
|
|
|
+ for _, v := range menuList {
|
|
|
+ v.MenuPath = strings.Replace(v.MenuPath, oldPath, menuModel.MenuPath, 1)
|
|
|
+ err := tx.Model(&v).Update("menu_path", v.MenuPath).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ _ = e.AddError(global.CreateFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 修改服务菜单后删除角色菜单缓存,防止系统菜单与服务菜单不同步
|
|
|
+ err = model.DeleteAllRoleMenuCache()
|
|
|
+ if err != nil {
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+// Remove 删除SysMenu
|
|
|
+func (e *SysMenu) Remove(d *dto.SysMenuDeleteReq) *SysMenu {
|
|
|
+ var err error
|
|
|
+
|
|
|
+ tx := e.Orm.Begin()
|
|
|
+ defer func() {
|
|
|
+ if err != nil {
|
|
|
+ tx.Rollback()
|
|
|
+ } else {
|
|
|
+ tx.Commit()
|
|
|
+ }
|
|
|
+ }()
|
|
|
+
|
|
|
+ var menuModel model.SysMenu
|
|
|
+ err = tx.First(&menuModel, d.GetId()).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
+ err = global.DeleteNotFoundOrNoPermissionErr
|
|
|
+ } else {
|
|
|
+ err = global.DeleteFailedErr
|
|
|
+ }
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ err = e.Orm.Where("menu_path like ?", menuModel.MenuPath+"%").Delete(&model.SysMenu{}).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("Delete error: %s", err)
|
|
|
+ _ = e.AddError(global.DeleteFailedErr)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+
|
|
|
+ // 删除服务菜单后删除角色菜单缓存,防止系统菜单与服务菜单不同步
|
|
|
+ err = model.DeleteAllRoleMenuCache()
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("删除角色菜单缓存失败 %s", err)
|
|
|
+ _ = e.AddError(err)
|
|
|
+ return e
|
|
|
+ }
|
|
|
+ return e
|
|
|
+}
|
|
|
+
|
|
|
+// GetList 获取菜单数据
|
|
|
+func (e *SysMenu) GetList(c *dto.SysMenuGetPageReq, list *[]model.SysMenu) error {
|
|
|
+ var err error
|
|
|
+ var data model.SysMenu
|
|
|
+
|
|
|
+ err = e.Orm.Model(&data).
|
|
|
+ Scopes(
|
|
|
+ dto2.MakeCondition(c.GetNeedSearch()),
|
|
|
+ ).
|
|
|
+ Find(list).Error
|
|
|
+ if err != nil {
|
|
|
+ e.Log.Errorf("db error: %s", err)
|
|
|
+ return global.GetFailedErr
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// SysMenuCall 构建菜单树
|
|
|
+func SysMenuCall(menuList *[]model.SysMenu, menu model.SysMenu) model.SysMenu {
|
|
|
+ list := *menuList
|
|
|
+
|
|
|
+ min := make([]model.SysMenu, 0)
|
|
|
+ for j := 0; j < len(list); j++ {
|
|
|
+
|
|
|
+ if menu.Id != list[j].ParentId {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ mi := model.SysMenu{}
|
|
|
+ mi.Id = list[j].Id
|
|
|
+ mi.Title = list[j].Title
|
|
|
+ mi.Icon = list[j].Icon
|
|
|
+ mi.Path = list[j].Path
|
|
|
+ mi.MenuType = list[j].MenuType
|
|
|
+ mi.Name = list[j].Name
|
|
|
+ mi.ParentId = list[j].ParentId
|
|
|
+ mi.Component = list[j].Component
|
|
|
+ mi.Sort = list[j].Sort
|
|
|
+ mi.Visible = list[j].Visible
|
|
|
+ mi.Redirect = list[j].Redirect
|
|
|
+ mi.CreatedAt = list[j].CreatedAt
|
|
|
+ mi.Children = []model.SysMenu{}
|
|
|
+
|
|
|
+ if mi.MenuType != cModel.Button {
|
|
|
+ ms := SysMenuCall(menuList, mi)
|
|
|
+ min = append(min, ms)
|
|
|
+ } else {
|
|
|
+ min = append(min, mi)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ menu.Children = min
|
|
|
+ return menu
|
|
|
+}
|
|
|
+
|
|
|
+// DeduplicateSysMenu 菜单列表去重
|
|
|
+func DeduplicateSysMenu(list []model.SysMenu) []model.SysMenu {
|
|
|
+ seen := make(map[string]bool)
|
|
|
+ deduplicated := []model.SysMenu{}
|
|
|
+
|
|
|
+ for _, v := range list {
|
|
|
+ // 将结构体编码为 JSON 字符串
|
|
|
+ key, err := json.Marshal(v)
|
|
|
+ if err != nil {
|
|
|
+ // 处理错误
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ // 将 JSON 字符串作为键插入到 map 中
|
|
|
+ if _, ok := seen[string(key)]; !ok {
|
|
|
+ seen[string(key)] = true
|
|
|
+ deduplicated = append(deduplicated, v)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return deduplicated
|
|
|
+}
|
|
|
+
|
|
|
+// menuLabelCall 递归构造组织数据
|
|
|
+func menuLabelCall(eList *[]model.SysMenu, organ dto.MenuLabel) dto.MenuLabel {
|
|
|
+ list := *eList
|
|
|
+
|
|
|
+ min := make([]dto.MenuLabel, 0)
|
|
|
+ for j := 0; j < len(list); j++ {
|
|
|
+
|
|
|
+ if organ.Id != list[j].ParentId {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ mi := dto.MenuLabel{}
|
|
|
+ mi.Id = list[j].Id
|
|
|
+ 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 {
|
|
|
+ organ.Children = min
|
|
|
+ } else {
|
|
|
+ organ.Children = nil
|
|
|
+ }
|
|
|
+ return organ
|
|
|
+}
|
|
|
+
|
|
|
+// SetLabel 修改角色中 设置菜单基础数据
|
|
|
+func (e *SysMenu) SetLabel() (m []dto.MenuLabel, err error) {
|
|
|
+ var list []model.SysMenu
|
|
|
+ err = e.GetList(&dto.SysMenuGetPageReq{}, &list)
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ 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].Id
|
|
|
+ e.Label = list[i].Title
|
|
|
+ menuInfo := menuLabelCall(&list, e)
|
|
|
+
|
|
|
+ m = append(m, menuInfo)
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// menuCall 构建菜单树
|
|
|
+func menuCall(menuList *[]model.SysMenu, menu model.SysMenu) model.SysMenu {
|
|
|
+ list := *menuList
|
|
|
+
|
|
|
+ min := make([]model.SysMenu, 0)
|
|
|
+ for j := 0; j < len(list); j++ {
|
|
|
+
|
|
|
+ if menu.Id != list[j].ParentId {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ mi := model.SysMenu{}
|
|
|
+ mi.Id = list[j].Id
|
|
|
+ mi.Title = list[j].Title
|
|
|
+ mi.Icon = list[j].Icon
|
|
|
+ mi.Path = list[j].Path
|
|
|
+ mi.MenuType = list[j].MenuType
|
|
|
+ mi.Name = list[j].Name
|
|
|
+ mi.ParentId = list[j].ParentId
|
|
|
+ mi.Component = list[j].Component
|
|
|
+ mi.Sort = list[j].Sort
|
|
|
+ mi.Visible = list[j].Visible
|
|
|
+ mi.CreatedAt = list[j].CreatedAt
|
|
|
+ mi.Children = []model.SysMenu{}
|
|
|
+ ms := menuCall(menuList, mi)
|
|
|
+ min = append(min, ms)
|
|
|
+ }
|
|
|
+ menu.Children = min
|
|
|
+ return menu
|
|
|
+}
|
|
|
+
|
|
|
+// 通过roleName获取菜单列表
|
|
|
+func (e *SysMenu) getByRoleName(roleName string) ([]model.SysMenu, error) {
|
|
|
+ var err error
|
|
|
+
|
|
|
+ data := make([]model.SysMenu, 0)
|
|
|
+
|
|
|
+ if roleName == "admin" {
|
|
|
+ err = e.Orm.Find(&data).Error
|
|
|
+ } else {
|
|
|
+ roleMenuIds := make([]int, 0)
|
|
|
+ err = e.Orm.Model(&model.SysRoleMenu{}).Select("menu_id").Where("role_key = ? ", roleName).Scan(&roleMenuIds).Error
|
|
|
+
|
|
|
+ if len(roleMenuIds) > 0 {
|
|
|
+ if err = recursiveSetMenu(e.Orm, roleMenuIds, &data); err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ // 菜单id及父id 去重
|
|
|
+ data = menuDistinct(data)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return data, err
|
|
|
+}
|
|
|
+func recursiveSetMenu(orm *gorm.DB, mIds []int, menus *[]model.SysMenu) error {
|
|
|
+ if len(mIds) == 0 || menus == nil {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ var subMenus []model.SysMenu
|
|
|
+ err := orm.Where("id in ?", mIds).Find(&subMenus).Error
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ subIds := make([]int, 0)
|
|
|
+ for _, menu := range subMenus {
|
|
|
+ if menu.ParentId != 0 {
|
|
|
+ subIds = append(subIds, menu.ParentId)
|
|
|
+ }
|
|
|
+ //if menu.MenuType != cModel.Button {
|
|
|
+ // *menus = append(*menus, menu)
|
|
|
+ //}
|
|
|
+ *menus = append(*menus, menu)
|
|
|
+ }
|
|
|
+ return recursiveSetMenu(orm, subIds, menus)
|
|
|
+}
|
|
|
+
|
|
|
+// 去掉重复菜单
|
|
|
+func menuDistinct(menuList []model.SysMenu) (result []model.SysMenu) {
|
|
|
+ distinctMap := make(map[int]struct{}, len(menuList))
|
|
|
+ for _, menu := range menuList {
|
|
|
+ if _, ok := distinctMap[menu.Id]; !ok {
|
|
|
+ distinctMap[menu.Id] = struct{}{}
|
|
|
+ result = append(result, menu)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return result
|
|
|
+}
|
|
|
+
|
|
|
+// SetMenuRole 获取左侧菜单树使用
|
|
|
+func (e *SysMenu) SetMenuRole(roleKey string, userId int, p *actions.DataPermission) (m []model.SysMenu, err error) {
|
|
|
+ // 从缓存获取角色关联的菜单
|
|
|
+ var cacheKey string
|
|
|
+ if p.DeptId > 0 {
|
|
|
+ //cacheKey = fmt.Sprintf("%s-1-%d", roleKey, userId)
|
|
|
+ cacheKey = fmt.Sprintf("%s-1", roleKey)
|
|
|
+ } else {
|
|
|
+ //cacheKey = fmt.Sprintf("%s-0-%d", roleKey, userId)
|
|
|
+ cacheKey = fmt.Sprintf("%s-0", roleKey)
|
|
|
+ }
|
|
|
+ if err := model.GetRoleMenuCache(cacheKey, &m); err == nil {
|
|
|
+ return m, nil
|
|
|
+ }
|
|
|
+ menus, err := e.getByRoleName(roleKey)
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ m = make([]model.SysMenu, 0)
|
|
|
+
|
|
|
+ var menuList []model.SysMenu
|
|
|
+
|
|
|
+ if p.DeptId == 0 {
|
|
|
+ if roleKey == "admin" {
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.visible = 1 and sys_menu.deleted_at is null and is_dept = 0").
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ menuIds := make([]int, 0)
|
|
|
+ for _, menu := range menus {
|
|
|
+ menuIds = append(menuIds, menu.Id)
|
|
|
+ }
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.id in (?) and sys_menu.visible = 1 and sys_menu.deleted_at is null and is_dept = 0", menuIds).
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+
|
|
|
+ if roleKey == "admin" {
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.visible = 1 and sys_menu.deleted_at is null").
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ menuIds := make([]int, 0)
|
|
|
+ for _, menu := range menus {
|
|
|
+ menuIds = append(menuIds, menu.Id)
|
|
|
+ }
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.id in (?) and sys_menu.visible = 1 and sys_menu.deleted_at is null", menuIds).
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ //sort.Sort(model.SysMenuSlice(menuList))
|
|
|
+ for i := 0; i < len(menuList); i++ {
|
|
|
+ if menuList[i].ParentId != 0 {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ menusInfo := menuCall(&menuList, menuList[i])
|
|
|
+ m = append(m, menusInfo)
|
|
|
+ }
|
|
|
+ // 保存角色菜单到缓存
|
|
|
+ _ = model.SetRoleMenuCache(cacheKey, m)
|
|
|
+ return
|
|
|
+}
|
|
|
+func (e *SysMenu) MenuSelect(roleKey string, userId int, p *actions.DataPermission) (m []model.SysMenu, err error) {
|
|
|
+
|
|
|
+ menus, err := e.getByRoleName(roleKey)
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ m = make([]model.SysMenu, 0)
|
|
|
+
|
|
|
+ var menuList []model.SysMenu
|
|
|
+
|
|
|
+ if p.DeptId == 0 {
|
|
|
+ if roleKey == "admin" {
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.visible = 1 and sys_menu.deleted_at is null").
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ menuIds := make([]int, 0)
|
|
|
+ for _, menu := range menus {
|
|
|
+ menuIds = append(menuIds, menu.Id)
|
|
|
+ }
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.id in (?) and sys_menu.visible = 1 and sys_menu.deleted_at is null", menuIds).
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+
|
|
|
+ if roleKey == "admin" {
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.visible = 1 and sys_menu.deleted_at is null").
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ menuIds := make([]int, 0)
|
|
|
+ for _, menu := range menus {
|
|
|
+ menuIds = append(menuIds, menu.Id)
|
|
|
+ }
|
|
|
+ err = e.Orm.Table("sys_menu").
|
|
|
+ Where("sys_menu.id in (?) and sys_menu.visible = 1 and sys_menu.deleted_at is null", menuIds).
|
|
|
+ Order("sys_menu.sort").
|
|
|
+ Order("id").
|
|
|
+ Scan(&menuList).Error
|
|
|
+ if err != nil {
|
|
|
+ return m, global.GetFailedErr
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ //sort.Sort(model.SysMenuSlice(menuList))
|
|
|
+ for i := 0; i < len(menuList); i++ {
|
|
|
+ if menuList[i].ParentId != 0 {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ menusInfo := menuCall(&menuList, menuList[i])
|
|
|
+ m = append(m, menusInfo)
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|