executor: reuse memory for parallelSortSpillHelper (#53859) (#53879)

close #53861
pingcap · Jun 8, 2024 · b7a6c25 · b7a6c25
1 parent 03990e9
commit b7a6c25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/executor/sortexec/parallel_sort_spill_helper.go b/pkg/executor/sortexec/parallel_sort_spill_helper.go
@@ -52,14 +52,15 @@ func newParallelSortSpillHelper(sortExec *SortExec, fieldTypes []*types.FieldTyp
 		errOutputChan: errOutputChan,
 		finishCh:      finishCh,
 		fieldTypes:    fieldTypes,
-		tmpSpillChunk: chunk.NewChunkWithCapacity(fieldTypes, spillChunkSize),
+		tmpSpillChunk: chunk.NewChunkFromPoolWithCapacity(fieldTypes, spillChunkSize),
 	}
 }
 
 func (p *parallelSortSpillHelper) close() {
 	for _, inDisk := range p.sortedRowsInDisk {
 		inDisk.Close()
 	}
+	p.tmpSpillChunk.Destroy(spillChunkSize, p.fieldTypes)
 }
 
 func (p *parallelSortSpillHelper) isNotSpilledNoLock() bool {