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.
40 lines
1.0 KiB
40 lines
1.0 KiB
package dbModel
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
"main/init/qmsql"
|
|
)
|
|
|
|
type Menu struct {
|
|
gorm.Model `json:"-"`
|
|
MenuLevel uint `json:"-"`
|
|
AuthorityId uint `json:"-"`
|
|
ParentId uint `json:"parentId"`
|
|
Path string `json:"path"`
|
|
Name string `json:"name"`
|
|
Hidden bool `json:"hidden"`
|
|
Component string `json:"component"`
|
|
Meta `json:"meta"`
|
|
Children []Menu `json:"children"`
|
|
}
|
|
|
|
type Meta struct {
|
|
Title string `json:"title"`
|
|
Icon string `json:"icon"`
|
|
}
|
|
|
|
func (m *Menu) GetMenuTree(authorityId float64) (err error, menus []Menu) {
|
|
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", authorityId, 0).Find(&menus).Error
|
|
for i := 0; i < len(menus); i++ {
|
|
err = getChildrenList(&menus[i])
|
|
}
|
|
return err, menus
|
|
}
|
|
|
|
func getChildrenList(menu *Menu) (err error) {
|
|
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", menu.AuthorityId, menu.ID).Find(&menu.Children).Error
|
|
for i := 0; i < len(menu.Children); i++ {
|
|
err = getChildrenList(&menu.Children[i])
|
|
}
|
|
return err
|
|
}
|