实现优先使用库存的配方一键生成

internal/app/api/router.go

@@ -260,6 +260,7 @@ func (a *API) setupRoutes() {
 			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)
+			feedGroup.POST("/recipes/generate-prioritized-stock/:pig_type_id", a.recipeController.GenerateFromAllMaterials)
 		}
 		logger.Debug("饲料管理相关接口注册成功 (需要认证和审计)")
 

internal/app/controller/feed/recipe_controller.go

@@ -225,3 +225,34 @@ func (c *RecipeController) GenerateFromAllMaterials(ctx echo.Context) error {
 	logger.Infof("%s: 配方生成成功, 新配方ID: %d", actionType, resp.ID)
 	return controller.SendSuccessWithAudit(ctx, controller.CodeCreated, "配方生成成功", resp, actionType, "配方生成成功", resp)
 }
+
+// GenerateRecipeWithPrioritizedStockRawMaterials 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-prioritized-stock/{pig_type_id} [post]
+func (c *RecipeController) GenerateRecipeWithPrioritizedStockRawMaterials(ctx echo.Context) error {
+	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "GenerateRecipeWithPrioritizedStockRawMaterials")
+	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.GenerateRecipeWithPrioritizedStockRawMaterials(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/service/recipe_service.go

@@ -27,6 +27,8 @@ type RecipeService interface {
 	ListRecipes(ctx context.Context, req *dto.ListRecipeRequest) (*dto.ListRecipeResponse, error)
 	// GenerateRecipeWithAllRawMaterials 添加新方法
 	GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
+	// GenerateRecipeWithPrioritizedStockRawMaterials 生成新配方，优先使用有库存的原料
+	GenerateRecipeWithPrioritizedStockRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
 }
 
 // recipeServiceImpl 是 RecipeService 接口的实现
@@ -50,6 +52,13 @@ func (s *recipeServiceImpl) GenerateRecipeWithAllRawMaterials(ctx context.Contex
 	return s.recipeSvc.GenerateRecipeWithAllRawMaterials(serviceCtx, pigTypeID)
 }
 
+// GenerateRecipeWithPrioritizedStockRawMaterials 实现生成优先使用库存原料配方的方法
+func (s *recipeServiceImpl) GenerateRecipeWithPrioritizedStockRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error) {
+	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GenerateRecipeWithPrioritizedStockRawMaterials")
+	// 直接调用领域服务的方法
+	return s.recipeSvc.GenerateRecipeWithPrioritizedStockRawMaterials(serviceCtx, pigTypeID)
+}
+
 // CreateRecipe 创建配方
 func (s *recipeServiceImpl) CreateRecipe(ctx context.Context, req *dto.CreateRecipeRequest) (*dto.RecipeResponse, error) {
 	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateRecipe")

internal/domain/recipe/recipe_service.go

@@ -20,6 +20,8 @@ type Service interface {
 	RecipeCoreService
 	RecipeGenerateManager
 	GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
+	// GenerateRecipeWithPrioritizedStockRawMaterials 生成新配方，优先使用有库存的原料
+	GenerateRecipeWithPrioritizedStockRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
 }
 
 // recipeServiceImpl 是 Service 的实现，通过组合各个子服务来实现
@@ -123,3 +125,99 @@ func (r *recipeServiceImpl) GenerateRecipeWithAllRawMaterials(ctx context.Contex
 	// 6. 返回创建的配方 (现在它应该已经有了ID)
 	return recipe, nil
 }
+
+// GenerateRecipeWithPrioritizedStockRawMaterials 使用优先有库存原料的策略为特定猪类型生成一个新配方。
+// 通过大幅调低有库存原料的参考价格，诱导生成器优先使用。
+// pigTypeID: 目标猪类型的ID。
+// 返回: 生成的配方对象指针和可能的错误。
+func (r *recipeServiceImpl) GenerateRecipeWithPrioritizedStockRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error) {
+	serviceCtx, logger := logs.Trace(ctx, r.ctx, "GenerateRecipeWithPrioritizedStockRawMaterials")
+
+	// 1. 获取猪只类型信息，确保包含了营养需求
+	pigType, err := r.GetPigTypeByID(serviceCtx, pigTypeID)
+	if err != nil {
+		return nil, fmt.Errorf("获取猪类型信息失败: %w", err)
+	}
+
+	// 2. 获取所有原料，并区分有库存和无库存的原料
+	// 获取有库存的原料
+	hasStock := true
+	stockOpts := repository.RawMaterialListOptions{HasStock: &hasStock}
+	stockMaterials, _, err := r.ListRawMaterials(serviceCtx, stockOpts, 1, 9999)
+	if err != nil {
+		return nil, fmt.Errorf("获取有库存原料列表失败: %w", err)
+	}
+
+	// 获取无库存的原料
+	hasStock = false
+	noStockOpts := repository.RawMaterialListOptions{HasStock: &hasStock}
+	noStockMaterials, _, err := r.ListRawMaterials(serviceCtx, noStockOpts, 1, 9999)
+	if err != nil {
+		return nil, fmt.Errorf("获取无库存原料列表失败: %w", err)
+	}
+
+	// 合并有库存和无库存的原料，作为所有原始原料的列表，用于后续计算最终参考价格
+	allOriginalMaterials := make([]models.RawMaterial, 0, len(stockMaterials)+len(noStockMaterials))
+	allOriginalMaterials = append(allOriginalMaterials, stockMaterials...)
+	allOriginalMaterials = append(allOriginalMaterials, noStockMaterials...)
+
+	// 3. 创建一个用于配方生成的原料列表，并调整有库存原料的价格
+	var materialsForGeneration []models.RawMaterial
+
+	// 先添加有库存的原料，并调整价格
+	for _, mat := range stockMaterials {
+		adjustedMat := mat // 复制一份
+		// 大幅调低有库存原料的参考价格，诱导生成器优先使用
+		adjustedMat.ReferencePrice = 0.01 // 设置一个非常小的价格
+		materialsForGeneration = append(materialsForGeneration, adjustedMat)
+		logger.Debugf("原料 '%s' (ID: %d) 有库存，生成配方时参考价格调整为 %.2f", mat.Name, mat.ID, adjustedMat.ReferencePrice)
+	}
+	// 再添加无库存的原料，保持原价
+	for _, mat := range noStockMaterials {
+		materialsForGeneration = append(materialsForGeneration, mat)
+	}
+
+	// 4. 调用生成器生成配方
+	recipe, err := r.GenerateRecipe(serviceCtx, *pigType, materialsForGeneration)
+	if err != nil {
+		return nil, fmt.Errorf("生成配方失败: %w", err)
+	}
+
+	// 5. 丰富配方描述：计算并添加参考价格信息
+	// 注意：这里需要使用原始的、未调整价格的原料信息来计算最终的参考价格
+	// rawMaterialMap 从 allOriginalMaterials 构建，确保使用原始价格
+	rawMaterialMap := make(map[uint32]models.RawMaterial)
+	for _, mat := range allOriginalMaterials {
+		rawMaterialMap[mat.ID] = mat
+	}
+
+	// 填充 RecipeIngredients 中的 RawMaterial 字段，以便后续计算成本
+	for i := range recipe.RecipeIngredients {
+		if rawMat, ok := rawMaterialMap[recipe.RecipeIngredients[i].RawMaterialID]; ok {
+			recipe.RecipeIngredients[i].RawMaterial = rawMat
+		} else {
+			logger.Warnf("未找到 RecipeIngredient (RawMaterialID: %d) 对应的原始 RawMaterial，成本计算可能不准确", recipe.RecipeIngredients[i].RawMaterialID)
+		}
+	}
+
+	referencePrice := recipe.CalculateReferencePricePerKilogram() / 100
+	recipe.Description = fmt.Sprintf("%s 计算时预估成本: %.2f元/kg。", recipe.Description, referencePrice)
+
+	// 如果 totalPercentage 小于 100%，说明填充料被使用，这是符合预期的。
+	// 此时需要在描述中说明需要添加的廉价填充料的百分比。
+	totalPercentage := recipe.CalculateTotalRawMaterialProportion()
+	if totalPercentage < 99.99 { // 允许微小的浮点误差
+		fillerPercentage := 100 - totalPercentage
+		recipe.Description = fmt.Sprintf("%s 注意：配方中实际原料占比 %.2f%%，需额外补充 %.2f%% 廉价填充料", recipe.Description, totalPercentage, fillerPercentage)
+	}
+
+	// 6. 保存新生成的配方到数据库
+	if recipe, err = r.CreateRecipe(serviceCtx, recipe); err != nil {
+
+		return nil, fmt.Errorf("保存生成的配方失败: %w", err)
+	}
+	logger.Infof("成功生成优先使用库存原料的配方: %+v", recipe)
+
+	// 7. 返回创建的配方
+	return recipe, nil
+}