配方领域层方法+重构配方领域

internal/domain/recipe/raw_material_service.go

@@ -0,0 +1,184 @@
+package recipe
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	"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 (
+	ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在")
+	ErrRawMaterialNotFound     = fmt.Errorf("原料不存在")
+)
+
+// RawMaterialService 定义了原料领域的核心业务服务接口
+type RawMaterialService interface {
+	CreateRawMaterial(ctx context.Context, name, description string) (*models.RawMaterial, error)
+	UpdateRawMaterial(ctx context.Context, id uint32, name, description string) (*models.RawMaterial, error)
+	DeleteRawMaterial(ctx context.Context, id uint32) error
+	GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error)
+	ListRawMaterials(ctx context.Context, opts repository.RawMaterialListOptions, page, pageSize int) ([]models.RawMaterial, int64, error)
+	UpdateRawMaterialNutrients(ctx context.Context, rawMaterialID uint32, nutrients []models.RawMaterialNutrient) error
+}
+
+// rawMaterialServiceImpl 是 RawMaterialService 的实现
+type rawMaterialServiceImpl struct {
+	ctx             context.Context
+	uow             repository.UnitOfWork
+	rawMaterialRepo repository.RawMaterialRepository
+}
+
+// NewRawMaterialService 创建一个新的 RawMaterialService 实例
+func NewRawMaterialService(ctx context.Context, uow repository.UnitOfWork, rawMaterialRepo repository.RawMaterialRepository) RawMaterialService {
+	return &rawMaterialServiceImpl{
+		ctx:             ctx,
+		uow:             uow,
+		rawMaterialRepo: rawMaterialRepo,
+	}
+}
+
+// CreateRawMaterial 实现了创建原料的核心业务逻辑
+func (s *rawMaterialServiceImpl) CreateRawMaterial(ctx context.Context, name, description string) (*models.RawMaterial, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateRawMaterial")
+
+	// 检查名称是否已存在
+	existing, err := s.rawMaterialRepo.GetRawMaterialByName(serviceCtx, name)
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, fmt.Errorf("检查原料名称失败: %w", err)
+	}
+	if existing != nil {
+		return nil, ErrRawMaterialNameConflict
+	}
+
+	rawMaterial := &models.RawMaterial{
+		Name:        name,
+		Description: description,
+	}
+
+	if err := s.rawMaterialRepo.CreateRawMaterial(serviceCtx, rawMaterial); err != nil {
+		return nil, fmt.Errorf("创建原料失败: %w", err)
+	}
+
+	return rawMaterial, nil
+}
+
+// UpdateRawMaterial 实现了更新原料的核心业务逻辑
+func (s *rawMaterialServiceImpl) UpdateRawMaterial(ctx context.Context, id uint32, name, description string) (*models.RawMaterial, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdateRawMaterial")
+
+	// 检查要更新的实体是否存在
+	rawMaterial, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrRawMaterialNotFound
+		}
+		return nil, fmt.Errorf("获取待更新的原料失败: %w", err)
+	}
+
+	// 如果名称有变动，检查新名称是否与其它记录冲突
+	if rawMaterial.Name != name {
+		existing, err := s.rawMaterialRepo.GetRawMaterialByName(serviceCtx, name)
+		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, fmt.Errorf("检查新的原料名称失败: %w", err)
+		}
+		if existing != nil && existing.ID != id {
+			return nil, ErrRawMaterialNameConflict
+		}
+	}
+
+	rawMaterial.Name = name
+	rawMaterial.Description = description
+
+	if err := s.rawMaterialRepo.UpdateRawMaterial(serviceCtx, rawMaterial); err != nil {
+		return nil, fmt.Errorf("更新原料失败: %w", err)
+	}
+
+	return rawMaterial, nil
+}
+
+// DeleteRawMaterial 实现了删除原料的核心业务逻辑
+func (s *rawMaterialServiceImpl) DeleteRawMaterial(ctx context.Context, id uint32) error {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeleteRawMaterial")
+
+	// 检查实体是否存在
+	_, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return ErrRawMaterialNotFound
+		}
+		return fmt.Errorf("获取待删除的原料失败: %w", err)
+	}
+
+	if err := s.rawMaterialRepo.DeleteRawMaterial(serviceCtx, id); err != nil {
+		return fmt.Errorf("删除原料失败: %w", err)
+	}
+
+	return nil
+}
+
+// GetRawMaterial 实现了获取单个原料的逻辑
+func (s *rawMaterialServiceImpl) GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetRawMaterial")
+
+	rawMaterial, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrRawMaterialNotFound
+		}
+		return nil, fmt.Errorf("获取原料失败: %w", err)
+	}
+	return rawMaterial, nil
+}
+
+// ListRawMaterials 实现了列出原料的逻辑
+func (s *rawMaterialServiceImpl) ListRawMaterials(ctx context.Context, opts repository.RawMaterialListOptions, page, pageSize int) ([]models.RawMaterial, int64, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListRawMaterials")
+
+	rawMaterials, total, err := s.rawMaterialRepo.ListRawMaterials(serviceCtx, opts, page, pageSize)
+	if err != nil {
+		return nil, 0, fmt.Errorf("获取原料列表失败: %w", err)
+	}
+	return rawMaterials, total, nil
+}
+
+// UpdateRawMaterialNutrients 实现了全量更新原料营养成分的业务逻辑
+func (s *rawMaterialServiceImpl) UpdateRawMaterialNutrients(ctx context.Context, rawMaterialID uint32, nutrients []models.RawMaterialNutrient) error {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdateRawMaterialNutrients")
+
+	// 1. 检查原料是否存在
+	if _, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, rawMaterialID); err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return ErrRawMaterialNotFound
+		}
+		return fmt.Errorf("获取待更新的原料失败: %w", err)
+	}
+
+	// 2. 在事务中执行替换操作
+	err := s.uow.ExecuteInTransaction(serviceCtx, func(tx *gorm.DB) error {
+		// 2.1. 删除旧的关联记录
+		if err := s.rawMaterialRepo.DeleteNutrientsByRawMaterialIDTx(serviceCtx, tx, rawMaterialID); err != nil {
+			return err // 错误已在仓库层封装，直接返回
+		}
+
+		// 2.2. 创建新的关联记录
+		if err := s.rawMaterialRepo.CreateBatchRawMaterialNutrientsTx(serviceCtx, tx, nutrients); err != nil {
+			return err // 错误已在仓库层封装，直接返回
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return fmt.Errorf("更新原料营养成分事务执行失败: %w", err)
+	}
+
+	// 3. 操作成功，直接返回 nil
+	return nil
+}