From 36864612f16bf4f37e8cea5524067f9879efde38 Mon Sep 17 00:00:00 2001
From: zhuqi-lucas <821684824@qq.com>
Date: Thu, 3 Jul 2025 19:28:00 +0800
Subject: [PATCH] benchmark: Support sort_tpch10 for benchmark

---
 benchmarks/bench.sh | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/benchmarks/bench.sh b/benchmarks/bench.sh
index effce26d1cd2e..1a687ac0241d7 100755
--- a/benchmarks/bench.sh
+++ b/benchmarks/bench.sh
@@ -89,6 +89,7 @@ tpch_mem10:             TPCH inspired benchmark on Scale Factor (SF) 10 (~10GB),
 
 # Extended TPC-H Benchmarks
 sort_tpch:              Benchmark of sorting speed for end-to-end sort queries on TPC-H dataset (SF=1)
+sort_tpch10:            Benchmark of sorting speed for end-to-end sort queries on TPC-H dataset (SF=10)
 topk_tpch:              Benchmark of top-k (sorting with limit) queries on TPC-H dataset (SF=1)
 external_aggr:          External aggregation benchmark on TPC-H dataset (SF=1)
 
@@ -250,6 +251,10 @@ main() {
                     # same data as for tpch
                     data_tpch "1"
                     ;;
+                sort_tpch10)
+                    # same data as for tpch10
+                    data_tpch "10"
+                    ;;
                 topk_tpch)
                     # same data as for tpch
                     data_tpch "1"
@@ -382,7 +387,10 @@ main() {
                     run_external_aggr
                     ;;
                 sort_tpch)
-                    run_sort_tpch
+                    run_sort_tpch "1"
+                    ;;
+                sort_tpch10)
+                    run_sort_tpch "10"
                     ;;
                 topk_tpch)
                     run_topk_tpch
@@ -997,8 +1005,13 @@ run_external_aggr() {
 
 # Runs the sort integration benchmark
 run_sort_tpch() {
-    TPCH_DIR="${DATA_DIR}/tpch_sf1"
-    RESULTS_FILE="${RESULTS_DIR}/sort_tpch.json"
+    SCALE_FACTOR=$1
+    if [ -z "$SCALE_FACTOR" ] ; then
+        echo "Internal error: Scale factor not specified"
+        exit 1
+    fi
+    TPCH_DIR="${DATA_DIR}/tpch_sf${SCALE_FACTOR}"
+    RESULTS_FILE="${RESULTS_DIR}/sort_tpch${SCALE_FACTOR}.json"
     echo "RESULTS_FILE: ${RESULTS_FILE}"
     echo "Running sort tpch benchmark..."