90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/beego/beego/v2/client/orm"
|
|
)
|
|
|
|
// Menu 菜单模型
|
|
type Menu struct {
|
|
Id int `orm:"auto"`
|
|
Name string `orm:"size(100)"`
|
|
Path string `orm:"size(255)"`
|
|
ParentId int `orm:"default(0)"`
|
|
Icon string `orm:"size(100)"`
|
|
Order int `orm:"default(0)"`
|
|
Status int8 `orm:"default(1)"`
|
|
ComponentPath string `orm:"size(500);null"`
|
|
IsExternal int8 `orm:"default(0)"`
|
|
ExternalUrl string `orm:"size(1000);null"`
|
|
MenuType int8 `orm:"default(1)"`
|
|
Permission string `orm:"size(200);null"`
|
|
CreateTime time.Time `orm:"auto_now_add;type(datetime)"`
|
|
UpdateTime time.Time `orm:"auto_now;type(datetime)"`
|
|
}
|
|
|
|
// TableName 设置表名
|
|
func (m *Menu) TableName() string {
|
|
return "yz_menus"
|
|
}
|
|
|
|
// GetAllMenus 获取所有菜单
|
|
func GetAllMenus() ([]map[string]interface{}, error) {
|
|
o := orm.NewOrm()
|
|
var menus []*Menu
|
|
_, err := o.QueryTable("yz_menus").Filter("Status", 1).All(&menus)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result := make([]map[string]interface{}, 0, len(menus))
|
|
for _, m := range menus {
|
|
item := map[string]interface{}{
|
|
"id": m.Id,
|
|
"name": m.Name,
|
|
"path": m.Path,
|
|
"parentId": m.ParentId,
|
|
"icon": m.Icon,
|
|
"order": m.Order,
|
|
"componentPath": m.ComponentPath,
|
|
"isExternal": m.IsExternal,
|
|
"externalUrl": m.ExternalUrl,
|
|
}
|
|
result = append(result, item)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// AddMenu 添加新菜单
|
|
func AddMenu(menu *Menu) (int64, error) {
|
|
o := orm.NewOrm()
|
|
id, err := o.Insert(menu)
|
|
return id, err
|
|
}
|
|
|
|
// UpdateMenu 更新菜单
|
|
func UpdateMenu(menu *Menu) error {
|
|
o := orm.NewOrm()
|
|
_, err := o.Update(menu)
|
|
return err
|
|
}
|
|
|
|
// UpdateMenuStatus 更新菜单状态
|
|
func UpdateMenuStatus(id int, status int8) error {
|
|
o := orm.NewOrm()
|
|
menu := Menu{Id: id}
|
|
if err := o.Read(&menu); err != nil {
|
|
return err
|
|
}
|
|
menu.Status = status
|
|
_, err := o.Update(&menu, "Status")
|
|
return err
|
|
}
|
|
|
|
// DeleteMenu 删除菜单
|
|
func DeleteMenu(id int) error {
|
|
o := orm.NewOrm()
|
|
_, err := o.Delete(&Menu{Id: id})
|
|
return err
|
|
}
|