51 lines
863 B
Go
51 lines
863 B
Go
package mysql
|
|
|
|
import (
|
|
"myschools.me/heritage/heritage-api/model"
|
|
)
|
|
|
|
func MenuList() ([]model.Menu, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var menus []model.Menu
|
|
if err := db.Order("sort ASC").Find(&menus).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return menus, nil
|
|
}
|
|
|
|
func MenuCreate(obj *model.Menu) (*model.Menu, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := db.Create(obj).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return obj, nil
|
|
}
|
|
|
|
func MenuCount() (*int64, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var count int64
|
|
if err := db.Model(&model.Menu{}).Count(&count).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &count, nil
|
|
}
|
|
|
|
func MenuCreateBatch(menus []model.Menu) error {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return db.Create(&menus).Error
|
|
}
|