@@ -103,10 +103,8 @@ fn build_struct_row(depth: usize, num_leaf_fields: usize) -> Vec<u8> {
103103 let inter_level_size = inter_bitset + UNSAFE_ROW_POINTER_SIZE ;
104104 let leaf_level_size = leaf_bitset + num_leaf_fields * INT64_SIZE ;
105105
106- let total = top_bitset
107- + UNSAFE_ROW_POINTER_SIZE
108- + ( depth - 1 ) * inter_level_size
109- + leaf_level_size;
106+ let total =
107+ top_bitset + UNSAFE_ROW_POINTER_SIZE + ( depth - 1 ) * inter_level_size + leaf_level_size;
110108 let mut data = vec ! [ 0u8 ; total] ;
111109
112110 // Absolute start position of each struct level in the buffer
@@ -248,29 +246,25 @@ fn run_benchmark(
248246 let addr_ptr = addrs. as_mut_ptr ( ) ;
249247 let size_ptr = sizes. as_mut_ptr ( ) ;
250248
251- group. bench_with_input (
252- BenchmarkId :: new ( name, param) ,
253- & num_rows,
254- |b, & n| {
255- b. iter ( || {
256- let tmp = Builder :: new ( ) . tempfile ( ) . unwrap ( ) ;
257- process_sorted_row_partition (
258- n,
259- BATCH_SIZE ,
260- addr_ptr,
261- size_ptr,
262- schema,
263- tmp. path ( ) . to_str ( ) . unwrap ( ) . to_string ( ) ,
264- 1.0 ,
265- false ,
266- 0 ,
267- None ,
268- & CompressionCodec :: Zstd ( 1 ) ,
269- )
270- . unwrap ( ) ;
271- } ) ;
272- } ,
273- ) ;
249+ group. bench_with_input ( BenchmarkId :: new ( name, param) , & num_rows, |b, & n| {
250+ b. iter ( || {
251+ let tmp = Builder :: new ( ) . tempfile ( ) . unwrap ( ) ;
252+ process_sorted_row_partition (
253+ n,
254+ BATCH_SIZE ,
255+ addr_ptr,
256+ size_ptr,
257+ schema,
258+ tmp. path ( ) . to_str ( ) . unwrap ( ) . to_string ( ) ,
259+ 1.0 ,
260+ false ,
261+ 0 ,
262+ None ,
263+ & CompressionCodec :: Zstd ( 1 ) ,
264+ )
265+ . unwrap ( ) ;
266+ } ) ;
267+ } ) ;
274268
275269 drop ( spark_rows) ;
276270}
@@ -368,9 +362,7 @@ fn benchmark_map_conversion(c: &mut Criterion) {
368362 for num_entries in [ 10 , 100 ] {
369363 for num_rows in [ 1000 , 10000 ] {
370364 let schema = vec ! [ make_map_schema( ) ] ;
371- let rows: Vec < Vec < u8 > > = ( 0 ..num_rows)
372- . map ( |_| build_map_row ( num_entries) )
373- . collect ( ) ;
365+ let rows: Vec < Vec < u8 > > = ( 0 ..num_rows) . map ( |_| build_map_row ( num_entries) ) . collect ( ) ;
374366
375367 run_benchmark (
376368 & mut group,
0 commit comments