Benchmark window function with multiple partitioning columns

Author: findepi

datafusion/core/benches/sql_planner.rs

@@ -476,6 +476,27 @@ fn criterion_benchmark(c: &mut Criterion) {
         });
     });
 
+    c.bench_function("physical_window_function_partition_by_8_on_values", |b| {
+        let partitioning_columns = 8;
+        let source = format!(
+            "SELECT 1 AS n{}",
+            (0..partitioning_columns)
+                .map(|i| format!(", {i} AS c{i}"))
+                .collect::<String>()
+        );
+        let window = format!(
+            "SUM(n) OVER (PARTITION BY {}) AS sum_n",
+            (0..partitioning_columns)
+                .map(|i| format!("c{i}"))
+                .collect::<Vec<_>>()
+                .join(", ")
+        );
+        let query = format!("SELECT {window} FROM ({source})");
+        b.iter(|| {
+            physical_plan(&ctx, &rt, &query);
+        });
+    });
+
     // Benchmark for Physical Planning Joins
     c.bench_function("physical_join_consider_sort", |b| {
         b.iter(|| {