fix test

Signed-off-by: AilinKid <[email protected]>
pingcap · Mar 28, 2024 · ef616c4 · ef616c4
1 parent a2f7d45
commit ef616c4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/planner/core/optimizer.go b/pkg/planner/core/optimizer.go
@@ -1165,7 +1165,7 @@ func logicalOptimize(ctx context.Context, flag uint64, logic LogicalPlan) (Logic
 		}
 	}
 
-	opt.RecordFinalLogicalPlan(logic.BuildPlanTrace())
+	opt.RecordFinalLogicalPlan(logic.BuildPlanTrace)
 	return logic, err
 }
 

diff --git a/pkg/planner/util/optTracer.go b/pkg/planner/util/optTracer.go
@@ -60,9 +60,9 @@ func (op *LogicalOptimizeOp) AppendStepToCurrent(id int, tp string, reason, acti
 }
 
 // RecordFinalLogicalPlan records the final logical plan.
-func (op *LogicalOptimizeOp) RecordFinalLogicalPlan(planTrace *tracing.PlanTrace) {
+func (op *LogicalOptimizeOp) RecordFinalLogicalPlan(build func() *tracing.PlanTrace) {
 	if op == nil || op.tracer == nil {
 		return
 	}
-	op.tracer.RecordFinalLogicalPlan(planTrace)
+	op.tracer.RecordFinalLogicalPlan(build())
 }