Spanner CI bench

Author: Torch3333

.github/workflows/spanner-thread-benchmark.yaml

@@ -0,0 +1,68 @@
+name: Spanner Thread Count Benchmark
+run-name: "Spanner Thread Benchmark"
+
+on:
+  workflow_dispatch:
+  push:
+
+env:
+  SPANNER_JDBC_URL: "jdbc:cloudspanner://localhost:9010/projects/emulator-project/instances/test-instance/databases/test-db;autoConfigEmulator=true;dialect=POSTGRESQL"
+
+jobs:
+  benchmark:
+    name: "${{ matrix.test_class }} threads=${{ matrix.thread_num }}"
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        test_class:
+          - JdbcDatabaseConditionalMutationIntegrationTest
+          - JdbcDatabaseMultipleClusteringKeyScanIntegrationTest
+        thread_num: [1, 2, 3, 4, 5, 6]
+    steps:
+      - uses: actions/checkout@v6
+
+      - name: Install and start Spanner emulator
+        run: |
+          echo "deb [signed-by=/usr/share/keyrings/cloud.google.asc] https://packages.cloud.google.com/apt cloud-sdk main" \
+            | sudo tee /etc/apt/sources.list.d/google-cloud-sdk.list
+          curl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg \
+            | sudo tee /usr/share/keyrings/cloud.google.asc > /dev/null
+          sudo apt-get update -qq
+          sudo apt-get install -y -qq google-cloud-cli-spanner-emulator
+          EMULATOR_DIR=/usr/lib/google-cloud-sdk/bin/cloud_spanner_emulator
+          ${EMULATOR_DIR}/gateway_main \
+            --hostname 0.0.0.0 \
+            --grpc_port 9010 \
+            --http_port 9020 \
+            --grpc_binary ${EMULATOR_DIR}/emulator_main \
+            &
+          until curl -sf http://localhost:9020/v1/projects/emulator-project/instances > /dev/null 2>&1; do sleep 1; done
+
+      - name: Set up JDK 8
+        uses: actions/setup-java@v5
+        with:
+          java-version: 8
+          distribution: temurin
+
+      - name: Setup Gradle
+        uses: gradle/actions/setup-gradle@v5
+
+      - name: Run ${{ matrix.test_class }} with ${{ matrix.thread_num }} threads
+        env:
+          SCALARDB_INTEGRATION_TEST_THREAD_NUM: ${{ matrix.thread_num }}
+          ORG_GRADLE_PROJECT_javaVersion: '8'
+          ORG_GRADLE_PROJECT_javaVendor: 'temurin'
+          ORG_GRADLE_PROJECT_integrationTestJavaRuntimeVersion: '8'
+          ORG_GRADLE_PROJECT_integrationTestJavaRuntimeVendor: 'temurin'
+        run: |
+          ./gradlew core:integrationTestJdbc \
+            "-Dscalardb.jdbc.url=${SPANNER_JDBC_URL}" \
+            --tests ${{ matrix.test_class }}
+
+      - name: Upload test report
+        if: always()
+        uses: actions/upload-artifact@v7
+        with:
+          name: benchmark_${{ matrix.test_class }}_threads_${{ matrix.thread_num }}
+          path: core/build/reports/tests/

core/src/integration-test/java/com/scalar/db/storage/jdbc/JdbcDatabaseConditionalMutationIntegrationTest.java

@@ -28,6 +28,11 @@ protected int getThreadNum() {
       return 1;
     }
 
+    String envThreadNum = System.getenv("SCALARDB_INTEGRATION_TEST_THREAD_NUM");
+    if (envThreadNum != null) {
+      return Integer.parseInt(envThreadNum);
+    }
+
     return super.getThreadNum();
   }
 
@@ -48,4 +53,4 @@ protected Column<?> getColumnWithRandomValue(
   protected boolean isConditionOnBlobColumnSupported() {
     return !JdbcTestUtils.isOracle(rdbEngine);
   }
-}
+}

core/src/integration-test/java/com/scalar/db/storage/jdbc/JdbcDatabaseMultipleClusteringKeyScanIntegrationTest.java

@@ -26,9 +26,15 @@ protected Properties getProperties(String testName) {
 
   @Override
   protected int getThreadNum() {
-    if (JdbcTestUtils.isOracle(rdbEngine) || JdbcTestUtils.isSpanner(rdbEngine)) {
+    if (JdbcTestUtils.isOracle(rdbEngine)) {
       return 1;
     }
+
+    String envThreadNum = System.getenv("SCALARDB_INTEGRATION_TEST_THREAD_NUM");
+    if (envThreadNum != null) {
+      return Integer.parseInt(envThreadNum);
+    }
+
     return super.getThreadNum();
   }