调整仓库方法归属

internal/app/service/pig_farm_service.go

@@ -31,19 +31,25 @@ type PigFarmService interface {
 }
 
 type pigFarmService struct {
-	logger         *logs.Logger
-	farmRepository repository.PigFarmRepository
-	penRepository  repository.PigPenRepository
-	uow            repository.UnitOfWork // 工作单元，用于事务管理
+	logger          *logs.Logger
+	farmRepository  repository.PigFarmRepository
+	penRepository   repository.PigPenRepository
+	batchRepository repository.PigBatchRepository
+	uow             repository.UnitOfWork // 工作单元，用于事务管理
 }
 
 // NewPigFarmService 创建一个新的 PigFarmService 实例
-func NewPigFarmService(farmRepository repository.PigFarmRepository, penRepository repository.PigPenRepository, uow repository.UnitOfWork, logger *logs.Logger) PigFarmService {
+func NewPigFarmService(farmRepository repository.PigFarmRepository,
+	penRepository repository.PigPenRepository,
+	batchRepository repository.PigBatchRepository,
+	uow repository.UnitOfWork,
+	logger *logs.Logger) PigFarmService {
 	return &pigFarmService{
-		logger:         logger,
-		farmRepository: farmRepository,
-		penRepository:  penRepository,
-		uow:            uow,
+		logger:          logger,
+		farmRepository:  farmRepository,
+		penRepository:   penRepository,
+		batchRepository: batchRepository,
+		uow:             uow,
 	}
 }
 
@@ -175,7 +181,7 @@ func (s *pigFarmService) DeletePen(id uint) error {
 	// 检查猪栏是否关联了活跃批次
 	// 注意：pen.PigBatchID 是指针类型，需要检查是否为 nil
 	if pen.PigBatchID != nil && *pen.PigBatchID != 0 {
-		pigBatch, err := s.farmRepository.GetPigBatchByID(*pen.PigBatchID)
+		pigBatch, err := s.batchRepository.GetPigBatchByID(*pen.PigBatchID)
 		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 			return err
 		}