address comments

pingcap · Nov 24, 2023 · 626fd5e · 626fd5e
1 parent 47f7681
commit 626fd5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/planner/core/rule_column_pruning.go b/pkg/planner/core/rule_column_pruning.go
@@ -18,6 +18,7 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"slices"
 
 	"github.com/pingcap/tidb/pkg/expression"
 	"github.com/pingcap/tidb/pkg/expression/aggregation"
@@ -290,7 +291,7 @@ func (p *LogicalUnionAll) PruneColumns(parentUsedCols []*expression.Column, opt
 	for i := len(used) - 1; i >= 0; i-- {
 		if !used[i] {
 			prunedColumns = append(prunedColumns, p.schema.Columns[i])
-			p.schema.Columns = append(p.schema.Columns[:i], p.schema.Columns[i+1:]...)
+			p.schema.Columns = slices.Delete(p.schema.Columns, i, i)
 		}
 	}
 	appendColumnPruneTraceStep(p, prunedColumns, opt)