refactor_boolean_cast_ops_add_benchmarks

Author: coderfender

native/spark-expr/benches/cast_boolean.rs

@@ -15,23 +15,26 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use std::sync::Arc;
 use arrow::array::{BooleanBuilder, RecordBatch};
 use arrow::datatypes::{DataType, Field, Schema};
 use criterion::{criterion_group, criterion_main, Criterion};
 use datafusion::physical_expr::expressions::Column;
 use datafusion::physical_expr::PhysicalExpr;
 use datafusion_comet_spark_expr::{Cast, EvalMode, SparkCastOptions};
+use std::sync::Arc;
 
 fn criterion_benchmark(c: &mut Criterion) {
     let expr = Arc::new(Column::new("a", 0));
     let boolean_batch = create_boolean_batch();
     let spark_cast_options = SparkCastOptions::new(EvalMode::Legacy, "UTC", false);
     Cast::new(expr.clone(), DataType::Int8, spark_cast_options.clone());
-    let cast_to_i8 = Cast::new(expr.clone(), DataType::Boolean, spark_cast_options.clone());
-    let cast_to_i16 = Cast::new(expr.clone(), DataType::Boolean, spark_cast_options.clone());
-    let cast_to_i32 = Cast::new(expr.clone(), DataType::Boolean, spark_cast_options.clone());
-    let cast_to_i64 = Cast::new(expr.clone(), DataType::Boolean, spark_cast_options);
+    let cast_to_i8 = Cast::new(expr.clone(), DataType::Int8, spark_cast_options.clone());
+    let cast_to_i16 = Cast::new(expr.clone(), DataType::Int16, spark_cast_options.clone());
+    let cast_to_i32 = Cast::new(expr.clone(), DataType::Int32, spark_cast_options.clone());
+    let cast_to_i64 = Cast::new(expr.clone(), DataType::Int64, spark_cast_options.clone());
+    let cast_to_f32 = Cast::new(expr.clone(), DataType::Float32, spark_cast_options.clone());
+    let cast_to_f64 = Cast::new(expr.clone(), DataType::Float64, spark_cast_options.clone());
+    let cast_to_str = Cast::new(expr, DataType::Utf8, spark_cast_options);
 
     let mut group = c.benchmark_group(format!("cast_bool_to_int"));
     group.bench_function("i8", |b| {
@@ -46,6 +49,15 @@ fn criterion_benchmark(c: &mut Criterion) {
     group.bench_function("i64", |b| {
         b.iter(|| cast_to_i64.evaluate(&boolean_batch).unwrap());
     });
+    group.bench_function("f32", |b| {
+        b.iter(|| cast_to_f32.evaluate(&boolean_batch).unwrap());
+    });
+    group.bench_function("f64", |b| {
+        b.iter(|| cast_to_f64.evaluate(&boolean_batch).unwrap());
+    });
+    group.bench_function("str", |b| {
+        b.iter(|| cast_to_str.evaluate(&boolean_batch).unwrap());
+    });
 }
 
 fn create_boolean_batch() -> RecordBatch {
@@ -71,4 +83,4 @@ criterion_group! {
     config = config();
     targets = criterion_benchmark
 }
-criterion_main!(benches);
+criterion_main!(benches);