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.
78 lines
2.1 KiB
78 lines
2.1 KiB
package dbModel
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/pkg/errors"
|
|
"main/controller/servers"
|
|
"main/init/qmsql"
|
|
"main/model/modelInterface"
|
|
)
|
|
|
|
type BaseMenu struct {
|
|
gorm.Model
|
|
MenuLevel uint `json:"-"`
|
|
ParentId string `json:"parentId"`
|
|
Path string `json:"path"`
|
|
Name string `json:"name"`
|
|
Hidden bool `json:"hidden"`
|
|
Component string `json:"component"`
|
|
Meta `json:"meta"`
|
|
NickName string `json:"nickName"`
|
|
Children []BaseMenu `json:"children"`
|
|
}
|
|
|
|
func (b *BaseMenu) AddBaseMenu() (err error) {
|
|
findOne := qmsql.DEFAULTDB.Where("name = ?", b.Name).Find(&BaseMenu{}).Error
|
|
if findOne != nil {
|
|
b.NickName = b.Title
|
|
err = qmsql.DEFAULTDB.Create(b).Error
|
|
} else {
|
|
err = errors.New("存在重复name，请修改name")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (b *BaseMenu) DeleteBaseMenu(id float64) (err error) {
|
|
err = qmsql.DEFAULTDB.Where("parent_id = ?",id).First(&BaseMenu{}).Error
|
|
if(err!=nil){
|
|
err = qmsql.DEFAULTDB.Where("id = ?", id).Delete(&b).Error
|
|
err = qmsql.DEFAULTDB.Where("menu_id = ?", id).Delete(&Menu{}).Error
|
|
}else{
|
|
return errors.New("此菜单存在子菜单不可删除")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (b *BaseMenu) UpdataBaseMenu(name string) (err error) {
|
|
err = qmsql.DEFAULTDB.Where("name = ?", name).Update(&b).Update(&Menu{}).Error
|
|
return err
|
|
}
|
|
|
|
func (b *BaseMenu) GetInfoList(info modelInterface.PageInfo) (err error, list interface{}, total int) {
|
|
// 封装分页方法 调用即可 传入 当前的结构体和分页信息
|
|
err, db, total := servers.PagingServer(b, info)
|
|
if err != nil {
|
|
return
|
|
} else {
|
|
var menuList []BaseMenu
|
|
err = db.Find(&menuList).Error
|
|
return err, menuList, total
|
|
}
|
|
}
|
|
|
|
//获取基础路由树
|
|
func (m *BaseMenu) GetBaseMenuTree() (err error, menus []BaseMenu) {
|
|
err = qmsql.DEFAULTDB.Where(" parent_id = ?", 0).Find(&menus).Error
|
|
for i := 0; i < len(menus); i++ {
|
|
err = getBaseChildrenList(&menus[i])
|
|
}
|
|
return err, menus
|
|
}
|
|
|
|
func getBaseChildrenList(menu *BaseMenu) (err error) {
|
|
err = qmsql.DEFAULTDB.Where("parent_id = ?", menu.ID).Find(&menu.Children).Error
|
|
for i := 0; i < len(menu.Children); i++ {
|
|
err = getBaseChildrenList(&menu.Children[i])
|
|
}
|
|
return err
|
|
}
|