refactor: (GH-apache#85) Changes in benchmark CI

Signed-off-by: Saurabh Kumar Singh <[email protected]>

.github/workflows/benchmark.yml

@@ -19,6 +19,8 @@ name: Benchmarks
 on:
   push:
     branches: [main]
+  pull_request:
+    branches: [main]
     paths:
       - ".github/workflows/benchmark.yml"
       - "ci/scripts/bench_adapt.py"
@@ -35,6 +37,12 @@ jobs:
         arch: ['amd64']
     steps:
       - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - name: Install dependencies
+        run: |
+          apt-get update
+          apt-get install -y ca-certificates
       - name: Set up Python
         uses: actions/setup-python@v4
         with:
@@ -45,13 +53,14 @@ jobs:
           go-version: ${{ matrix.go }}
           cache: true
           cache-dependency-path: go.sum
+          check-latest: false
       - name: Run Benchmarks
-        run: |
-          ci/scripts/bench.sh --json
+        if: github.event_name != 'push'
+        run: bash ci/scripts/bench.sh $(pwd) --json
       - name: Upload results
         if: github.event_name == 'push' && github.repository == 'apache/arrow-go' && github.ref_name == 'main'
         env:
-          CONBENCH_URL: https://conbench.ursa.dev
+          CONBENCH_URL: https://conbench.ursa.dev 
           CONBENCH_EMAIL: ${{ secrets.CONBENCH_EMAIL }}
           CONBENCH_PASSWORD: ${{ secrets.CONBENCH_PASS }}
           CONBENCH_REF: ${{ github.ref_name }}

ci/scripts/bench.sh

@@ -23,9 +23,17 @@
 
 set -ex
 
-source_dir=${1}
+# Validate input arguments
+if [ -z "$1" ]; then
+  echo "Error: Missing source directory argument"
+  exit 1
+fi
+
+source_dir="$1"
+
+PARQUET_TEST_DATA="${source_dir}/parquet-testing/data"
+export PARQUET_TEST_DATA
 
-export PARQUET_TEST_DATA=${1}/cpp/submodules/parquet-testing/data
 pushd "${source_dir}"
 
 # lots of benchmarks, they can take a while