Add benchmark for array_reverse of list view

Author: vegarsti

datafusion/functions-nested/benches/array_expression.rs

@@ -19,16 +19,22 @@
 extern crate criterion;
 extern crate arrow;
 
+use std::sync::Arc;
+
 use crate::criterion::Criterion;
+use arrow::{
+    array::{Int32Array, ListArray, ListViewArray},
+    buffer::{OffsetBuffer, ScalarBuffer},
+    datatypes::{DataType, Field},
+};
 use datafusion_expr::lit;
 use datafusion_functions_nested::expr_fn::{array_replace_all, make_array};
+use datafusion_functions_nested::reverse::array_reverse_inner;
 use std::hint::black_box;
 
 fn criterion_benchmark(c: &mut Criterion) {
     // Construct large arrays for benchmarking
-
-    let array_len = 100000000;
-
+    let array_len = 100000;
     let array = (0..array_len).map(|_| lit(2_i64)).collect::<Vec<_>>();
     let list_array = make_array(vec![make_array(array); 3]);
     let from_array = make_array(vec![lit(2_i64); 3]);
@@ -37,7 +43,6 @@ fn criterion_benchmark(c: &mut Criterion) {
     let expected_array = list_array.clone();
 
     // Benchmark array functions
-
     c.bench_function("array_replace", |b| {
         b.iter(|| {
             assert_eq!(
@@ -50,6 +55,38 @@ fn criterion_benchmark(c: &mut Criterion) {
             )
         })
     });
+
+    // Construct large arrays for reverse
+    let step_size: usize = 1000;
+    let offsets: Vec<i32> = (0..array_len as i32)
+        .step_by(step_size.try_into().unwrap())
+        .collect();
+    let offsets = ScalarBuffer::from(offsets);
+    let sizes: Vec<i32> = vec![step_size as i32; array_len / step_size as usize];
+    let values = (0..array_len as i32).collect::<Vec<i32>>();
+
+    // Reverse list array
+    let list_array = ListArray::new(
+        Arc::new(Field::new("a", DataType::Int32, false)),
+        OffsetBuffer::new(offsets.clone()),
+        Arc::new(Int32Array::from(values.clone())),
+        None,
+    );
+    c.bench_function("array_reverse_list", |b| {
+        b.iter(|| array_reverse_inner(&[Arc::new(list_array.clone())]))
+    });
+
+    // Reverse list view array
+    let list_view_array = ListViewArray::new(
+        Arc::new(Field::new("a", DataType::Int32, false)),
+        offsets,
+        ScalarBuffer::from(sizes),
+        Arc::new(Int32Array::from(values)),
+        None,
+    );
+    c.bench_function("array_reverse_list_view", |b| {
+        b.iter(|| array_reverse_inner(&[Arc::new(list_view_array.clone())]))
+    });
 }
 
 criterion_group!(benches, criterion_benchmark);