mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
package services
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/PandaXGO/PandaKit/casbin"
|
|
"pandax/apps/system/entity"
|
|
"pandax/pkg/global"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type (
|
|
SysApiModel interface {
|
|
Insert(data entity.SysApi) (*entity.SysApi, error)
|
|
FindOne(id int64) (*entity.SysApi, error)
|
|
FindListPage(page, pageSize int, data entity.SysApi) (*[]entity.SysApi, int64, error)
|
|
FindList(data entity.SysApi) (*[]entity.SysApi, error)
|
|
Update(data entity.SysApi) error
|
|
Delete(ids []int64) error
|
|
}
|
|
|
|
sysApiModelImpl struct {
|
|
table string
|
|
}
|
|
)
|
|
|
|
var SysApiModelDao SysApiModel = &sysApiModelImpl{
|
|
table: `sys_apis`,
|
|
}
|
|
|
|
func (m *sysApiModelImpl) Insert(api entity.SysApi) (*entity.SysApi, error) {
|
|
err := global.Db.Table(m.table).Where("path = ? AND method = ?", api.Path, api.Method).First(&entity.SysApi{}).Error
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, errors.New("存在相同api")
|
|
}
|
|
err = global.Db.Table(m.table).Create(&api).Error
|
|
return &api, err
|
|
}
|
|
|
|
func (m *sysApiModelImpl) FindOne(id int64) (resData *entity.SysApi, err error) {
|
|
resData = new(entity.SysApi)
|
|
err = global.Db.Table(m.table).Where("id = ?", id).First(&resData).Error
|
|
return
|
|
}
|
|
|
|
func (m *sysApiModelImpl) FindListPage(page, pageSize int, data entity.SysApi) (*[]entity.SysApi, int64, error) {
|
|
list := make([]entity.SysApi, 0)
|
|
var total int64 = 0
|
|
offset := pageSize * (page - 1)
|
|
|
|
db := global.Db.Table(m.table)
|
|
|
|
if data.Path != "" {
|
|
db = db.Where("path LIKE ?", "%"+data.Path+"%")
|
|
}
|
|
|
|
if data.Description != "" {
|
|
db = db.Where("description LIKE ?", "%"+data.Description+"%")
|
|
}
|
|
|
|
if data.Method != "" {
|
|
db = db.Where("method = ?", data.Method)
|
|
}
|
|
|
|
if data.ApiGroup != "" {
|
|
db = db.Where("api_group = ?", data.ApiGroup)
|
|
}
|
|
|
|
db.Where("delete_time IS NULL")
|
|
err := db.Count(&total).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
err = db.Order("api_group").Limit(pageSize).Offset(offset).Find(&list).Error
|
|
return &list, total, err
|
|
}
|
|
|
|
func (m *sysApiModelImpl) FindList(data entity.SysApi) (*[]entity.SysApi, error) {
|
|
list := make([]entity.SysApi, 0)
|
|
db := global.Db.Table(m.table)
|
|
|
|
if data.Path != "" {
|
|
db = db.Where("path LIKE ?", "%"+data.Path+"%")
|
|
}
|
|
|
|
if data.Description != "" {
|
|
db = db.Where("description LIKE ?", "%"+data.Description+"%")
|
|
}
|
|
|
|
if data.Method != "" {
|
|
db = db.Where("method = ?", data.Method)
|
|
}
|
|
|
|
if data.ApiGroup != "" {
|
|
db = db.Where("api_group = ?", data.ApiGroup)
|
|
}
|
|
db.Where("delete_time IS NULL")
|
|
err := db.Order("api_group").Find(&list).Error
|
|
return &list, err
|
|
}
|
|
|
|
func (m *sysApiModelImpl) Update(api entity.SysApi) error {
|
|
var oldA entity.SysApi
|
|
err := global.Db.Table(m.table).Where("id = ?", api.Id).First(&oldA).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if oldA.Path != api.Path || oldA.Method != api.Method {
|
|
err := global.Db.Table(m.table).Where("path = ? AND method = ?", api.Path, api.Method).First(&entity.SysApi{}).Error
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("存在相同api路径")
|
|
}
|
|
}
|
|
// 异常直接抛错误
|
|
ca := casbin.CasbinService{ModelPath: global.Conf.Casbin.ModelPath}
|
|
err = ca.UpdateCasbinApi(oldA.Path, api.Path, oldA.Method, api.Method)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = global.Db.Table(m.table).Model(&api).Updates(&api).Error
|
|
return err
|
|
}
|
|
|
|
func (m *sysApiModelImpl) Delete(ids []int64) error {
|
|
return global.Db.Table(m.table).Delete(&entity.SysApi{}, "id in (?)", ids).Error
|
|
}
|