猪群管理聚合服务
This commit is contained in:
@@ -1,55 +1,37 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/app/dto"
|
||||
domain_pig "git.huangwc.com/pig/pig-farm-controller/internal/domain/pig"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
||||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrPigBatchNotFound = errors.New("指定的猪批次不存在")
|
||||
ErrPigBatchActive = errors.New("活跃的猪批次不能被删除")
|
||||
ErrPigBatchNotActive = errors.New("猪批次不处于活跃状态，无法修改关联猪栏")
|
||||
ErrPenOccupiedByOtherBatch = errors.New("猪栏已被其他批次使用")
|
||||
ErrPenStatusInvalidForAllocation = errors.New("猪栏状态不允许分配")
|
||||
ErrPenNotAssociatedWithBatch = errors.New("猪栏未与该批次关联")
|
||||
)
|
||||
|
||||
// PigBatchService 提供了猪批次管理的业务逻辑
|
||||
// PigBatchService 接口定义保持不变，继续作为应用层对外的契约。
|
||||
type PigBatchService interface {
|
||||
CreatePigBatch(dto *dto.PigBatchCreateDTO) (*dto.PigBatchResponseDTO, error)
|
||||
GetPigBatch(id uint) (*dto.PigBatchResponseDTO, error)
|
||||
UpdatePigBatch(id uint, dto *dto.PigBatchUpdateDTO) (*dto.PigBatchResponseDTO, error)
|
||||
DeletePigBatch(id uint) error
|
||||
ListPigBatches(isActive *bool) ([]*dto.PigBatchResponseDTO, error)
|
||||
// UpdatePigBatchPens 更新猪批次关联的猪栏
|
||||
UpdatePigBatchPens(batchID uint, desiredPenIDs []uint) error
|
||||
}
|
||||
|
||||
// pigBatchService 的实现现在依赖于领域服务接口。
|
||||
type pigBatchService struct {
|
||||
logger *logs.Logger
|
||||
pigBatchRepo repository.PigBatchRepository // 猪批次仓库
|
||||
pigFarmRepo repository.PigFarmRepository // 猪场资产仓库 (包含猪栏操作)
|
||||
uow repository.UnitOfWork // 工作单元，用于事务管理
|
||||
logger *logs.Logger
|
||||
domainService domain_pig.PigBatchService // 依赖注入领域服务
|
||||
}
|
||||
|
||||
// NewPigBatchService 创建一个新的 PigBatchService 实例
|
||||
func NewPigBatchService(pigBatchRepo repository.PigBatchRepository, pigFarmRepo repository.PigFarmRepository, uow repository.UnitOfWork, logger *logs.Logger) PigBatchService {
|
||||
// NewPigBatchService 构造函数被修改，以注入领域服务。
|
||||
func NewPigBatchService(domainService domain_pig.PigBatchService, logger *logs.Logger) PigBatchService {
|
||||
return &pigBatchService{
|
||||
logger: logger,
|
||||
pigBatchRepo: pigBatchRepo,
|
||||
pigFarmRepo: pigFarmRepo,
|
||||
uow: uow,
|
||||
logger: logger,
|
||||
domainService: domainService,
|
||||
}
|
||||
}
|
||||
|
||||
// toPigBatchResponseDTO 将 models.PigBatch 转换为 dto.PigBatchResponseDTO
|
||||
// toPigBatchResponseDTO 负责将领域模型转换为应用层DTO，这个职责保留在应用层。
|
||||
func (s *pigBatchService) toPigBatchResponseDTO(batch *models.PigBatch) *dto.PigBatchResponseDTO {
|
||||
if batch == nil {
|
||||
return nil
|
||||
@@ -62,14 +44,15 @@ func (s *pigBatchService) toPigBatchResponseDTO(batch *models.PigBatch) *dto.Pig
|
||||
EndDate: batch.EndDate,
|
||||
InitialCount: batch.InitialCount,
|
||||
Status: batch.Status,
|
||||
IsActive: batch.IsActive(), // 使用模型自带的 IsActive 方法
|
||||
IsActive: batch.IsActive(),
|
||||
CreateTime: batch.CreatedAt,
|
||||
UpdateTime: batch.UpdatedAt,
|
||||
}
|
||||
}
|
||||
|
||||
// CreatePigBatch 处理创建猪批次的业务逻辑
|
||||
// CreatePigBatch 现在将请求委托给领域服务处理。
|
||||
func (s *pigBatchService) CreatePigBatch(dto *dto.PigBatchCreateDTO) (*dto.PigBatchResponseDTO, error) {
|
||||
// 1. DTO -> 领域模型
|
||||
batch := &models.PigBatch{
|
||||
BatchNumber: dto.BatchNumber,
|
||||
OriginType: dto.OriginType,
|
||||
@@ -78,41 +61,38 @@ func (s *pigBatchService) CreatePigBatch(dto *dto.PigBatchCreateDTO) (*dto.PigBa
|
||||
Status: dto.Status,
|
||||
}
|
||||
|
||||
createdBatch, err := s.pigBatchRepo.CreatePigBatch(batch)
|
||||
// 2. 调用领域服务
|
||||
createdBatch, err := s.domainService.CreatePigBatch(batch)
|
||||
if err != nil {
|
||||
s.logger.Errorf("创建猪批次失败: %v", err)
|
||||
return nil, err
|
||||
s.logger.Errorf("应用层: 创建猪批次失败: %v", err)
|
||||
return nil, err // 将领域层的错误传递上去
|
||||
}
|
||||
|
||||
// 3. 领域模型 -> DTO
|
||||
return s.toPigBatchResponseDTO(createdBatch), nil
|
||||
}
|
||||
|
||||
// GetPigBatch 处理获取单个猪批次的业务逻辑
|
||||
// GetPigBatch 从领域服务获取数据并转换为DTO。
|
||||
func (s *pigBatchService) GetPigBatch(id uint) (*dto.PigBatchResponseDTO, error) {
|
||||
batch, err := s.pigBatchRepo.GetPigBatchByID(id)
|
||||
batch, err := s.domainService.GetPigBatch(id)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrPigBatchNotFound
|
||||
}
|
||||
s.logger.Errorf("获取猪批次失败，ID: %d, 错误: %v", id, err)
|
||||
s.logger.Warnf("应用层: 获取猪批次失败, ID: %d, 错误: %v", id, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return s.toPigBatchResponseDTO(batch), nil
|
||||
}
|
||||
|
||||
// UpdatePigBatch 处理更新猪批次的业务逻辑
|
||||
// UpdatePigBatch 协调获取、更新和保存的流程。
|
||||
func (s *pigBatchService) UpdatePigBatch(id uint, dto *dto.PigBatchUpdateDTO) (*dto.PigBatchResponseDTO, error) {
|
||||
existingBatch, err := s.pigBatchRepo.GetPigBatchByID(id)
|
||||
// 1. 先获取最新的领域模型
|
||||
existingBatch, err := s.domainService.GetPigBatch(id)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrPigBatchNotFound
|
||||
}
|
||||
s.logger.Errorf("更新猪批次失败，获取原批次信息错误，ID: %d, 错误: %v", id, err)
|
||||
s.logger.Warnf("应用层: 更新猪批次失败，获取原批次信息错误, ID: %d, 错误: %v", id, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 根据 DTO 中的非空字段更新模型
|
||||
// 2. 将DTO中的变更应用到模型上
|
||||
if dto.BatchNumber != nil {
|
||||
existingBatch.BatchNumber = *dto.BatchNumber
|
||||
}
|
||||
@@ -132,55 +112,32 @@ func (s *pigBatchService) UpdatePigBatch(id uint, dto *dto.PigBatchUpdateDTO) (*
|
||||
existingBatch.Status = *dto.Status
|
||||
}
|
||||
|
||||
updatedBatch, rowsAffected, err := s.pigBatchRepo.UpdatePigBatch(existingBatch)
|
||||
// 3. 调用领域服务执行更新
|
||||
updatedBatch, err := s.domainService.UpdatePigBatch(existingBatch)
|
||||
if err != nil {
|
||||
s.logger.Errorf("更新猪批次失败，ID: %d, 错误: %v", id, err)
|
||||
s.logger.Errorf("应用层: 更新猪批次失败, ID: %d, 错误: %v", id, err)
|
||||
return nil, err
|
||||
}
|
||||
// 如果没有行受影响，则认为猪批次不存在
|
||||
if rowsAffected == 0 {
|
||||
return nil, ErrPigBatchNotFound
|
||||
}
|
||||
|
||||
// 4. 转换并返回结果
|
||||
return s.toPigBatchResponseDTO(updatedBatch), nil
|
||||
}
|
||||
|
||||
// DeletePigBatch 处理删除猪批次的业务逻辑
|
||||
// DeletePigBatch 将删除操作委托给领域服务。
|
||||
func (s *pigBatchService) DeletePigBatch(id uint) error {
|
||||
// 1. 获取猪批次信息
|
||||
batch, err := s.pigBatchRepo.GetPigBatchByID(id)
|
||||
err := s.domainService.DeletePigBatch(id)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return ErrPigBatchNotFound
|
||||
}
|
||||
s.logger.Errorf("删除猪批次失败，获取批次信息错误，ID: %d, 错误: %v", id, err)
|
||||
s.logger.Errorf("应用层: 删除猪批次失败, ID: %d, 错误: %v", id, err)
|
||||
return err
|
||||
}
|
||||
|
||||
// 2. 检查猪批次是否活跃
|
||||
if batch.IsActive() {
|
||||
return ErrPigBatchActive // 如果活跃，则不允许删除
|
||||
}
|
||||
|
||||
// 3. 执行删除操作
|
||||
rowsAffected, err := s.pigBatchRepo.DeletePigBatch(id)
|
||||
if err != nil {
|
||||
s.logger.Errorf("删除猪批次失败，ID: %d, 错误: %v", id, err)
|
||||
return err
|
||||
}
|
||||
// 如果没有行受影响，则认为猪批次不存在
|
||||
if rowsAffected == 0 {
|
||||
return ErrPigBatchNotFound
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ListPigBatches 处理批量查询猪批次的业务逻辑
|
||||
// ListPigBatches 从领域服务获取列表并进行转换。
|
||||
func (s *pigBatchService) ListPigBatches(isActive *bool) ([]*dto.PigBatchResponseDTO, error) {
|
||||
batches, err := s.pigBatchRepo.ListPigBatches(isActive)
|
||||
batches, err := s.domainService.ListPigBatches(isActive)
|
||||
if err != nil {
|
||||
s.logger.Errorf("批量查询猪批次失败，错误: %v", err)
|
||||
s.logger.Errorf("应用层: 批量查询猪批次失败: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@@ -192,111 +149,12 @@ func (s *pigBatchService) ListPigBatches(isActive *bool) ([]*dto.PigBatchRespons
|
||||
return responseDTOs, nil
|
||||
}
|
||||
|
||||
// UpdatePigBatchPens 更新猪批次关联的猪栏
|
||||
// UpdatePigBatchPens 将关联猪栏的复杂操作委托给领域服务。
|
||||
func (s *pigBatchService) UpdatePigBatchPens(batchID uint, desiredPenIDs []uint) error {
|
||||
// 使用工作单元执行事务
|
||||
return s.uow.ExecuteInTransaction(func(tx *gorm.DB) error {
|
||||
// 1. 验证猪批次
|
||||
pigBatch, err := s.pigFarmRepo.GetPigBatchByIDTx(tx, batchID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return ErrPigBatchNotFound
|
||||
}
|
||||
s.logger.Errorf("更新猪批次猪栏失败: 获取猪批次信息错误，ID: %d, 错误: %v", batchID, err)
|
||||
return fmt.Errorf("获取猪批次信息失败: %w", err)
|
||||
}
|
||||
|
||||
if !pigBatch.IsActive() {
|
||||
return ErrPigBatchNotActive
|
||||
}
|
||||
|
||||
// 2. 获取当前关联的猪栏
|
||||
currentPens, err := s.pigFarmRepo.GetPensByBatchID(tx, batchID)
|
||||
if err != nil {
|
||||
s.logger.Errorf("更新猪批次猪栏失败: 获取当前关联猪栏错误，批次ID: %d, 错误: %v", batchID, err)
|
||||
return fmt.Errorf("获取当前关联猪栏失败: %w", err)
|
||||
}
|
||||
|
||||
currentPenMap := make(map[uint]models.Pen)
|
||||
currentPenIDsSet := make(map[uint]struct{})
|
||||
for _, pen := range currentPens {
|
||||
currentPenMap[pen.ID] = pen
|
||||
currentPenIDsSet[pen.ID] = struct{}{} // 用于快速查找
|
||||
}
|
||||
|
||||
// 3. 构建期望猪栏集合
|
||||
desiredPenIDsSet := make(map[uint]struct{})
|
||||
for _, penID := range desiredPenIDs {
|
||||
desiredPenIDsSet[penID] = struct{}{} // 用于快速查找
|
||||
}
|
||||
|
||||
// 4. 计算需要添加和移除的猪栏
|
||||
var pensToRemove []uint
|
||||
for penID := range currentPenIDsSet {
|
||||
if _, found := desiredPenIDsSet[penID]; !found {
|
||||
pensToRemove = append(pensToRemove, penID)
|
||||
}
|
||||
}
|
||||
|
||||
var pensToAdd []uint
|
||||
for _, penID := range desiredPenIDs {
|
||||
if _, found := currentPenIDsSet[penID]; !found {
|
||||
pensToAdd = append(pensToAdd, penID)
|
||||
}
|
||||
}
|
||||
|
||||
// 5. 处理移除猪栏
|
||||
for _, penID := range pensToRemove {
|
||||
currentPen := currentPenMap[penID]
|
||||
// 验证：确保猪栏确实与当前批次关联
|
||||
if currentPen.PigBatchID == nil || *currentPen.PigBatchID != batchID {
|
||||
s.logger.Warnf("尝试移除未与批次 %d 关联的猪栏 %d", batchID, penID)
|
||||
return fmt.Errorf("猪栏 %d 未与该批次关联，无法移除", penID)
|
||||
}
|
||||
|
||||
updates := make(map[string]interface{})
|
||||
updates["pig_batch_id"] = nil // 总是将 PigBatchID 设为 nil
|
||||
|
||||
// 只有当猪栏当前状态是“使用中”时，才将其状态改回“空闲”
|
||||
if currentPen.Status == models.PenStatusOccupied {
|
||||
updates["status"] = models.PenStatusEmpty
|
||||
}
|
||||
|
||||
if err := s.pigFarmRepo.UpdatePenFields(tx, penID, updates); err != nil {
|
||||
s.logger.Errorf("更新猪批次猪栏失败: 移除猪栏 %d 失败: %v", penID, err)
|
||||
return fmt.Errorf("移除猪栏 %d 失败: %w", penID, err)
|
||||
}
|
||||
}
|
||||
|
||||
// 6. 处理添加猪栏
|
||||
for _, penID := range pensToAdd {
|
||||
actualPen, err := s.pigFarmRepo.GetPenByIDTx(tx, penID)
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return fmt.Errorf("猪栏 %d 不存在: %w", penID, ErrPenNotFound)
|
||||
}
|
||||
s.logger.Errorf("更新猪批次猪栏失败: 获取猪栏 %d 信息错误: %v", penID, err)
|
||||
return fmt.Errorf("获取猪栏 %d 信息失败: %w", penID, err)
|
||||
}
|
||||
|
||||
// 验证：猪栏必须是“空闲”状态且未被任何批次使用，才能被分配
|
||||
if actualPen.Status != models.PenStatusEmpty {
|
||||
return fmt.Errorf("猪栏 %s 状态为 %s，无法分配: %w", actualPen.PenNumber, actualPen.Status, ErrPenStatusInvalidForAllocation)
|
||||
}
|
||||
if actualPen.PigBatchID != nil {
|
||||
return fmt.Errorf("猪栏 %s 已被其他批次 %d 使用，无法分配: %w", actualPen.PenNumber, *actualPen.PigBatchID, ErrPenOccupiedByOtherBatch)
|
||||
}
|
||||
|
||||
updates := map[string]interface{}{
|
||||
"pig_batch_id": &batchID, // 将 PigBatchID 设为当前批次ID的指针
|
||||
"status": models.PenStatusOccupied, // 分配后，状态变为“使用中”
|
||||
}
|
||||
if err := s.pigFarmRepo.UpdatePenFields(tx, penID, updates); err != nil {
|
||||
s.logger.Errorf("更新猪批次猪栏失败: 添加猪栏 %d 失败: %v", penID, err)
|
||||
return fmt.Errorf("添加猪栏 %d 失败: %w", penID, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
err := s.domainService.UpdatePigBatchPens(batchID, desiredPenIDs)
|
||||
if err != nil {
|
||||
s.logger.Errorf("应用层: 更新猪批次猪栏关联失败, 批次ID: %d, 错误: %v", batchID, err)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user