-
Notifications
You must be signed in to change notification settings - Fork 0
/
menu_srvc.go
221 lines (178 loc) · 6.07 KB
/
menu_srvc.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
package service_swa
import (
"context"
"errors"
"fmt"
"gorm.io/gorm"
)
type TMenu interface {
GetRoleMenu(ctx context.Context, roleId uint) ([]SysBaseMenu, error)
GetInfoList(ctx context.Context) (list []SysBaseMenu, total int64, err error)
AddMenuRole(ctx context.Context, menus []SysBaseMenu, roleId uint) (err error)
GetBaseMenuById(_ context.Context, id int) (menu SysBaseMenu, err error)
GetAllBaseMenu(_ context.Context) (menus []SysBaseMenu, err error)
AddBaseMenu(_ context.Context, menu SysBaseMenu) error
DeleteBaseMenu(_ context.Context, id int) (err error)
UpdateBaseMenu(ctx context.Context, menu SysBaseMenu) (err error)
GetMenuRole(_ context.Context, id uint) (menus []SysBaseMenu, err error)
}
func (imp *impl) GetRoleMenu(_ context.Context, roleId uint) (menus []SysBaseMenu, err error) {
var MenuIds []string
err = imp.gormDB.Model(&Role2Menus{}).Where("role_id = ?", roleId).Pluck("menu_id", &MenuIds).Error
if err != nil {
return
}
err = imp.gormDB.Where("id in (?)", MenuIds).Order("sort").Find(&menus).Error
if err != nil {
return menus, err
}
return menus, err
}
func (imp *impl) GetInfoList(_ context.Context) (list []SysBaseMenu, total int64, err error) {
var allMenus []SysBaseMenu
err = imp.gormDB.Order("sort").Find(&allMenus).Error
return allMenus, total, err
}
func (imp *impl) AddBaseMenu(_ context.Context, menu SysBaseMenu) error {
if !errors.Is(imp.gormDB.Where("name = ?", menu.Name).First(&SysBaseMenu{}).Error, gorm.ErrRecordNotFound) {
return errors.New("存在重复name,请修改name")
}
return imp.gormDB.Create(&menu).Error
}
func (imp *impl) getBaseMenuTreeMap() (treeMap map[string][]SysBaseMenu, err error) {
var allMenus []SysBaseMenu
treeMap = make(map[string][]SysBaseMenu)
err = imp.gormDB.Order("sort").Preload("MenuBtn").Preload("Parameters").Find(&allMenus).Error
for _, v := range allMenus {
treeMap[v.ParentId] = append(treeMap[v.ParentId], v)
}
return treeMap, err
}
func (imp *impl) AddMenuRole(ctx context.Context, menus []SysBaseMenu, roleId uint) (err error) {
var auth RoleUserMenu
auth.RoleId = roleId
auth.SysBaseMenus = menus
return err
}
func (imp *impl) GetMenuRole(_ context.Context, id uint) (menus []SysBaseMenu, err error) {
var baseMenu []SysBaseMenu
var roleMenu []Role2Menus
err = imp.gormDB.Where("role_id = ?", id).Find(&roleMenu).Error
if err != nil {
return
}
var MenuIds []uint
for i := range roleMenu {
MenuIds = append(MenuIds, roleMenu[i].MenuId)
}
err = imp.gormDB.Where("id in (?) ", MenuIds).Order("sort").Find(&baseMenu).Error
return baseMenu, err
}
func (imp *impl) GetBaseMenuById(_ context.Context, id int) (menu SysBaseMenu, err error) {
err = imp.gormDB.Where("id = ?", id).First(&menu).Error
return
}
func (imp *impl) GetAllBaseMenu(_ context.Context) (menus []SysBaseMenu, err error) {
err = imp.gormDB.Order("sort").Find(&menus).Error
return
}
func (imp *impl) DeleteBaseMenu(_ context.Context, id int) (err error) {
err = imp.gormDB.Where("parent_id = ?", id).First(&SysBaseMenu{}).Error
if err != nil {
err := imp.gormDB.Delete(&SwaRoleMenu{}, "sys_base_menu_id = ?", id).Error
if err != nil {
if fmt.Sprintf("%v", err) != "record not found" {
return errors.New("删除菜单角色数据错误")
}
}
err = imp.gormDB.Delete(&SysBaseMenu{}, "id = ?", id).Error
if err != nil {
if fmt.Sprintf("%v", err) != "record not found" {
return errors.New("删除菜单数据错误")
}
}
err = imp.gormDB.Delete(&SysBaseMenuParameter{}, "sys_base_menu_id = ?", id).Error
if err != nil {
if fmt.Sprintf("%v", err) != "record not found" {
return errors.New("删除菜单参数数据错误")
}
}
err = imp.gormDB.Delete(&SysBaseMenuBtn{}, "sys_base_menu_id = ?", id).Error
if err != nil {
if fmt.Sprintf("%v", err) != "record not found" {
return errors.New("删除菜单按键数据错误")
}
}
err = imp.gormDB.Delete(&SwaRoleBtn{}, "sys_menu_id = ?", id).Error
if err != nil {
if fmt.Sprintf("%v", err) != "record not found" {
return errors.New("删除角色按键数据错误")
}
}
} else {
return errors.New("此菜单存在子菜单不可删除")
}
return nil
}
func (imp *impl) UpdateBaseMenu(ctx context.Context, menu SysBaseMenu) (err error) {
var oldMenu SysBaseMenu
upDateMap := make(map[string]interface{})
upDateMap["keep_alive"] = menu.KeepAlive
upDateMap["close_tab"] = menu.CloseTab
upDateMap["default_menu"] = menu.DefaultMenu
upDateMap["parent_id"] = menu.ParentId
upDateMap["path"] = menu.Path
upDateMap["name"] = menu.Name
upDateMap["hidden"] = menu.Hidden
upDateMap["component"] = menu.Component
upDateMap["title"] = menu.Title
upDateMap["active_name"] = menu.ActiveName
upDateMap["icon"] = menu.Icon
upDateMap["sort"] = menu.Sort
err = imp.gormDB.Transaction(func(tx *gorm.DB) error {
db := tx.Where("id = ?", menu.ID).Find(&oldMenu)
if oldMenu.Name != menu.Name {
if !errors.Is(tx.Where("id <> ? AND name = ?", menu.ID, menu.Name).First(&SysBaseMenu{}).Error, gorm.ErrRecordNotFound) {
imp.Logger(ctx).Debug("存在相同name修改失败")
return errors.New("存在相同name修改失败")
}
}
txErr := tx.Unscoped().Delete(&SysBaseMenuParameter{}, "sys_base_menu_id = ?", menu.ID).Error
if txErr != nil {
imp.Logger(ctx).Debug(txErr.Error())
return txErr
}
txErr = tx.Unscoped().Delete(&SysBaseMenuBtn{}, "sys_base_menu_id = ?", menu.ID).Error
if txErr != nil {
imp.Logger(ctx).Debug(txErr.Error())
return txErr
}
if len(menu.Parameters) > 0 {
for k := range menu.Parameters {
menu.Parameters[k].SysBaseMenuID = menu.ID
}
txErr = tx.Create(&menu.Parameters).Error
if txErr != nil {
imp.Logger(ctx).Debug(txErr.Error())
return txErr
}
}
if len(menu.MenuBtn) > 0 {
for k := range menu.MenuBtn {
menu.MenuBtn[k].SysBaseMenuID = menu.ID
}
txErr = tx.Create(&menu.MenuBtn).Error
if txErr != nil {
imp.Logger(ctx).Debug(txErr.Error())
return txErr
}
}
txErr = db.Updates(upDateMap).Error
if txErr != nil {
imp.Logger(ctx).Debug(txErr.Error())
return txErr
}
return nil
})
return err
}