Skip to content

Commit 1384247

Browse files
committed
refactor: improve readability
1 parent 305bf8e commit 1384247

File tree

6 files changed

+19
-20
lines changed

6 files changed

+19
-20
lines changed

benches/mode.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ use arrow::util::bench_util::{create_primitive_array, create_string_array};
1919
use criterion::{Criterion, criterion_group, criterion_main};
2020
use datafusion::{arrow, logical_expr::Accumulator};
2121
use datafusion_functions_extra::common::mode::{BytesModeAccumulator, PrimitiveModeAccumulator};
22-
use std::{hint, sync};
22+
use std::{hint, slice, sync};
2323

2424
fn prepare_primitive_mode_accumulator() -> Box<dyn Accumulator> {
2525
Box::new(PrimitiveModeAccumulator::<arrow::datatypes::Int32Type>::new(
@@ -35,7 +35,7 @@ fn mode_bench_primitive(c: &mut Criterion, name: &str, values: arrow::array::Arr
3535
let mut accumulator = prepare_primitive_mode_accumulator();
3636
c.bench_function(name, |b| {
3737
b.iter(|| {
38-
accumulator.update_batch(&[values.clone()]).unwrap();
38+
accumulator.update_batch(slice::from_ref(&values)).unwrap();
3939
hint::black_box(accumulator.evaluate().unwrap());
4040
});
4141
});
@@ -45,7 +45,7 @@ fn mode_bench_bytes(c: &mut Criterion, name: &str, values: arrow::array::ArrayRe
4545
let mut accumulator = prepare_bytes_mode_accumulator();
4646
c.bench_function(name, |b| {
4747
b.iter(|| {
48-
accumulator.update_batch(&[values.clone()]).unwrap();
48+
accumulator.update_batch(slice::from_ref(&values)).unwrap();
4949
hint::black_box(accumulator.evaluate().unwrap());
5050
});
5151
});

src/common/mode/bytes.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
use datafusion::arrow::array::AsArray;
1919
use datafusion::{arrow, common, error, logical_expr, scalar};
20-
use std::collections;
20+
use std::{collections, mem};
2121

2222
#[derive(Debug)]
2323
pub struct BytesModeAccumulator {
@@ -142,7 +142,7 @@ impl logical_expr::Accumulator for BytesModeAccumulator {
142142
}
143143

144144
fn size(&self) -> usize {
145-
self.value_counts.capacity() * std::mem::size_of::<(String, i64)>() + std::mem::size_of_val(&self.data_type)
145+
self.value_counts.capacity() * mem::size_of::<(String, i64)>() + mem::size_of_val(&self.data_type)
146146
}
147147
}
148148

src/common/mode/native.rs

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
use datafusion::physical_expr::aggregate::utils::Hashable;
1919
use datafusion::{arrow, common, error, logical_expr, scalar};
20-
use std::{collections, fmt, hash};
20+
use std::{cmp, collections, fmt, hash, mem};
2121

2222
#[derive(fmt::Debug)]
2323
pub struct PrimitiveModeAccumulator<T>
@@ -108,11 +108,11 @@ where
108108

109109
self.value_counts.iter().for_each(|(value, &count)| {
110110
match count.cmp(&max_count) {
111-
std::cmp::Ordering::Greater => {
111+
cmp::Ordering::Greater => {
112112
max_value = Some(*value);
113113
max_count = count;
114114
}
115-
std::cmp::Ordering::Equal => {
115+
cmp::Ordering::Equal => {
116116
max_value = match max_value {
117117
Some(ref current_max_value) if value > current_max_value => Some(*value),
118118
Some(ref current_max_value) => Some(*current_max_value),
@@ -130,7 +130,7 @@ where
130130
}
131131

132132
fn size(&self) -> usize {
133-
std::mem::size_of_val(&self.value_counts) + self.value_counts.len() * std::mem::size_of::<(T::Native, i64)>()
133+
mem::size_of_val(&self.value_counts) + self.value_counts.len() * mem::size_of::<(T::Native, i64)>()
134134
}
135135
}
136136

@@ -221,11 +221,11 @@ where
221221

222222
self.value_counts.iter().for_each(|(value, &count)| {
223223
match count.cmp(&max_count) {
224-
std::cmp::Ordering::Greater => {
224+
cmp::Ordering::Greater => {
225225
max_value = Some(value.0);
226226
max_count = count;
227227
}
228-
std::cmp::Ordering::Equal => {
228+
cmp::Ordering::Equal => {
229229
max_value = match max_value {
230230
Some(current_max_value) if value.0 > current_max_value => Some(value.0),
231231
Some(current_max_value) => Some(current_max_value),
@@ -243,8 +243,7 @@ where
243243
}
244244

245245
fn size(&self) -> usize {
246-
std::mem::size_of_val(&self.value_counts)
247-
+ self.value_counts.len() * std::mem::size_of::<(Hashable<T::Native>, i64)>()
246+
mem::size_of_val(&self.value_counts) + self.value_counts.len() * mem::size_of::<(Hashable<T::Native>, i64)>()
248247
}
249248
}
250249

src/kurtosis.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
use datafusion::arrow::array::{Float64Array, UInt64Array};
1919
use datafusion::{arrow, common, error, logical_expr, scalar};
20-
use std::{any, fmt};
20+
use std::{any, fmt, mem};
2121

2222
make_udaf_expr_and_func!(
2323
KurtosisFunction,
@@ -179,7 +179,7 @@ impl logical_expr::Accumulator for KurtosisAccumulator {
179179
}
180180

181181
fn size(&self) -> usize {
182-
std::mem::size_of_val(self)
182+
mem::size_of_val(self)
183183
}
184184

185185
fn state(&mut self) -> error::Result<Vec<scalar::ScalarValue>> {

src/kurtosis_pop.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
use datafusion::arrow::array::{Float64Array, UInt64Array};
2222
use datafusion::{arrow, common, error, logical_expr, scalar};
23-
use std::{any, fmt};
23+
use std::{any, fmt, mem};
2424

2525
make_udaf_expr_and_func!(
2626
KurtosisPopFunction,
@@ -177,7 +177,7 @@ impl logical_expr::Accumulator for KurtosisPopAccumulator {
177177
}
178178

179179
fn size(&self) -> usize {
180-
std::mem::size_of_val(self)
180+
mem::size_of_val(self)
181181
}
182182

183183
fn state(&mut self) -> error::Result<Vec<scalar::ScalarValue>> {

src/skewness.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
use datafusion::arrow::array::AsArray;
1919
use datafusion::{arrow, logical_expr, scalar};
2020
use std::ops::{Div, Mul, Sub};
21-
use std::{any, fmt};
21+
use std::{any, fmt, mem};
2222

2323
make_udaf_expr_and_func!(SkewnessFunc, skewness, x, "Computes the skewness value.", skewness_udaf);
2424

@@ -28,7 +28,7 @@ pub struct SkewnessFunc {
2828
}
2929

3030
impl fmt::Debug for SkewnessFunc {
31-
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
31+
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
3232
f.debug_struct("SkewnessFunc")
3333
.field("signature", &self.signature)
3434
.finish()
@@ -143,7 +143,7 @@ impl logical_expr::Accumulator for SkewnessAccumulator {
143143
}
144144

145145
fn size(&self) -> usize {
146-
std::mem::size_of_val(self)
146+
mem::size_of_val(self)
147147
}
148148

149149
fn state(&mut self) -> datafusion::common::Result<Vec<scalar::ScalarValue>> {

0 commit comments

Comments
 (0)