pig/pig-farm-controller
1
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

修复createPlan逻辑错误

Browse Source
This commit is contained in:
huang
2025-09-14 14:45:56 +08:00
parent 67707c0d41
commit 352c9d9246
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
internal/infra/repository/plan_repository.go
View File

@@ -137,10 +137,10 @@ func (r *gormPlanRepository) CreatePlan(plan *models.Plan) error {
if plan.ContentType == models.PlanContentTypeSubPlans { if plan.ContentType == models.PlanContentTypeSubPlans {
childIDsToValidate := make(map[uint]bool) childIDsToValidate := make(map[uint]bool)
for _, subPlanLink := range plan.SubPlans { for _, subPlanLink := range plan.SubPlans {
if subPlanLink.ChildPlan == nil || subPlanLink.ChildPlan.ID == 0 { if subPlanLink.ChildPlanID == 0 {
return ErrSubPlanIDIsZeroOnCreate return ErrSubPlanIDIsZeroOnCreate
} }
childIDsToValidate[subPlanLink.ChildPlan.ID] = true childIDsToValidate[subPlanLink.ChildPlanID] = true
} }
var ids []uint var ids []uint

2
internal/infra/repository/plan_repository_test.go
View File

@@ -1134,7 +1134,7 @@ func TestPlanRepository_Create(t *testing.T) {
Name: "父计划", Name: "父计划",
ContentType: models.PlanContentTypeSubPlans, ContentType: models.PlanContentTypeSubPlans,
SubPlans: []models.SubPlan{ SubPlans: []models.SubPlan{
{ChildPlanID: 1, ExecutionOrder: 1, ChildPlan: &models.Plan{Model: gorm.Model{ID: 1}}}, // 关联已存在的子计划1 {ChildPlanID: 1, ExecutionOrder: 1}, // 关联已存在的子计划1
{ChildPlanID: 2, ExecutionOrder: 2, ChildPlan: &models.Plan{Model: gorm.Model{ID: 2}}}, // 关联已存在的子计划2 {ChildPlanID: 2, ExecutionOrder: 2, ChildPlan: &models.Plan{Model: gorm.Model{ID: 2}}}, // 关联已存在的子计划2
}, },
}, },