Skip to content

Commit 70ff89c

Browse files
committed
lint
1 parent c2af4bd commit 70ff89c

1 file changed

Lines changed: 22 additions & 30 deletions

File tree

native/core/benches/row_columnar.rs

Lines changed: 22 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)