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

design/recipe-management/index.md

@@ -63,4 +63,5 @@ http://git.huangwc.com/pig/pig-farm-controller/issues/66
 13. 重构配方领域
 14. 配方增删改查服务层和控制器
 15. 实现库存管理相关逻辑
-16. 实现配方生成器
+16. 实现配方生成器
+17. 实现使用系统中所有可用的原料一键生成配方

docs/docs.go

@@ -3145,6 +3145,52 @@ const docTemplate = `{
                 }
             }
         },
+        "/api/v1/feed/recipes/generate-from-all-materials/{pig_type_id}": {
+            "post": {
+                "security": [
+                    {
+                        "BearerAuth": []
+                    }
+                ],
+                "description": "根据指定的猪类型ID，使用系统中所有可用的原料，自动计算并创建一个成本最优的配方。",
+                "produces": [
+                    "application/json"
+                ],
+                "tags": [
+                    "饲料管理-配方"
+                ],
+                "summary": "使用系统中所有可用的原料一键生成配方",
+                "parameters": [
+                    {
+                        "type": "integer",
+                        "description": "猪类型ID",
+                        "name": "pig_type_id",
+                        "in": "path",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "201": {
+                        "description": "业务码为201代表创建成功",
+                        "schema": {
+                            "allOf": [
+                                {
+                                    "$ref": "#/definitions/controller.Response"
+                                },
+                                {
+                                    "type": "object",
+                                    "properties": {
+                                        "data": {
+                                            "$ref": "#/definitions/dto.GenerateRecipeResponse"
+                                        }
+                                    }
+                                }
+                            ]
+                        }
+                    }
+                }
+            }
+        },
         "/api/v1/feed/recipes/{id}": {
             "get": {
                 "security": [
@@ -3669,7 +3715,6 @@ const docTemplate = `{
                     },
                     {
                         "enum": [
-                            7,
                             -1,
                             0,
                             1,
@@ -3679,12 +3724,12 @@ const docTemplate = `{
                             5,
                             -1,
                             5,
-                            6
+                            6,
+                            7
                         ],
                         "type": "integer",
                         "format": "int32",
                         "x-enum-varnames": [
-                            "_numLevels",
                             "DebugLevel",
                             "InfoLevel",
                             "WarnLevel",
@@ -3694,7 +3739,8 @@ const docTemplate = `{
                             "FatalLevel",
                             "_minLevel",
                             "_maxLevel",
-                            "InvalidLevel"
+                            "InvalidLevel",
+                            "_numLevels"
                         ],
                         "name": "level",
                         "in": "query"
@@ -7439,6 +7485,23 @@ const docTemplate = `{
                 }
             }
         },
+        "dto.GenerateRecipeResponse": {
+            "type": "object",
+            "properties": {
+                "description": {
+                    "description": "新生成的配方描述",
+                    "type": "string"
+                },
+                "id": {
+                    "description": "新生成的配方ID",
+                    "type": "integer"
+                },
+                "name": {
+                    "description": "新生成的配方名称",
+                    "type": "string"
+                }
+            }
+        },
         "dto.HistoricalAlarmDTO": {
             "type": "object",
             "properties": {
@@ -10523,7 +10586,6 @@ const docTemplate = `{
             "type": "integer",
             "format": "int32",
             "enum": [
-                7,
                 -1,
                 0,
                 1,
@@ -10533,10 +10595,10 @@ const docTemplate = `{
                 5,
                 -1,
                 5,
-                6
+                6,
+                7
             ],
             "x-enum-varnames": [
-                "_numLevels",
                 "DebugLevel",
                 "InfoLevel",
                 "WarnLevel",
@@ -10546,7 +10608,8 @@ const docTemplate = `{
                 "FatalLevel",
                 "_minLevel",
                 "_maxLevel",
-                "InvalidLevel"
+                "InvalidLevel",
+                "_numLevels"
             ]
         }
     },

docs/swagger.json

@@ -3137,6 +3137,52 @@
                 }
             }
         },
+        "/api/v1/feed/recipes/generate-from-all-materials/{pig_type_id}": {
+            "post": {
+                "security": [
+                    {
+                        "BearerAuth": []
+                    }
+                ],
+                "description": "根据指定的猪类型ID，使用系统中所有可用的原料，自动计算并创建一个成本最优的配方。",
+                "produces": [
+                    "application/json"
+                ],
+                "tags": [
+                    "饲料管理-配方"
+                ],
+                "summary": "使用系统中所有可用的原料一键生成配方",
+                "parameters": [
+                    {
+                        "type": "integer",
+                        "description": "猪类型ID",
+                        "name": "pig_type_id",
+                        "in": "path",
+                        "required": true
+                    }
+                ],
+                "responses": {
+                    "201": {
+                        "description": "业务码为201代表创建成功",
+                        "schema": {
+                            "allOf": [
+                                {
+                                    "$ref": "#/definitions/controller.Response"
+                                },
+                                {
+                                    "type": "object",
+                                    "properties": {
+                                        "data": {
+                                            "$ref": "#/definitions/dto.GenerateRecipeResponse"
+                                        }
+                                    }
+                                }
+                            ]
+                        }
+                    }
+                }
+            }
+        },
         "/api/v1/feed/recipes/{id}": {
             "get": {
                 "security": [
@@ -3661,7 +3707,6 @@
                     },
                     {
                         "enum": [
-                            7,
                             -1,
                             0,
                             1,
@@ -3671,12 +3716,12 @@
                             5,
                             -1,
                             5,
-                            6
+                            6,
+                            7
                         ],
                         "type": "integer",
                         "format": "int32",
                         "x-enum-varnames": [
-                            "_numLevels",
                             "DebugLevel",
                             "InfoLevel",
                             "WarnLevel",
@@ -3686,7 +3731,8 @@
                             "FatalLevel",
                             "_minLevel",
                             "_maxLevel",
-                            "InvalidLevel"
+                            "InvalidLevel",
+                            "_numLevels"
                         ],
                         "name": "level",
                         "in": "query"
@@ -7431,6 +7477,23 @@
                 }
             }
         },
+        "dto.GenerateRecipeResponse": {
+            "type": "object",
+            "properties": {
+                "description": {
+                    "description": "新生成的配方描述",
+                    "type": "string"
+                },
+                "id": {
+                    "description": "新生成的配方ID",
+                    "type": "integer"
+                },
+                "name": {
+                    "description": "新生成的配方名称",
+                    "type": "string"
+                }
+            }
+        },
         "dto.HistoricalAlarmDTO": {
             "type": "object",
             "properties": {
@@ -10515,7 +10578,6 @@
             "type": "integer",
             "format": "int32",
             "enum": [
-                7,
                 -1,
                 0,
                 1,
@@ -10525,10 +10587,10 @@
                 5,
                 -1,
                 5,
-                6
+                6,
+                7
             ],
             "x-enum-varnames": [
-                "_numLevels",
                 "DebugLevel",
                 "InfoLevel",
                 "WarnLevel",
@@ -10538,7 +10600,8 @@
                 "FatalLevel",
                 "_minLevel",
                 "_maxLevel",
-                "InvalidLevel"
+                "InvalidLevel",
+                "_numLevels"
             ]
         }
     },

docs/swagger.yaml

@@ -565,6 +565,18 @@ definitions:
       thresholds:
         type: number
     type: object
+  dto.GenerateRecipeResponse:
+    properties:
+      description:
+        description: 新生成的配方描述
+        type: string
+      id:
+        description: 新生成的配方ID
+        type: integer
+      name:
+        description: 新生成的配方名称
+        type: string
+    type: object
   dto.HistoricalAlarmDTO:
     properties:
       alarm_code:
@@ -2719,7 +2731,6 @@ definitions:
     - PlanTypeFilterSystem
   zapcore.Level:
     enum:
-    - 7
     - -1
     - 0
     - 1
@@ -2730,10 +2741,10 @@ definitions:
     - -1
     - 5
     - 6
+    - 7
     format: int32
     type: integer
     x-enum-varnames:
-    - _numLevels
     - DebugLevel
     - InfoLevel
     - WarnLevel
@@ -2744,6 +2755,7 @@ definitions:
     - _minLevel
     - _maxLevel
     - InvalidLevel
+    - _numLevels
 info:
   contact:
     email: divano@example.com
@@ -4722,6 +4734,32 @@ paths:
       summary: 更新配方
       tags:
       - 饲料管理-配方
+  /api/v1/feed/recipes/generate-from-all-materials/{pig_type_id}:
+    post:
+      description: 根据指定的猪类型ID，使用系统中所有可用的原料，自动计算并创建一个成本最优的配方。
+      parameters:
+      - description: 猪类型ID
+        in: path
+        name: pig_type_id
+        required: true
+        type: integer
+      produces:
+      - application/json
+      responses:
+        "201":
+          description: 业务码为201代表创建成功
+          schema:
+            allOf:
+            - $ref: '#/definitions/controller.Response'
+            - properties:
+                data:
+                  $ref: '#/definitions/dto.GenerateRecipeResponse'
+              type: object
+      security:
+      - BearerAuth: []
+      summary: 使用系统中所有可用的原料一键生成配方
+      tags:
+      - 饲料管理-配方
   /api/v1/inventory/stock/adjust:
     post:
       consumes:
@@ -4947,7 +4985,6 @@ paths:
         name: end_time
         type: string
       - enum:
-        - 7
         - -1
         - 0
         - 1
@@ -4958,12 +4995,12 @@ paths:
         - -1
         - 5
         - 6
+        - 7
         format: int32
         in: query
         name: level
         type: integer
         x-enum-varnames:
-        - _numLevels
         - DebugLevel
         - InfoLevel
         - WarnLevel
@@ -4974,6 +5011,7 @@ paths:
         - _minLevel
         - _maxLevel
         - InvalidLevel
+        - _numLevels
       - enum:
         - 邮件
         - 企业微信

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("饲料管理相关接口注册成功 (需要认证和审计)")
 

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)
+}

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,
+	}
+}

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"` // 新生成的配方描述
+}

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")

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{

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"
 )

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,
+		ctx:                   ctx,
-		NutrientService:    nutrientService,
+		NutrientService:       nutrientService,
-		RawMaterialService: rawMaterialService,
+		RawMaterialService:    rawMaterialService,
-		PigBreedService:    pigBreedService,
+		PigBreedService:       pigBreedService,
-		PigAgeStageService: pigAgeStageService,
+		PigAgeStageService:    pigAgeStageService,
-		PigTypeService:     pigTypeService,
+		PigTypeService:        pigTypeService,
-		RecipeCoreService:  recipeCoreService,
+		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
+}