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-22 21:29:23 +08:00
parent b40eb35016
commit 851682d579
11 changed files with 982 additions and 587 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/infra/repository/raw_material_repository.go
View File

@@ -176,13 +176,21 @@ func (r *gormRawMaterialRepository) DeleteNutrientsByRawMaterialIDTx(ctx context
// CreateBatchRawMaterialNutrientsTx 在事务中批量创建原料营养成分
func (r *gormRawMaterialRepository) CreateBatchRawMaterialNutrientsTx(ctx context.Context, db *gorm.DB, nutrients []models.RawMaterialNutrient) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateBatchRawMaterialNutrientsTx")
// 如果没有要创建的记录，直接返回成功，避免执行空的Create语句
if len(nutrients) == 0 {
return nil
}
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateBatchRawMaterialNutrientsTx")
tx := db.WithContext(repoCtx)
if err := tx.Create(&nutrients).Error; err != nil {
// 确保每个营养都关联到正确的原料ID
// 注意：这里假设传入的 nutrients 已经设置了正确的 RawMaterialID
for i := range nutrients {
if nutrients[i].RawMaterialID == 0 {
return fmt.Errorf("创建原料营养时 RecipeID 不能为空")
}
}
if err := db.WithContext(repoCtx).Create(&nutrients).Error; err != nil {
return fmt.Errorf("批量创建原料营养成分失败: %w", err)
}
return nil

68
internal/infra/repository/recipe_repository.go
View File

@@ -25,8 +25,11 @@ type RecipeRepository interface {
GetRecipeByName(ctx context.Context, name string) (*models.Recipe, error)
ListRecipes(ctx context.Context, opts RecipeListOptions, page, pageSize int) ([]models.Recipe, int64, error)
UpdateRecipe(ctx context.Context, recipe *models.Recipe) error
UpdateRecipeIngredients(ctx context.Context, recipeID uint32, ingredients []models.RecipeIngredient) error
DeleteRecipe(ctx context.Context, id uint32) error
// 在事务中删除配方原料
DeleteRecipeIngredientsByRecipeIDTx(ctx context.Context, tx *gorm.DB, recipeID uint32) error
// 在事务中批量创建配方原料
CreateBatchRecipeIngredientsTx(ctx context.Context, tx *gorm.DB, ingredients []models.RecipeIngredient) error
}
// gormRecipeRepository 是 RecipeRepository 的 GORM 实现
@@ -43,12 +46,12 @@ func NewGormRecipeRepository(ctx context.Context, db *gorm.DB) RecipeRepository
// CreateRecipe 创建一个新的配方，并处理其关联的配方原料
func (r *gormRecipeRepository) CreateRecipe(ctx context.Context, recipe *models.Recipe) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateRecipe")
return r.db.WithContext(repoCtx).Transaction(func(tx *gorm.DB) error {
if err := tx.Create(recipe).Error; err != nil {
return fmt.Errorf("创建配方失败: %w", err)
}
return nil
})
// 注意：这里的事务只针对 Recipe 本身，如果 RecipeIngredient 也在同一个 Create 中，GORM 会自动处理。
// 但如果 RecipeIngredient 是单独操作，则需要外部事务。
if err := r.db.WithContext(repoCtx).Create(recipe).Error; err != nil {
return fmt.Errorf("创建配方失败: %w", err)
}
return nil
}
// GetRecipeByID 根据ID获取单个配方，并预加载其关联的配方原料和原料信息
@@ -130,29 +133,6 @@ func (r *gormRecipeRepository) UpdateRecipe(ctx context.Context, recipe *models.
return nil
}
// UpdateRecipeIngredients 更新配方关联的原料列表
func (r *gormRecipeRepository) UpdateRecipeIngredients(ctx context.Context, recipeID uint32, ingredients []models.RecipeIngredient) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "UpdateRecipeIngredients")
return r.db.WithContext(repoCtx).Transaction(func(tx *gorm.DB) error {
// 1. 删除所有旧的关联配方原料
if err := tx.Where("recipe_id = ?", recipeID).Delete(&models.RecipeIngredient{}).Error; err != nil {
return fmt.Errorf("删除旧的配方原料失败: %w", err)
}
// 2. 批量创建新的关联配方原料
if len(ingredients) > 0 {
for i := range ingredients {
ingredients[i].RecipeID = recipeID
}
if err := tx.Create(&ingredients).Error; err != nil {
return fmt.Errorf("创建新的配方原料失败: %w", err)
}
}
return nil
})
}
// DeleteRecipe 根据ID删除一个配方，并级联软删除关联的 RecipeIngredient 记录
func (r *gormRecipeRepository) DeleteRecipe(ctx context.Context, id uint32) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "DeleteRecipe")
@@ -180,3 +160,31 @@ func (r *gormRecipeRepository) DeleteRecipe(ctx context.Context, id uint32) erro
return nil
})
}
// DeleteRecipeIngredientsByRecipeIDTx 在给定事务中删除配方原料
func (r *gormRecipeRepository) DeleteRecipeIngredientsByRecipeIDTx(ctx context.Context, tx *gorm.DB, recipeID uint32) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "DeleteRecipeIngredientsByRecipeIDTx")
if err := tx.WithContext(repoCtx).Where("recipe_id = ?", recipeID).Delete(&models.RecipeIngredient{}).Error; err != nil {
return fmt.Errorf("删除配方 %d 的原料失败: %w", recipeID, err)
}
return nil
}
// CreateBatchRecipeIngredientsTx 在给定事务中批量创建配方原料
func (r *gormRecipeRepository) CreateBatchRecipeIngredientsTx(ctx context.Context, tx *gorm.DB, ingredients []models.RecipeIngredient) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreateBatchRecipeIngredientsTx")
if len(ingredients) == 0 {
return nil // 没有原料需要创建
}
// 确保每个原料都关联到正确的配方ID
// 注意：这里假设传入的 ingredients 已经设置了正确的 RecipeID
for i := range ingredients {
if ingredients[i].RecipeID == 0 {
return fmt.Errorf("创建配方原料时 RecipeID 不能为空")
}
}
if err := tx.WithContext(repoCtx).Create(&ingredients).Error; err != nil {
return fmt.Errorf("批量创建配方原料失败: %w", err)
}
return nil
}