70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package repo
|
|
|
|
import (
|
|
"admin/apps/game/domain/entity"
|
|
"admin/apps/game/model"
|
|
"admin/internal/errcode"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type IServerRepo interface {
|
|
List(pageNo, pageLen int) ([]*entity.Server, error)
|
|
Create(et *entity.Server) error
|
|
Edit(et *entity.Server) error
|
|
Delete(id int) error
|
|
}
|
|
|
|
func NewServerRepo(db *gorm.DB) IServerRepo {
|
|
return newServerRepoImpl(db)
|
|
}
|
|
|
|
type ServerRepoImpl struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func newServerRepoImpl(db *gorm.DB) *ServerRepoImpl {
|
|
return &ServerRepoImpl{db: db}
|
|
}
|
|
|
|
func (repo *ServerRepoImpl) List(pageNo, pageLen int) ([]*entity.Server, error) {
|
|
list := make([]*model.Server, 0)
|
|
err := repo.db.Find(&list).Error
|
|
if err != nil {
|
|
return nil, errcode.New(errcode.DBError, "find Server error:%v", err)
|
|
}
|
|
|
|
entityList := make([]*entity.Server, 0, len(list))
|
|
for _, Server := range list {
|
|
entityList = append(entityList, entity.FromServerPo(Server))
|
|
}
|
|
|
|
return entityList, nil
|
|
}
|
|
|
|
func (repo *ServerRepoImpl) Create(et *entity.Server) 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 *ServerRepoImpl) Edit(et *entity.Server) 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 *ServerRepoImpl) Delete(id int) error {
|
|
err := repo.db.Where("id=?", id).Unscoped().Delete(&model.Server{}).Error
|
|
if err != nil {
|
|
return errcode.New(errcode.DBError, "delete obj:%+v error:%v", id, err)
|
|
}
|
|
return nil
|
|
}
|