@@ -2,7 +2,6 @@ package system
2
2
3
3
import (
4
4
"errors"
5
- "strconv"
6
5
7
6
"github.com/flipped-aurora/gin-vue-admin/server/global"
8
7
"github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
@@ -20,11 +19,11 @@ type MenuService struct{}
20
19
21
20
var MenuServiceApp = new (MenuService )
22
21
23
- func (menuService * MenuService ) getMenuTreeMap (authorityId uint ) (treeMap map [string ][]system.SysMenu , err error ) {
22
+ func (menuService * MenuService ) getMenuTreeMap (authorityId uint ) (treeMap map [uint ][]system.SysMenu , err error ) {
24
23
var allMenus []system.SysMenu
25
24
var baseMenu []system.SysBaseMenu
26
25
var btns []system.SysAuthorityBtn
27
- treeMap = make (map [string ][]system.SysMenu )
26
+ treeMap = make (map [uint ][]system.SysMenu )
28
27
29
28
var SysAuthorityMenus []system.SysAuthorityMenu
30
29
err = global .GVA_DB .Where ("sys_authority_authority_id = ?" , authorityId ).Find (& SysAuthorityMenus ).Error
@@ -47,7 +46,7 @@ func (menuService *MenuService) getMenuTreeMap(authorityId uint) (treeMap map[st
47
46
allMenus = append (allMenus , system.SysMenu {
48
47
SysBaseMenu : baseMenu [i ],
49
48
AuthorityId : authorityId ,
50
- MenuId : strconv . Itoa ( int ( baseMenu [i ].ID )) ,
49
+ MenuId : baseMenu [i ].ID ,
51
50
Parameters : baseMenu [i ].Parameters ,
52
51
})
53
52
}
@@ -78,7 +77,7 @@ func (menuService *MenuService) getMenuTreeMap(authorityId uint) (treeMap map[st
78
77
79
78
func (menuService * MenuService ) GetMenuTree (authorityId uint ) (menus []system.SysMenu , err error ) {
80
79
menuTree , err := menuService .getMenuTreeMap (authorityId )
81
- menus = menuTree ["0" ]
80
+ menus = menuTree [0 ]
82
81
for i := 0 ; i < len (menus ); i ++ {
83
82
err = menuService .getChildrenList (& menus [i ], menuTree )
84
83
}
@@ -91,7 +90,7 @@ func (menuService *MenuService) GetMenuTree(authorityId uint) (menus []system.Sy
91
90
//@param: menu *model.SysMenu, treeMap map[string][]model.SysMenu
92
91
//@return: err error
93
92
94
- func (menuService * MenuService ) getChildrenList (menu * system.SysMenu , treeMap map [string ][]system.SysMenu ) (err error ) {
93
+ func (menuService * MenuService ) getChildrenList (menu * system.SysMenu , treeMap map [uint ][]system.SysMenu ) (err error ) {
95
94
menu .Children = treeMap [menu .MenuId ]
96
95
for i := 0 ; i < len (menu .Children ); i ++ {
97
96
err = menuService .getChildrenList (& menu .Children [i ], treeMap )
@@ -107,7 +106,7 @@ func (menuService *MenuService) getChildrenList(menu *system.SysMenu, treeMap ma
107
106
func (menuService * MenuService ) GetInfoList () (list interface {}, total int64 , err error ) {
108
107
var menuList []system.SysBaseMenu
109
108
treeMap , err := menuService .getBaseMenuTreeMap ()
110
- menuList = treeMap ["0" ]
109
+ menuList = treeMap [0 ]
111
110
for i := 0 ; i < len (menuList ); i ++ {
112
111
err = menuService .getBaseChildrenList (& menuList [i ], treeMap )
113
112
}
@@ -120,8 +119,8 @@ func (menuService *MenuService) GetInfoList() (list interface{}, total int64, er
120
119
//@param: menu *model.SysBaseMenu, treeMap map[string][]model.SysBaseMenu
121
120
//@return: err error
122
121
123
- func (menuService * MenuService ) getBaseChildrenList (menu * system.SysBaseMenu , treeMap map [string ][]system.SysBaseMenu ) (err error ) {
124
- menu .Children = treeMap [strconv . Itoa ( int ( menu .ID )) ]
122
+ func (menuService * MenuService ) getBaseChildrenList (menu * system.SysBaseMenu , treeMap map [uint ][]system.SysBaseMenu ) (err error ) {
123
+ menu .Children = treeMap [menu .ID ]
125
124
for i := 0 ; i < len (menu .Children ); i ++ {
126
125
err = menuService .getBaseChildrenList (& menu .Children [i ], treeMap )
127
126
}
@@ -146,9 +145,9 @@ func (menuService *MenuService) AddBaseMenu(menu system.SysBaseMenu) error {
146
145
//@description: 获取路由总树map
147
146
//@return: treeMap map[string][]system.SysBaseMenu, err error
148
147
149
- func (menuService * MenuService ) getBaseMenuTreeMap () (treeMap map [string ][]system.SysBaseMenu , err error ) {
148
+ func (menuService * MenuService ) getBaseMenuTreeMap () (treeMap map [uint ][]system.SysBaseMenu , err error ) {
150
149
var allMenus []system.SysBaseMenu
151
- treeMap = make (map [string ][]system.SysBaseMenu )
150
+ treeMap = make (map [uint ][]system.SysBaseMenu )
152
151
err = global .GVA_DB .Order ("sort" ).Preload ("MenuBtn" ).Preload ("Parameters" ).Find (& allMenus ).Error
153
152
for _ , v := range allMenus {
154
153
treeMap [v .ParentId ] = append (treeMap [v .ParentId ], v )
@@ -163,7 +162,7 @@ func (menuService *MenuService) getBaseMenuTreeMap() (treeMap map[string][]syste
163
162
164
163
func (menuService * MenuService ) GetBaseMenuTree () (menus []system.SysBaseMenu , err error ) {
165
164
treeMap , err := menuService .getBaseMenuTreeMap ()
166
- menus = treeMap ["0" ]
165
+ menus = treeMap [0 ]
167
166
for i := 0 ; i < len (menus ); i ++ {
168
167
err = menuService .getBaseChildrenList (& menus [i ], treeMap )
169
168
}
@@ -210,14 +209,15 @@ func (menuService *MenuService) GetMenuAuthority(info *request.GetAuthorityId) (
210
209
menus = append (menus , system.SysMenu {
211
210
SysBaseMenu : baseMenu [i ],
212
211
AuthorityId : info .AuthorityId ,
213
- MenuId : strconv . Itoa ( int ( baseMenu [i ].ID )) ,
212
+ MenuId : baseMenu [i ].ID ,
214
213
Parameters : baseMenu [i ].Parameters ,
215
214
})
216
215
}
217
216
return menus , err
218
217
}
219
218
220
219
// UserAuthorityDefaultRouter 用户角色默认路由检查
220
+ //
221
221
// Author [SliverHorn](https://github.com/SliverHorn)
222
222
func (menuService * MenuService ) UserAuthorityDefaultRouter (user * system.SysUser ) {
223
223
var menuIds []string
0 commit comments