业务逻辑移动到服务层

internal/infra/repository/pig_farm_repository.go

@@ -1,18 +1,10 @@
 package repository
 
 import (
-	"errors"
-
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"gorm.io/gorm"
 )
 
-var (
-	ErrHouseContainsPens = errors.New("请在移除所有猪圈后移除当前猪舍")
-	ErrHouseNotFound     = errors.New("猪舍不存在")
-	ErrPenInUse          = errors.New("猪栏正在被活跃批次使用，无法删除")
-)
-
 // PigFarmRepository 定义了与猪场资产（猪舍、猪栏）相关的数据库操作接口
 type PigFarmRepository interface {
 	// PigHouse methods
@@ -21,6 +13,7 @@ type PigFarmRepository interface {
 	ListPigHouses() ([]models.PigHouse, error)
 	UpdatePigHouse(house *models.PigHouse) error
 	DeletePigHouse(id uint) error
+	CountPensInHouse(houseID uint) (int64, error)
 
 	// Pen methods
 	CreatePen(pen *models.Pen) error
@@ -28,6 +21,9 @@ type PigFarmRepository interface {
 	ListPens() ([]models.Pen, error)
 	UpdatePen(pen *models.Pen) error
 	DeletePen(id uint) error
+
+	// PigBatch methods
+	GetPigBatchByID(id uint) (*models.PigBatch, error)
 }
 
 // gormPigFarmRepository 是 PigFarmRepository 的 GORM 实现
@@ -74,39 +70,26 @@ func (r *gormPigFarmRepository) UpdatePigHouse(house *models.PigHouse) error {
 }
 
 func (r *gormPigFarmRepository) DeletePigHouse(id uint) error {
-	return r.db.Transaction(func(tx *gorm.DB) error {
-		var penCount int64
-		if err := tx.Model(&models.Pen{}).Where("house_id = ?", id).Count(&penCount).Error; err != nil {
-			return err
-		}
-		if penCount > 0 {
-			return ErrHouseContainsPens
-		}
+	result := r.db.Delete(&models.PigHouse{}, id)
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+	return nil
+}
 
-		result := tx.Delete(&models.PigHouse{}, id)
-		if result.Error != nil {
-			return result.Error
-		}
-		if result.RowsAffected == 0 {
-			return gorm.ErrRecordNotFound
-		}
-		return nil
-	})
+func (r *gormPigFarmRepository) CountPensInHouse(houseID uint) (int64, error) {
+	var count int64
+	err := r.db.Model(&models.Pen{}).Where("house_id = ?", houseID).Count(&count).Error
+	return count, err
 }
 
 // --- Pen Implementation ---
 
 func (r *gormPigFarmRepository) CreatePen(pen *models.Pen) error {
-	return r.db.Transaction(func(tx *gorm.DB) error {
-		// 验证所属猪舍是否存在
-		if err := tx.First(&models.PigHouse{}, pen.HouseID).Error; err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				return ErrHouseNotFound
-			}
-			return err
-		}
-		return tx.Create(pen).Error
-	})
+	return r.db.Create(pen).Error
 }
 
 func (r *gormPigFarmRepository) GetPenByID(id uint) (*models.Pen, error) {
@@ -126,55 +109,33 @@ func (r *gormPigFarmRepository) ListPens() ([]models.Pen, error) {
 }
 
 func (r *gormPigFarmRepository) UpdatePen(pen *models.Pen) error {
-	return r.db.Transaction(func(tx *gorm.DB) error {
-		// 验证所属猪舍是否存在
-		if err := tx.First(&models.PigHouse{}, pen.HouseID).Error; err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				return ErrHouseNotFound
-			}
-			return err
-		}
-
-		result := tx.Model(&models.Pen{}).Where("id = ?", pen.ID).Updates(pen)
-		if result.Error != nil {
-			return result.Error
-		}
-		if result.RowsAffected == 0 {
-			return gorm.ErrRecordNotFound
-		}
-		return nil
-	})
+	result := r.db.Model(&models.Pen{}).Where("id = ?", pen.ID).Updates(pen)
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+	return nil
 }
 
 func (r *gormPigFarmRepository) DeletePen(id uint) error {
-	return r.db.Transaction(func(tx *gorm.DB) error {
-		var pen models.Pen
-		if err := tx.First(&pen, id).Error; err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				return gorm.ErrRecordNotFound
-			}
-			return err
-		}
-
-		// 检查猪栏是否被活跃批次使用
-		if pen.PigBatchID != 0 {
-			var pigBatch models.PigBatch
-			err := tx.First(&pigBatch, pen.PigBatchID).Error
-			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
-				return err
-			}
-			if pigBatch.IsActive() {
-				return ErrPenInUse
-			}
-		}
-
-		result := tx.Delete(&models.Pen{}, id)
-		if result.Error != nil {
-			return result.Error
-		}
-		if result.RowsAffected == 0 {
-			return gorm.ErrRecordNotFound
-		}
-		return nil
-	})
+	result := r.db.Delete(&models.Pen{}, id)
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+	return nil
+}
+
+// --- PigBatch Implementation ---
+
+func (r *gormPigFarmRepository) GetPigBatchByID(id uint) (*models.PigBatch, error) {
+	var batch models.PigBatch
+	if err := r.db.First(&batch, id).Error; err != nil {
+		return nil, err
+	}
+	return &batch, nil
 }