pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

调整仓库方法归属

Browse Source
This commit is contained in:
huang
2025-10-05 17:42:27 +08:00
parent 01327eb8d2
commit b3933b6d63
6 changed files with 158 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
internal/infra/repository/pig_farm_repository.go
View File

@@ -17,22 +17,6 @@ type PigFarmRepository interface {
DeletePigHouse(id uint) (int64, error)
CountPensInHouse(houseID uint) (int64, error)
// Pen methods
CreatePen(pen *models.Pen) error
// GetPenByID 根据ID获取单个猪栏 (非事务性)
GetPenByID(id uint) (*models.Pen, error)
// GetPenByIDTx 根据ID获取单个猪栏 (事务性)
GetPenByIDTx(tx *gorm.DB, id uint) (*models.Pen, error)
ListPens() ([]models.Pen, error)
// UpdatePen 更新一个猪栏，返回受影响的行数和错误
UpdatePen(pen *models.Pen) (int64, error)
// DeletePen 根据ID删除一个猪栏，返回受影响的行数和错误
DeletePen(id uint) (int64, error)
// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error)
// UpdatePenFields 更新猪栏的指定字段 (事务性)
UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error
// PigBatch methods
// GetPigBatchByID 根据ID获取单个猪批次 (非事务性)
GetPigBatchByID(id uint) (*models.PigBatch, error)
@@ -100,75 +84,6 @@ func (r *gormPigFarmRepository) CountPensInHouse(houseID uint) (int64, error) {
return count, err
}
// --- Pen Implementation ---
// CreatePen 创建一个新的猪栏
func (r *gormPigFarmRepository) CreatePen(pen *models.Pen) error {
return r.db.Create(pen).Error
}
// GetPenByID 根据ID获取单个猪栏 (非事务性)
func (r *gormPigFarmRepository) GetPenByID(id uint) (*models.Pen, error) {
var pen models.Pen
if err := r.db.First(&pen, id).Error; err != nil {
return nil, err
}
return &pen, nil
}
// GetPenByIDTx 根据ID获取单个猪栏 (事务性)
func (r *gormPigFarmRepository) GetPenByIDTx(tx *gorm.DB, id uint) (*models.Pen, error) {
var pen models.Pen
if err := tx.First(&pen, id).Error; err != nil {
return nil, err
}
return &pen, nil
}
// ListPens 列出所有猪栏
func (r *gormPigFarmRepository) ListPens() ([]models.Pen, error) {
var pens []models.Pen
if err := r.db.Find(&pens).Error; err != nil {
return nil, err
}
return pens, nil
}
// UpdatePen 更新一个猪栏，返回受影响的行数和错误
func (r *gormPigFarmRepository) UpdatePen(pen *models.Pen) (int64, error) {
result := r.db.Model(&models.Pen{}).Where("id = ?", pen.ID).Updates(pen)
if result.Error != nil {
return 0, result.Error
}
return result.RowsAffected, nil
}
// DeletePen 根据ID删除一个猪栏，返回受影响的行数和错误
func (r *gormPigFarmRepository) DeletePen(id uint) (int64, error) {
result := r.db.Delete(&models.Pen{}, id)
if result.Error != nil {
return 0, result.Error
}
return result.RowsAffected, nil
}
// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性)
func (r *gormPigFarmRepository) GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error) {
var pens []models.Pen
// 注意：PigBatchID 是指针类型，需要处理 nil 值
result := tx.Where("pig_batch_id = ?", batchID).Find(&pens)
if result.Error != nil {
return nil, result.Error
}
return pens, nil
}
// UpdatePenFields 更新猪栏的指定字段 (事务性)
func (r *gormPigFarmRepository) UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
result := tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates)
return result.Error
}
// --- PigBatch Implementation ---
// GetPigBatchByID 根据ID获取单个猪批次 (非事务性)