修改方法命名

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)
 		}

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获取所有关联的猪栏 (事务性)
+	// GetPensByBatchIDTx 根据批次ID获取所有关联的猪栏 (事务性)
-	GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error)
-	// UpdatePenFields 更新猪栏的指定字段 (事务性)
-	UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error
-
 	GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error)
+	// UpdatePenFieldsTx 更新猪栏的指定字段 (事务性)
 	UpdatePenFieldsTx(tx *gorm.DB, penID uint, updates map[string]interface{}) error
 }
 
-// pigPenRepository 是 PigPenRepository 接口的 gorm 实现。
+// gormPigPenRepository 是 PigPenRepository 接口的 GORM 实现。
-type pigPenRepository struct {
+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获取所有关联的猪栏 (事务性)
+// GetPensByBatchIDTx 在指定的事务中，获取一个猪群当前关联的所有猪栏。
-func (r *pigPenRepository) GetPensByBatchID(tx *gorm.DB, batchID uint) ([]models.Pen, error) {
+func (r *gormPigPenRepository) GetPensByBatchIDTx(tx *gorm.DB, batchID uint) ([]*models.Pen, error) {
-	var pens []models.Pen
+	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 更新猪栏的指定字段 (事务性)
+// UpdatePenFieldsTx 在指定的事务中，更新一个猪栏的指定字段。
-func (r *pigPenRepository) UpdatePenFields(tx *gorm.DB, penID uint, updates map[string]interface{}) error {
+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
 }