planner: avoid slicesgrow in the buildDataSource

Signed-off-by: Weizhen Wang <[email protected]>
pingcap · Jan 10, 2025 · cd9d3c0 · cd9d3c0
1 parent d9dc3f4
commit cd9d3c0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/planner/core/logical_plan_builder.go b/pkg/planner/core/logical_plan_builder.go
@@ -4604,16 +4604,16 @@ func (b *PlanBuilder) buildDataSource(ctx context.Context, tn *ast.TableName, as
 		IndexHints:          b.TableHints().IndexHintList,
 		IndexMergeHints:     indexMergeHints,
 		PossibleAccessPaths: possiblePaths,
-		Columns:             make([]*model.ColumnInfo, 0, len(columns)),
+		Columns:             make([]*model.ColumnInfo, 0, len(columns)+1),
 		PartitionNames:      tn.PartitionNames,
-		TblCols:             make([]*expression.Column, 0, len(columns)),
+		TblCols:             make([]*expression.Column, 0, len(columns)+1),
 		PreferPartitions:    make(map[int][]ast.CIStr),
 		IS:                  b.is,
 		IsForUpdateRead:     b.isForUpdateRead,
 	}.Init(b.ctx, b.getSelectOffset())
 	var handleCols util.HandleCols
-	schema := expression.NewSchema(make([]*expression.Column, 0, len(columns))...)
-	names := make([]*types.FieldName, 0, len(columns))
+	schema := expression.NewSchema(make([]*expression.Column, 0, len(columns)+1)...)
+	names := make([]*types.FieldName, 0, len(columns)+1)
 	for i, col := range columns {
 		ds.Columns = append(ds.Columns, col.ToInfo())
 		names = append(names, &types.FieldName{