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 }