猪群相关接口增加当前总量和当前总存栏量

internal/domain/pig/pig_batch_service.go

@@ -6,7 +6,6 @@ import (
 
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
-	"gorm.io/gorm"
 )
 
 // --- 业务错误定义 ---
@@ -60,6 +59,8 @@ type PigBatchService interface {
 	GetCurrentPigQuantity(batchID uint) (int, error)
 	// GetCurrentPigsInPen 获取指定猪栏的当前存栏量。
 	GetCurrentPigsInPen(penID uint) (int, error)
+	// GetTotalPigsInPensForBatch 获取指定猪群下所有猪栏的当前总存栏数
+	GetTotalPigsInPensForBatch(batchID uint) (int, error)
 
 	UpdatePigBatchQuantity(operatorID uint, batchID uint, changeType models.LogChangeType, changeAmount int, changeReason string, happenedAt time.Time) error
 
@@ -120,61 +121,3 @@ func NewPigBatchService(
 		sickSvc:         sickSvc,
 	}
 }
-
-func (s *pigBatchService) RemoveEmptyPenFromBatch(batchID uint, penID uint) error {
-	return s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
-		// 1. 检查猪批次是否存在且活跃
-		batch, err := s.pigBatchRepo.GetPigBatchByIDTx(tx, batchID)
-		if err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				return ErrPigBatchNotFound
-			}
-			return err
-		}
-		if !batch.IsActive() {
-			return ErrPigBatchNotActive
-		}
-
-		// 2. 检查猪栏是否存在
-		pen, err := s.transferSvc.GetPenByID(tx, penID)
-		if err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				return ErrPenNotFound
-			}
-			return err
-		}
-
-		// 3. 检查猪栏是否与当前批次关联
-		if pen.PigBatchID == nil || *pen.PigBatchID != batchID {
-			return ErrPenNotAssociatedWithBatch
-		}
-
-		// 4. 检查猪栏是否为空
-		pigsInPen, err := s.transferSvc.GetCurrentPigsInPen(tx, penID)
-		if err != nil {
-			return err
-		}
-		if pigsInPen > 0 {
-			return ErrPenNotEmpty
-		}
-
-		// 5. 释放猪栏 (将 pig_batch_id 设置为 nil，状态设置为空闲)
-		if err := s.transferSvc.ReleasePen(tx, penID); err != nil {
-			return err
-		}
-		return nil
-	})
-}
-
-func (s *pigBatchService) GetCurrentPigsInPen(penID uint) (int, error) {
-	var currentPigs int
-	err := s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
-		pigs, err := s.transferSvc.GetCurrentPigsInPen(tx, penID)
-		if err != nil {
-			return err
-		}
-		currentPigs = pigs
-		return nil
-	})
-	return currentPigs, err
-}

internal/domain/pig/pig_batch_service_pen_transfer.go

@@ -381,3 +381,75 @@ func (s *pigBatchService) ReclassifyPenToNewBatch(fromBatchID uint, toBatchID ui
 		return nil
 	})
 }
+
+func (s *pigBatchService) RemoveEmptyPenFromBatch(batchID uint, penID uint) error {
+	return s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
+		// 1. 检查猪批次是否存在且活跃
+		batch, err := s.pigBatchRepo.GetPigBatchByIDTx(tx, batchID)
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				return ErrPigBatchNotFound
+			}
+			return err
+		}
+		if !batch.IsActive() {
+			return ErrPigBatchNotActive
+		}
+
+		// 2. 检查猪栏是否存在
+		pen, err := s.transferSvc.GetPenByID(tx, penID)
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				return ErrPenNotFound
+			}
+			return err
+		}
+
+		// 3. 检查猪栏是否与当前批次关联
+		if pen.PigBatchID == nil || *pen.PigBatchID != batchID {
+			return ErrPenNotAssociatedWithBatch
+		}
+
+		// 4. 检查猪栏是否为空
+		pigsInPen, err := s.transferSvc.GetCurrentPigsInPen(tx, penID)
+		if err != nil {
+			return err
+		}
+		if pigsInPen > 0 {
+			return ErrPenNotEmpty
+		}
+
+		// 5. 释放猪栏 (将 pig_batch_id 设置为 nil，状态设置为空闲)
+		if err := s.transferSvc.ReleasePen(tx, penID); err != nil {
+			return err
+		}
+		return nil
+	})
+}
+
+func (s *pigBatchService) GetCurrentPigsInPen(penID uint) (int, error) {
+	var currentPigs int
+	err := s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
+		pigs, err := s.transferSvc.GetCurrentPigsInPen(tx, penID)
+		if err != nil {
+			return err
+		}
+		currentPigs = pigs
+		return nil
+	})
+	return currentPigs, err
+}
+
+// GetTotalPigsInPensForBatch 实现了获取指定猪群下所有猪栏的当前总存栏数的逻辑。
+func (s *pigBatchService) GetTotalPigsInPensForBatch(batchID uint) (int, error) {
+	var totalPigs int
+	err := s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
+		pigs, err := s.transferSvc.GetTotalPigsInPensForBatchTx(tx, batchID)
+		if err != nil {
+			return err
+		}
+		totalPigs = pigs
+		return nil
+	})
+	return totalPigs, err
+}