实现使用系统中所有可用的原料一键生成配方

2025-11-26 20:44:41 +08:00
parent ba60ed541c
commit 34311889e8
12 changed files with 301 additions and 30 deletions
--- a/internal/app/api/router.go
+++ b/internal/app/api/router.go
@@ -259,6 +259,7 @@ func (a *API) setupRoutes() {
 			feedGroup.DELETE("/recipes/:id", a.recipeController.DeleteRecipe)
 			feedGroup.GET("/recipes/:id", a.recipeController.GetRecipe)
 			feedGroup.GET("/recipes", a.recipeController.ListRecipes)
+			feedGroup.POST("/recipes/generate-from-all-materials/:pig_type_id", a.recipeController.GenerateFromAllMaterials)
 		}
 		logger.Debug("饲料管理相关接口注册成功 (需要认证和审计)")

--- a/internal/app/controller/feed/recipe_controller.go
+++ b/internal/app/controller/feed/recipe_controller.go
@@ -194,3 +194,34 @@ func (c *RecipeController) ListRecipes(ctx echo.Context) error {
 	logger.Infof("%s: 获取配方列表成功, 数量: %d", actionType, len(resp.List))
 	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取配方列表成功", resp, actionType, "获取配方列表成功", resp)
 }
+
+// GenerateFromAllMaterials godoc
+// @Summary      使用系统中所有可用的原料一键生成配方
+// @Description  根据指定的猪类型ID，使用系统中所有可用的原料，自动计算并创建一个成本最优的配方。
+// @Tags         饲料管理-配方
+// @Security     BearerAuth
+// @Produce      json
+// @Param        pig_type_id path int true "猪类型ID"
+// @Success      201 {object} controller.Response{data=dto.GenerateRecipeResponse} "业务码为201代表创建成功"
+// @Router       /api/v1/feed/recipes/generate-from-all-materials/{pig_type_id} [post]
+func (c *RecipeController) GenerateFromAllMaterials(ctx echo.Context) error {
+	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "GenerateFromAllMaterials")
+	const actionType = "使用系统中所有可用的原料一键生成配方"
+
+	idStr := ctx.Param("pig_type_id")
+	id, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		logger.Errorf("%s: 猪类型ID格式错误: %v, ID: %s", actionType, err, idStr)
+		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的猪类型ID格式", actionType, "猪类型ID格式错误", idStr)
+	}
+
+	recipe, err := c.recipeService.GenerateRecipeWithAllRawMaterials(reqCtx, uint32(id))
+	if err != nil {
+		logger.Errorf("%s: 服务层生成配方失败: %v, PigTypeID: %d", actionType, err, id)
+		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "生成配方失败: "+err.Error(), actionType, "服务层生成配方失败", id)
+	}
+
+	resp := dto.ToGenerateRecipeResponse(recipe)
+	logger.Infof("%s: 配方生成成功, 新配方ID: %d", actionType, resp.ID)
+	return controller.SendSuccessWithAudit(ctx, controller.CodeCreated, "配方生成成功", resp, actionType, "配方生成成功", resp)
+}
--- a/internal/app/dto/feed_converter.go
+++ b/internal/app/dto/feed_converter.go
@@ -280,3 +280,15 @@ func ConvertUpdateRecipeRequestToModel(req *UpdateRecipeRequest) *models.Recipe
 		RecipeIngredients: ingredients,
 	}
 }
+
+// ToGenerateRecipeResponse 将 models.Recipe 转换为 GenerateRecipeResponse DTO
+func ToGenerateRecipeResponse(recipe *models.Recipe) *GenerateRecipeResponse {
+	if recipe == nil {
+		return nil
+	}
+	return &GenerateRecipeResponse{
+		ID:          recipe.ID,
+		Name:        recipe.Name,
+		Description: recipe.Description,
+	}
+}
--- a/internal/app/dto/feed_dto.go
+++ b/internal/app/dto/feed_dto.go
@@ -325,3 +325,10 @@ type ListRecipeResponse struct {
 	List       []RecipeResponse `json:"list"`
 	Pagination PaginationDTO    `json:"pagination"`
 }
+
+// GenerateRecipeResponse 是一键生成配方的响应体
+type GenerateRecipeResponse struct {
+	ID          uint32 `json:"id"`          // 新生成的配方ID
+	Name        string `json:"name"`        // 新生成的配方名称
+	Description string `json:"description"` // 新生成的配方描述
+}
--- a/internal/app/service/recipe_service.go
+++ b/internal/app/service/recipe_service.go
@@ -25,22 +25,31 @@ type RecipeService interface {
 	DeleteRecipe(ctx context.Context, id uint32) error
 	GetRecipeByID(ctx context.Context, id uint32) (*dto.RecipeResponse, error)
 	ListRecipes(ctx context.Context, req *dto.ListRecipeRequest) (*dto.ListRecipeResponse, error)
+	// GenerateRecipeWithAllRawMaterials 添加新方法
+	GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
 }

 // recipeServiceImpl 是 RecipeService 接口的实现
 type recipeServiceImpl struct {
 	ctx       context.Context
-	recipeSvc recipe.RecipeCoreService
+	recipeSvc recipe.Service
 }

 // NewRecipeService 创建一个新的 RecipeService 实例
-func NewRecipeService(ctx context.Context, recipeSvc recipe.RecipeCoreService) RecipeService {
+func NewRecipeService(ctx context.Context, recipeSvc recipe.Service) RecipeService {
 	return &recipeServiceImpl{
 		ctx:       ctx,
 		recipeSvc: recipeSvc,
 	}
 }

+// GenerateRecipeWithAllRawMaterials 实现新方法
+func (s *recipeServiceImpl) GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GenerateRecipeWithAllRawMaterials")
+	// 直接调用领域服务的方法
+	return s.recipeSvc.GenerateRecipeWithAllRawMaterials(serviceCtx, pigTypeID)
+}
+
 // CreateRecipe 创建配方
 func (s *recipeServiceImpl) CreateRecipe(ctx context.Context, req *dto.CreateRecipeRequest) (*dto.RecipeResponse, error) {
 	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateRecipe")
--- a/internal/core/component_initializers.go
+++ b/internal/core/component_initializers.go
@@ -228,6 +228,7 @@ func initDomainServices(ctx context.Context, cfg *config.Config, infra *Infrastr
 	pigTypeService := recipe.NewPigTypeService(logs.AddCompName(baseCtx, "PigTypeService"), infra.repos.unitOfWork, infra.repos.pigTypeRepo)
 	rawMaterialService := recipe.NewRawMaterialService(logs.AddCompName(baseCtx, "RawMaterialService"), infra.repos.unitOfWork, infra.repos.rawMaterialRepo, inventoryService)
 	recipeCoreService := recipe.NewRecipeCoreService(logs.AddCompName(baseCtx, "RecipeCoreService"), infra.repos.unitOfWork, infra.repos.recipeRepo)
+	recipeGenerateManager := recipe.NewRecipeGenerateManager(logs.AddCompName(baseCtx, "RecipeGenerateManager"))
 	recipeService := recipe.NewRecipeService(
 		logs.AddCompName(baseCtx, "RecipeService"),
 		nutrientService,
@@ -236,6 +237,7 @@ func initDomainServices(ctx context.Context, cfg *config.Config, infra *Infrastr
 		pigAgeStageService,
 		pigTypeService,
 		recipeCoreService,
+		recipeGenerateManager,
 	)

 	return &DomainServices{
--- a/internal/domain/recipe/recipe_generate_manager.go
+++ b/internal/domain/recipe/recipe_generate_manager.go
@@ -7,6 +7,7 @@ import (
 	"math"

 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+
 	"gonum.org/v1/gonum/mat"
 	"gonum.org/v1/gonum/optimize/convex/lp"
 )
--- a/internal/domain/recipe/recipe_service.go
+++ b/internal/domain/recipe/recipe_service.go
@@ -2,6 +2,10 @@ package recipe

 import (
 	"context"
+	"fmt"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
 )

 // Service 定义了配方与原料领域的核心业务服务接口
@@ -13,6 +17,8 @@ type Service interface {
 	PigAgeStageService
 	PigTypeService
 	RecipeCoreService
+	RecipeGenerateManager
+	GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
 }

 // recipeServiceImpl 是 Service 的实现，通过组合各个子服务来实现
@@ -24,6 +30,7 @@ type recipeServiceImpl struct {
 	PigAgeStageService
 	PigTypeService
 	RecipeCoreService
+	RecipeGenerateManager
 }

 // NewRecipeService 创建一个新的 Service 实例
@@ -35,14 +42,50 @@ func NewRecipeService(
 	pigAgeStageService PigAgeStageService,
 	pigTypeService PigTypeService,
 	recipeCoreService RecipeCoreService,
+	recipeGenerateManager RecipeGenerateManager,
 ) Service {
 	return &recipeServiceImpl{
-		ctx:                ctx,
-		NutrientService:    nutrientService,
-		RawMaterialService: rawMaterialService,
-		PigBreedService:    pigBreedService,
-		PigAgeStageService: pigAgeStageService,
-		PigTypeService:     pigTypeService,
-		RecipeCoreService:  recipeCoreService,
+		ctx:                   ctx,
+		NutrientService:       nutrientService,
+		RawMaterialService:    rawMaterialService,
+		PigBreedService:       pigBreedService,
+		PigAgeStageService:    pigAgeStageService,
+		PigTypeService:        pigTypeService,
+		RecipeCoreService:     recipeCoreService,
+		RecipeGenerateManager: recipeGenerateManager,
 	}
 }
+
+// GenerateRecipeWithAllRawMaterials 使用所有已知原料为特定猪类型生成一个新配方。
+// pigTypeID: 目标猪类型的ID。
+// 返回: 生成的配方对象指针和可能的错误。
+func (r *recipeServiceImpl) GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error) {
+	// 1. 获取猪只类型信息，确保包含了营养需求
+	pigType, err := r.GetPigTypeByID(ctx, pigTypeID)
+	if err != nil {
+		return nil, fmt.Errorf("获取猪类型信息失败: %w", err)
+	}
+
+	// 2. 获取所有原料
+	// 我们通过传递一个非常大的 pageSize 来获取所有原料，这在大多数情况下是可行的。
+	// 对于超大规模系统，可能需要考虑分页迭代，但目前这是一个简单有效的策略。
+	materials, _, err := r.ListRawMaterials(ctx, repository.RawMaterialListOptions{}, 1, 9999)
+	if err != nil {
+		return nil, fmt.Errorf("获取所有原料列表失败: %w", err)
+	}
+
+	// 3. 调用生成器生成配方
+	recipe, err := r.GenerateRecipe(ctx, *pigType, materials)
+	if err != nil {
+		return nil, fmt.Errorf("生成配方失败: %w", err)
+	}
+
+	// 4. 保存新生成的配方到数据库
+	// CreateRecipe 会处理配方及其成分的保存
+	if recipe, err = r.CreateRecipe(ctx, recipe); err != nil {
+		return nil, fmt.Errorf("保存生成的配方失败: %w", err)
+	}
+
+	// 5. 返回创建的配方 (现在它应该已经有了ID)
+	return recipe, nil
+}