70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
|
package repo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"admin/apps/mockpro/domain/entity"
|
||
|
|
"admin/apps/mockpro/internal/model"
|
||
|
|
"admin/internal/errcode"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type IWhiteListRepo interface {
|
||
|
|
List(pageNo, pageLen int) ([]*entity.WhiteList, error)
|
||
|
|
Create(et *entity.WhiteList) error
|
||
|
|
Edit(et *entity.WhiteList) error
|
||
|
|
Delete(id int) error
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWhiteListRepo(db *gorm.DB) IWhiteListRepo {
|
||
|
|
return newWhiteListRepoImpl(db)
|
||
|
|
}
|
||
|
|
|
||
|
|
type WhiteListRepoImpl struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func newWhiteListRepoImpl(db *gorm.DB) *WhiteListRepoImpl {
|
||
|
|
return &WhiteListRepoImpl{db: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *WhiteListRepoImpl) List(pageNo, pageLen int) ([]*entity.WhiteList, error) {
|
||
|
|
list := make([]*model.WhiteList, 0)
|
||
|
|
err := repo.db.Find(&list).Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, errcode.New(errcode.DBError, "find WhiteList error:%v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
entityList := make([]*entity.WhiteList, 0, len(list))
|
||
|
|
for _, WhiteList := range list {
|
||
|
|
entityList = append(entityList, entity.FromWhiteListPo(WhiteList))
|
||
|
|
}
|
||
|
|
|
||
|
|
return entityList, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *WhiteListRepoImpl) Create(et *entity.WhiteList) error {
|
||
|
|
po := et.ToPo()
|
||
|
|
err := repo.db.Create(po).Error
|
||
|
|
if err != nil {
|
||
|
|
return errcode.New(errcode.DBError, "create obj:%+v error:%v", et, err)
|
||
|
|
}
|
||
|
|
et.Id = po.ID
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *WhiteListRepoImpl) Edit(et *entity.WhiteList) error {
|
||
|
|
po := et.ToPo()
|
||
|
|
err := repo.db.Where("id=?", et.Id).Updates(po).Error
|
||
|
|
if err != nil {
|
||
|
|
return errcode.New(errcode.DBError, "edit obj:%+v error:%v", et, err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo *WhiteListRepoImpl) Delete(id int) error {
|
||
|
|
err := repo.db.Where("id=?", id).Unscoped().Delete(&model.WhiteList{}).Error
|
||
|
|
if err != nil {
|
||
|
|
return errcode.New(errcode.DBError, "delete obj:%+v error:%v", id, err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|