@@ -1578,7 +1578,7 @@ impl SortMergeJoinStream {
1578
1578
. append_nulls ( num_rows) ;
1579
1579
self . output_record_batches
1580
1580
. batch_ids
1581
- . extend ( vec ! [ 0 ; num_rows] ) ;
1581
+ . resize ( self . output_record_batches . batch_ids . len ( ) + num_rows, 0 ) ;
1582
1582
1583
1583
self . output_record_batches . batches . push ( record_batch) ;
1584
1584
}
@@ -1622,7 +1622,7 @@ impl SortMergeJoinStream {
1622
1622
. append_nulls ( num_rows) ;
1623
1623
self . output_record_batches
1624
1624
. batch_ids
1625
- . extend ( vec ! [ 0 ; num_rows] ) ;
1625
+ . resize ( self . output_record_batches . batch_ids . len ( ) + num_rows, 0 ) ;
1626
1626
self . output_record_batches . batches . push ( record_batch) ;
1627
1627
}
1628
1628
buffered_batch. join_filter_not_matched_map . clear ( ) ;
@@ -1757,10 +1757,10 @@ impl SortMergeJoinStream {
1757
1757
self . output_record_batches . filter_mask . extend ( pre_mask) ;
1758
1758
}
1759
1759
self . output_record_batches . row_indices . extend ( & left_indices) ;
1760
- self . output_record_batches . batch_ids . extend ( vec ! [
1761
- self . streamed_batch_counter . load ( Relaxed ) ;
1762
- left_indices . len ( )
1763
- ] ) ;
1760
+ self . output_record_batches . batch_ids . resize (
1761
+ self . output_record_batches . batch_ids . len ( ) + left_indices . len ( ) ,
1762
+ self . streamed_batch_counter . load ( Relaxed ) ,
1763
+ ) ;
1764
1764
1765
1765
// For outer joins, we need to push the null joined rows to the output if
1766
1766
// all joined rows are failed on the join filter.
0 commit comments