diff --git a/internal/app/service/pig_farm_service.go b/internal/app/service/pig_farm_service.go index bb6f746..5f4b0c5 100644 --- a/internal/app/service/pig_farm_service.go +++ b/internal/app/service/pig_farm_service.go @@ -236,7 +236,7 @@ func (s *pigFarmService) UpdatePenStatus(id uint, newStatus models.PenStatus) (* "status": newStatus, } - if err := s.penRepository.UpdatePenFields(tx, id, updates); err != nil { + if err := s.penRepository.UpdatePenFieldsTx(tx, id, updates); err != nil { s.logger.Errorf("更新猪栏 %d 状态失败: %v", id, err) return fmt.Errorf("更新猪栏 %d 状态失败: %w", id, err) } diff --git a/internal/infra/repository/pig_pen_repository.go b/internal/infra/repository/pig_pen_repository.go index 8d894cc..84a1b69 100644 --- a/internal/infra/repository/pig_pen_repository.go +++ b/internal/infra/repository/pig_pen_repository.go @@ -17,66 +17,43 @@ type PigPenRepository interface { 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 - + // GetPensByBatchIDTx 根据批次ID获取所有关联的猪栏 (事务性) GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error) + // UpdatePenFieldsTx 更新猪栏的指定字段 (事务性) UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error } -// pigPenRepository 是 PigPenRepository 接口的 gorm 实现。 -type pigPenRepository struct { +// gormPigPenRepository 是 PigPenRepository 接口的 GORM 实现。 +type gormPigPenRepository struct { db *gorm.DB } -// NewGormPigPenRepository 创建一个新的 PigPenRepository 实例。 +// NewGormPigPenRepository 创建一个新的 PigPenRepository GORM 实现实例。 func NewGormPigPenRepository(db *gorm.DB) PigPenRepository { - return &pigPenRepository{db: db} + return &gormPigPenRepository{db: db} } -// GetPenByIDTx 在指定的事务中,通过ID获取单个猪栏信息。 -func (r *pigPenRepository) GetPenByIDTx(tx *gorm.DB, penID uint) (*models.Pen, error) { - var pen models.Pen - if err := tx.First(&pen, penID).Error; err != nil { - return nil, err - } - return &pen, nil -} - -// GetPensByBatchIDTx 在指定的事务中,获取一个猪群当前关联的所有猪栏。 -func (r *pigPenRepository) GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error) { - var pens []*models.Pen - if err := tx.Where("pig_batch_id = ?", batchID).Find(&pens).Error; err != nil { - return nil, err - } - return pens, nil -} - -// UpdatePenFieldsTx 在指定的事务中,更新一个猪栏的指定字段。 -func (r *pigPenRepository) UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error { - return tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates).Error -} - -// --- Pen Implementation --- - // CreatePen 创建一个新的猪栏 -func (r *pigPenRepository) CreatePen(pen *models.Pen) error { +func (r *gormPigPenRepository) CreatePen(pen *models.Pen) error { return r.db.Create(pen).Error } // GetPenByID 根据ID获取单个猪栏 (非事务性) -func (r *pigPenRepository) GetPenByID(id uint) (*models.Pen, error) { +func (r *gormPigPenRepository) GetPenByID(id uint) (*models.Pen, error) { + return r.GetPenByIDTx(r.db, id) // 非Tx方法直接调用Tx方法 +} + +// GetPenByIDTx 在指定的事务中,通过ID获取单个猪栏信息。 +func (r *gormPigPenRepository) GetPenByIDTx(tx *gorm.DB, id uint) (*models.Pen, error) { var pen models.Pen - if err := r.db.First(&pen, id).Error; err != nil { + if err := tx.First(&pen, id).Error; err != nil { return nil, err } return &pen, nil } // ListPens 列出所有猪栏 -func (r *pigPenRepository) ListPens() ([]models.Pen, error) { +func (r *gormPigPenRepository) ListPens() ([]models.Pen, error) { var pens []models.Pen if err := r.db.Find(&pens).Error; err != nil { return nil, err @@ -85,7 +62,7 @@ func (r *pigPenRepository) ListPens() ([]models.Pen, error) { } // UpdatePen 更新一个猪栏,返回受影响的行数和错误 -func (r *pigPenRepository) UpdatePen(pen *models.Pen) (int64, error) { +func (r *gormPigPenRepository) 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 @@ -94,7 +71,7 @@ func (r *pigPenRepository) UpdatePen(pen *models.Pen) (int64, error) { } // DeletePen 根据ID删除一个猪栏,返回受影响的行数和错误 -func (r *pigPenRepository) DeletePen(id uint) (int64, error) { +func (r *gormPigPenRepository) DeletePen(id uint) (int64, error) { result := r.db.Delete(&models.Pen{}, id) if result.Error != nil { return 0, result.Error @@ -102,9 +79,9 @@ func (r *pigPenRepository) DeletePen(id uint) (int64, error) { return result.RowsAffected, nil } -// GetPensByBatchID 根据批次ID获取所有关联的猪栏 (事务性) -func (r *pigPenRepository) GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error) { - var pens []models.Pen +// GetPensByBatchIDTx 在指定的事务中,获取一个猪群当前关联的所有猪栏。 +func (r *gormPigPenRepository) GetPensByBatchIDTx(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 { @@ -113,8 +90,8 @@ func (r *pigPenRepository) GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models return pens, nil } -// UpdatePenFields 更新猪栏的指定字段 (事务性) -func (r *pigPenRepository) UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error { +// UpdatePenFieldsTx 在指定的事务中,更新一个猪栏的指定字段。 +func (r *gormPigPenRepository) UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error { result := tx.Model(&models.Pen{}).Where("id = ?", penID).Updates(updates) return result.Error }