pig/pig-farm-controller
1
0
Fork 0
Code Issues 16 Pull Requests Actions Packages Projects Releases Wiki Activity

实现原材料的增删改查和仓库层的原料库存记录表增查

Browse Source
This commit is contained in:
huang
2025-11-20 13:43:09 +08:00
parent fd0939fe0a
commit c697e668e3
5 changed files with 296 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
internal/domain/recipe/recipe_service.go
View File

@@ -2,11 +2,14 @@ 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"
)
// 定义领域特定的错误
@@ -14,28 +17,41 @@ var (
ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在")
ErrNutrientNotFound = fmt.Errorf("营养种类不存在")
ErrNutrientInUse = fmt.Errorf("营养种类正在被原料使用，无法删除")
ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在")
ErrRawMaterialNotFound = fmt.Errorf("原料不存在")
)
// Service 定义了配方与原料领域的核心业务服务接口
type Service interface {
// 营养种类相关接口
CreateNutrient(ctx context.Context, name, description string) (*models.Nutrient, error)
UpdateNutrient(ctx context.Context, id uint32, name, description string) (*models.Nutrient, error)
DeleteNutrient(ctx context.Context, id uint32) error
GetNutrient(ctx context.Context, id uint32) (*models.Nutrient, error)
ListNutrients(ctx context.Context, page, pageSize int) ([]models.Nutrient, int64, error)
// 原料相关接口
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, page, pageSize int) ([]models.RawMaterial, int64, error)
}
// recipeServiceImpl 是 RecipeService 的实现
type recipeServiceImpl struct {
ctx context.Context
nutrientRepo repository.NutrientRepository
ctx context.Context
nutrientRepo repository.NutrientRepository
rawMaterialRepo repository.RawMaterialRepository
}
// NewRecipeService 创建一个新的 RecipeService 实例
func NewRecipeService(ctx context.Context, nutrientRepo repository.NutrientRepository) Service {
func NewRecipeService(ctx context.Context, nutrientRepo repository.NutrientRepository, rawMaterialRepo repository.RawMaterialRepository) Service {
return &recipeServiceImpl{
ctx: ctx,
nutrientRepo: nutrientRepo,
ctx: ctx,
nutrientRepo: nutrientRepo,
rawMaterialRepo: rawMaterialRepo,
}
}
@@ -45,7 +61,7 @@ func (s *recipeServiceImpl) CreateNutrient(ctx context.Context, name, descriptio
// 检查名称是否已存在
existing, err := s.nutrientRepo.GetNutrientByName(serviceCtx, name)
if err != nil {
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) { // 只有不是记录未找到的错误才返回
return nil, fmt.Errorf("检查营养种类名称失败: %w", err)
}
if existing != nil {
@@ -71,16 +87,16 @@ func (s *recipeServiceImpl) UpdateNutrient(ctx context.Context, id uint32, name,
// 检查要更新的实体是否存在
nutrient, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) { // 如果是记录未找到错误，则返回领域错误
return nil, ErrNutrientNotFound
}
return nil, fmt.Errorf("获取待更新的营养种类失败: %w", err)
}
if nutrient == nil {
return nil, ErrNutrientNotFound
}
// 如果名称有变动，检查新名称是否与其它记录冲突
if nutrient.Name != name {
existing, err := s.nutrientRepo.GetNutrientByName(serviceCtx, name)
if err != nil {
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, fmt.Errorf("检查新的营养种类名称失败: %w", err)
}
if existing != nil && existing.ID != id {
@@ -103,13 +119,13 @@ func (s *recipeServiceImpl) DeleteNutrient(ctx context.Context, id uint32) error
serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeleteNutrient")
// 检查实体是否存在
nutrient, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
_, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrNutrientNotFound
}
return fmt.Errorf("获取待删除的营养种类失败: %w", err)
}
if nutrient == nil {
return ErrNutrientNotFound
}
if err := s.nutrientRepo.DeleteNutrient(serviceCtx, id); err != nil {
return fmt.Errorf("删除营养种类失败: %w", err)
@@ -124,11 +140,11 @@ func (s *recipeServiceImpl) GetNutrient(ctx context.Context, id uint32) (*models
nutrient, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrNutrientNotFound
}
return nil, fmt.Errorf("获取营养种类失败: %w", err)
}
if nutrient == nil {
return nil, ErrNutrientNotFound
}
return nutrient, nil
}
@@ -142,3 +158,107 @@ func (s *recipeServiceImpl) ListNutrients(ctx context.Context, page, pageSize in
}
return nutrients, total, nil
}
// CreateRawMaterial 实现了创建原料的核心业务逻辑
func (s *recipeServiceImpl) 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 *recipeServiceImpl) 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 *recipeServiceImpl) 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 *recipeServiceImpl) 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 *recipeServiceImpl) ListRawMaterials(ctx context.Context, page, pageSize int) ([]models.RawMaterial, int64, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListRawMaterials")
rawMaterials, total, err := s.rawMaterialRepo.ListRawMaterials(serviceCtx, page, pageSize)
if err != nil {
return nil, 0, fmt.Errorf("获取原料列表失败: %w", err)
}
return rawMaterials, total, nil
}