58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package dao
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"hospital-admin-api/api/model"
|
|
"hospital-admin-api/global"
|
|
)
|
|
|
|
type AdminMenuApiDao struct {
|
|
}
|
|
|
|
// GetAdminMenuApiListByMenuID 获取菜单api-菜单id
|
|
// menuId 菜单id
|
|
func (r *AdminMenuApiDao) GetAdminMenuApiListByMenuID(menuId int64) (m []*model.AdminMenuApi, err error) {
|
|
err = global.Db.Where("menu_id = ?", menuId).Find(&m).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// GetAdminMenuApiListWithAPIByMenuID 获取菜单api，关联api模型-菜单id
|
|
func (r *AdminMenuApiDao) GetAdminMenuApiListWithAPIByMenuID(menuId int64) (m []*model.AdminMenuApi, err error) {
|
|
err = global.Db.Where("menu_id = ?", menuId).Preload("API").Find(&m).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// AddAdminMenuApi 新增菜单api
|
|
func (r *AdminMenuApiDao) AddAdminMenuApi(tx *gorm.DB, model *model.AdminMenuApi) (*model.AdminMenuApi, error) {
|
|
if err := tx.Create(model).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return model, nil
|
|
}
|
|
|
|
// DeleteAdminMenuApiByMenuId 删除菜单关联api-菜单id
|
|
func (r *AdminMenuApiDao) DeleteAdminMenuApiByMenuId(tx *gorm.DB, menuId int64) error {
|
|
if err := tx.Where("menu_id = ?", menuId).Delete(&model.AdminMenuApi{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetAdminMenuApiApiIdByMenuId 获取菜单接口id-菜单id
|
|
func (r *AdminMenuApiDao) GetAdminMenuApiApiIdByMenuId(menuId int64) (m []int64, err error) {
|
|
err = global.Db.Model(&model.AdminRoleMenu{}).
|
|
Where("menu_id = ?", menuId).
|
|
Pluck("api_id", &m).
|
|
Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|