You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.5 KiB
68 lines
2.5 KiB
package model
|
|
|
|
import (
|
|
"gin-vue-admin/global"
|
|
)
|
|
|
|
type SysMenu struct {
|
|
SysBaseMenu
|
|
MenuId string `json:"MenuId"`
|
|
AuthorityId string `json:"-"`
|
|
Children []SysMenu `json:"children"`
|
|
}
|
|
|
|
// @title AddMenuAuthority
|
|
// @description 为角色增加menu树
|
|
// @auth (2020/04/05 20:22 )
|
|
// @param menus []SysBaseMenu
|
|
// @param authorityId string
|
|
// @return error
|
|
func (m *SysMenu) AddMenuAuthority(menus []SysBaseMenu, authorityId string) (err error) {
|
|
var auth SysAuthority
|
|
auth.AuthorityId = authorityId
|
|
auth.SysBaseMenus = menus
|
|
err = auth.SetMuneAuthority()
|
|
return err
|
|
}
|
|
|
|
// @title GetMenuAuthority
|
|
// @description 查看当前角色树
|
|
// @auth (2020/04/05 20:22 )
|
|
// @param authorityId string
|
|
// @return err error
|
|
// @return menus []SysBaseMenu
|
|
func (m *SysMenu) GetMenuAuthority(authorityId string) (err error, menus []SysBaseMenu) {
|
|
var a SysAuthority
|
|
err = global.GVA_DB.Preload("SysBaseMenus").Where("authority_id = ?", authorityId).First(&a).Error
|
|
return err, a.SysBaseMenus
|
|
}
|
|
|
|
// @title GetMenuTree
|
|
// @description 获取动态菜单树
|
|
// @auth (2020/04/05 20:22 )
|
|
// @param authorityId string
|
|
// @return err error
|
|
// @return menus []SysMenu
|
|
func (m *SysMenu) GetMenuTree(authorityId string) (err error, menus []SysMenu) {
|
|
SQLstatement := "SELECT authority_menu.created_at,authority_menu.updated_at,authority_menu.deleted_at,authority_menu.menu_level,authority_menu.parent_id,authority_menu.path,authority_menu.`name`,authority_menu.hidden,authority_menu.component,authority_menu.title,authority_menu.icon,authority_menu.sort,authority_menu.menu_id,authority_menu.authority_id FROM authority_menu WHERE authority_menu.authority_id = ? AND authority_menu.parent_id = ?"
|
|
|
|
err = global.GVA_DB.Raw(SQLstatement, authorityId, 0).Scan(&menus).Error
|
|
for i := 0; i < len(menus); i++ {
|
|
err = getChildrenList(&menus[i], SQLstatement)
|
|
}
|
|
return err, menus
|
|
}
|
|
|
|
// @title getChildrenList
|
|
// @description 获取子菜单
|
|
// @auth (2020/04/05 20:22 )
|
|
// @param menu *SysMenu
|
|
// @param SQLstatement string
|
|
// @return err error
|
|
func getChildrenList(menu *SysMenu, SQLstatement string) (err error) {
|
|
err = global.GVA_DB.Raw(SQLstatement, menu.AuthorityId, menu.MenuId).Scan(&menu.Children).Error
|
|
for i := 0; i < len(menu.Children); i++ {
|
|
err = getChildrenList(&menu.Children[i], SQLstatement)
|
|
}
|
|
return err
|
|
}
|