350 lines
14 KiB
Go
350 lines
14 KiB
Go
package recipe
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"git.huangwc.com/pig/pig-farm-controller/internal/infra/ai"
|
||
"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"
|
||
)
|
||
|
||
// Service 定义了配方与原料领域的核心业务服务接口
|
||
// 该接口聚合了所有子领域的服务接口
|
||
type Service interface {
|
||
NutrientService
|
||
RawMaterialService
|
||
PigBreedService
|
||
PigAgeStageService
|
||
PigTypeService
|
||
RecipeCoreService
|
||
RecipeGenerateManager
|
||
GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
|
||
// GenerateRecipeWithPrioritizedStockRawMaterials 生成新配方，优先使用有库存的原料
|
||
GenerateRecipeWithPrioritizedStockRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error)
|
||
// AIDiagnoseRecipe 智能诊断配方, 返回智能诊断结果
|
||
AIDiagnoseRecipe(ctx context.Context, recipeID uint32, pigTypeID uint32) (string, models.AIModel, error)
|
||
}
|
||
|
||
// recipeServiceImpl 是 Service 的实现，通过组合各个子服务来实现
|
||
type recipeServiceImpl struct {
|
||
ctx context.Context
|
||
NutrientService
|
||
RawMaterialService
|
||
PigBreedService
|
||
PigAgeStageService
|
||
PigTypeService
|
||
RecipeCoreService
|
||
RecipeGenerateManager
|
||
|
||
recipeRepo repository.RecipeRepository
|
||
ai ai.AI
|
||
}
|
||
|
||
// NewRecipeService 创建一个新的 Service 实例
|
||
func NewRecipeService(
|
||
ctx context.Context,
|
||
nutrientService NutrientService,
|
||
rawMaterialService RawMaterialService,
|
||
pigBreedService PigBreedService,
|
||
pigAgeStageService PigAgeStageService,
|
||
pigTypeService PigTypeService,
|
||
recipeCoreService RecipeCoreService,
|
||
recipeGenerateManager RecipeGenerateManager,
|
||
recipeRepo repository.RecipeRepository,
|
||
ai ai.AI,
|
||
) Service {
|
||
return &recipeServiceImpl{
|
||
ctx: ctx,
|
||
NutrientService: nutrientService,
|
||
RawMaterialService: rawMaterialService,
|
||
PigBreedService: pigBreedService,
|
||
PigAgeStageService: pigAgeStageService,
|
||
PigTypeService: pigTypeService,
|
||
RecipeCoreService: recipeCoreService,
|
||
RecipeGenerateManager: recipeGenerateManager,
|
||
recipeRepo: recipeRepo,
|
||
ai: ai,
|
||
}
|
||
}
|
||
|
||
// GenerateRecipeWithAllRawMaterials 使用所有已知原料为特定猪类型生成一个新配方。
|
||
// pigTypeID: 目标猪类型的ID。
|
||
// 返回: 生成的配方对象指针和可能的错误。
|
||
func (r *recipeServiceImpl) GenerateRecipeWithAllRawMaterials(ctx context.Context, pigTypeID uint32) (*models.Recipe, error) {
|
||
serviceCtx, logger := logs.Trace(ctx, r.ctx, "GenerateRecipeWithAllRawMaterials")
|
||
|
||
// 1. 获取猪只类型信息，确保包含了营养需求
|
||
pigType, err := r.GetPigTypeByID(serviceCtx, pigTypeID)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取猪类型信息失败: %w", err)
|
||
}
|
||
|
||
// 2. 获取所有原料
|
||
// 我们通过传递一个非常大的 pageSize 来获取所有原料，这在大多数情况下是可行的。
|
||
// 对于超大规模系统，可能需要考虑分页迭代，但目前这是一个简单有效的策略。
|
||
materials, _, err := r.ListRawMaterials(serviceCtx, repository.RawMaterialListOptions{}, 1, 9999)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取所有原料列表失败: %w", err)
|
||
}
|
||
|
||
// 3. 调用生成器生成配方
|
||
recipe, err := r.GenerateRecipe(serviceCtx, *pigType, materials)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("生成配方失败: %w", err)
|
||
}
|
||
|
||
// 4. 丰富配方描述：计算并添加参考价格信息
|
||
recipe.Name = fmt.Sprintf("%s - 使用所有已知原料", recipe.Name)
|
||
|
||
// 填充 RecipeIngredients 中的 RawMaterial 字段，以便后续计算成本
|
||
rawMaterialMap := make(map[uint32]models.RawMaterial)
|
||
for _, mat := range materials {
|
||
rawMaterialMap[mat.ID] = mat
|
||
}
|
||
for i := range recipe.RecipeIngredients {
|
||
if rawMat, ok := rawMaterialMap[recipe.RecipeIngredients[i].RawMaterialID]; ok {
|
||
recipe.RecipeIngredients[i].RawMaterial = rawMat
|
||
} else {
|
||
// 理论上 GenerateRecipe 应该只使用传入的 materials 中的 RawMaterialID
|
||
// 如果出现此情况，说明 GenerateRecipe 生成了不在当前 materials 列表中的 RawMaterialID
|
||
// 这可能是一个数据不一致或逻辑错误，记录警告以便排查
|
||
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)
|
||
|
||
}
|
||
|
||
// 5. 保存新生成的配方到数据库
|
||
// CreateRecipe 会处理配方及其成分的保存
|
||
if recipe, err = r.CreateRecipe(serviceCtx, recipe); err != nil {
|
||
return nil, fmt.Errorf("保存生成的配方失败: %w", err)
|
||
}
|
||
logger.Infof("成功生成配方: 配方名称: %v | 配方简介: %v", recipe.Name, recipe.Description)
|
||
|
||
// 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 // 复制一份
|
||
// 大幅调低有库存原料的参考价格，诱导生成器优先使用
|
||
// TODO 按理说应该尽量优先使用已有原料, 但如果搭配后购买缺失原料花的钱还不如不用已有原料的另一个组合钱少怎么办
|
||
adjustedMat.ReferencePrice = adjustedMat.ReferencePrice * 0.1
|
||
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. 丰富配方描述：计算并添加参考价格信息
|
||
recipe.Name = fmt.Sprintf("%s - 优先使用库存已有原料", recipe.Name)
|
||
|
||
// 注意：这里需要使用原始的、未调整价格的原料信息来计算最终的参考价格
|
||
// 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("使用 %v 种有库存原料和 %v 种无库存原料计算的库存原料优先使用的配方。 计算时预估成本: %.2f元/kg。", len(stockMaterials), len(noStockMaterials), 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 | 配方简介: %v", recipe.Name, recipe.Description)
|
||
|
||
// 7. 返回创建的配方
|
||
return recipe, nil
|
||
}
|
||
|
||
// AIDiagnoseRecipe 使用 AI 为指定食谱生成诊断。
|
||
func (s *recipeServiceImpl) AIDiagnoseRecipe(ctx context.Context, recipeID uint32, pigTypeID uint32) (string, models.AIModel, error) {
|
||
serviceCtx, logger := logs.Trace(ctx, context.Background(), "AIDiagnoseRecipe")
|
||
|
||
// 1. 根据 recipeID 获取配方详情
|
||
recipe, err := s.recipeRepo.GetRecipeByID(serviceCtx, recipeID)
|
||
if err != nil {
|
||
logger.Errorf("获取配方详情失败: %v", err)
|
||
return "", s.ai.AIModel(), fmt.Errorf("获取配方详情失败: %w", err)
|
||
}
|
||
if recipe == nil {
|
||
logger.Warnf("未找到配方，ID: %d", recipeID)
|
||
return "", s.ai.AIModel(), fmt.Errorf("未找到配方，ID: %d", recipeID)
|
||
}
|
||
|
||
// 2. 获取目标猪只类型信息
|
||
pigType, err := s.GetPigTypeByID(serviceCtx, pigTypeID)
|
||
if err != nil {
|
||
logger.Errorf("获取猪只类型信息失败: %v", err)
|
||
return "", s.ai.AIModel(), fmt.Errorf("获取猪只类型信息失败: %w", err)
|
||
}
|
||
if pigType == nil {
|
||
logger.Warnf("未找到猪只类型，ID: %d", pigTypeID)
|
||
return "", s.ai.AIModel(), fmt.Errorf("未找到猪只类型，ID: %d", pigTypeID)
|
||
}
|
||
|
||
// 3. 定义 AI 输入结构体
|
||
type ingredientNutrient struct {
|
||
NutrientName string `json:"nutrient_name"`
|
||
Value float32 `json:"value"`
|
||
}
|
||
|
||
type recipeIngredient struct {
|
||
RawMaterialName string `json:"raw_material_name"`
|
||
Percentage float32 `json:"percentage"`
|
||
Nutrients []ingredientNutrient `json:"nutrients"`
|
||
}
|
||
|
||
type aiDiagnosisInput struct {
|
||
RecipeName string `json:"recipe_name"`
|
||
TargetPigType struct {
|
||
Name string `json:"name"`
|
||
} `json:"target_pig_type"`
|
||
Ingredients []recipeIngredient `json:"ingredients"`
|
||
}
|
||
|
||
// 4. 填充 AI 输入结构体
|
||
input := aiDiagnosisInput{
|
||
RecipeName: recipe.Name,
|
||
}
|
||
|
||
input.TargetPigType.Name = fmt.Sprintf("%s-%s", pigType.Breed.Name, pigType.AgeStage.Name)
|
||
|
||
for _, ingredient := range recipe.RecipeIngredients {
|
||
if ingredient.RawMaterial.ID == 0 {
|
||
logger.Warnf("配方成分中存在未加载的原料信息，RecipeIngredientID: %d", ingredient.ID)
|
||
continue
|
||
}
|
||
ing := recipeIngredient{
|
||
RawMaterialName: ingredient.RawMaterial.Name,
|
||
Percentage: ingredient.Percentage,
|
||
}
|
||
for _, rmn := range ingredient.RawMaterial.RawMaterialNutrients {
|
||
if rmn.Nutrient.ID == 0 {
|
||
logger.Warnf("原料营养成分中存在未加载的营养素信息，RawMaterialNutrientID: %d", rmn.ID)
|
||
continue
|
||
}
|
||
ing.Nutrients = append(ing.Nutrients, ingredientNutrient{
|
||
NutrientName: rmn.Nutrient.Name,
|
||
Value: rmn.Value,
|
||
})
|
||
}
|
||
input.Ingredients = append(input.Ingredients, ing)
|
||
}
|
||
|
||
// 5. 序列化为 JSON 字符串
|
||
jsonBytes, err := json.Marshal(input)
|
||
if err != nil {
|
||
logger.Errorf("序列化配方和猪只类型信息为 JSON 失败: %v", err)
|
||
return "", s.ai.AIModel(), fmt.Errorf("序列化数据失败: %w", err)
|
||
}
|
||
jsonString := string(jsonBytes)
|
||
|
||
// 6. 构建 AI Prompt
|
||
var promptBuilder strings.Builder
|
||
promptBuilder.WriteString(`
|
||
你是一个专业的动物营养师。请根据以下猪饲料配方数据，生成一份详细的、对养殖户友好的说明报告。
|
||
说明报告应包括以下部分：
|
||
1. 诊断猪只配方是否合理，如合理需要说明为什么合理, 如不合理需给出详细的改进建议。
|
||
2. 关键成分分析：分析主要原料和营养成分的作用
|
||
3. 使用建议：提供使用此配方的最佳实践和注意事项。
|
||
\n`)
|
||
promptBuilder.WriteString("```")
|
||
promptBuilder.WriteString(jsonString)
|
||
promptBuilder.WriteString("```")
|
||
prompt := promptBuilder.String()
|
||
|
||
logger.Debugf("生成的 AI 诊断 Prompt: \n%s", prompt)
|
||
|
||
// 7. 调用 AI Manager 进行诊断
|
||
diagnosisResult, err := s.ai.GenerateReview(serviceCtx, prompt)
|
||
if err != nil {
|
||
logger.Errorf("调用 AI Manager 诊断配方失败: %v", err)
|
||
return "", s.ai.AIModel(), fmt.Errorf("AI 诊断失败: %w", err)
|
||
}
|
||
|
||
logger.Infof("成功对配方 ID: %d (目标猪只类型 ID: %d) 进行 AI 诊断。", recipeID, pigTypeID)
|
||
return diagnosisResult, s.ai.AIModel(), nil
|
||
}
|