Adding support for benchbase

PiperOrigin-RevId: 820364561

Author: ScottLinnn

perfkitbenchmarker/data/benchbase_conf.j2

@@ -0,0 +1,44 @@
+<?xml version="1.0"?>
+<parameters>
+    <type>{{ db_type }}</type>
+    <driver>{{ driver_class }}</driver>
+    <url>{{ jdbc_url }}</url>
+    {{ username_element }}
+    {{ password_element }}
+    <reconnectOnConnectionFailure>true</reconnectOnConnectionFailure>
+    <isolation>{{ isolation }}</isolation>
+    <scalefactor>{{ scalefactor }}</scalefactor>
+    <terminals>{{ terminals }}</terminals>
+    <batchsize>128</batchsize>
+
+    <!-- Benchmark Type -->
+    <type>TPCC</type>
+
+    <!-- Workload for TPC-C -->
+    <works>
+        <work>
+            <time>{{ workload_duration_seconds }}</time>
+            {{ rate_element }}
+            <weights>{{ weights }}</weights>
+        </work>
+    </works>
+
+    <!-- TPC-C Specific -->
+    <transactiontypes>
+        <transactiontype>
+            <name>NewOrder</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Payment</name>
+        </transactiontype>
+        <transactiontype>
+            <name>OrderStatus</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Delivery</name>
+        </transactiontype>
+        <transactiontype>
+            <name>StockLevel</name>
+        </transactiontype>
+    </transactiontypes>
+</parameters>

perfkitbenchmarker/linux_benchmarks/benchbase_benchmark.py

@@ -0,0 +1,122 @@
+# Copyright 2025 PerfKitBenchmarker Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License
+
+"""Benchbase Benchmark.
+
+This is a set of benchmarks that measures OLTP performance of managed
+postgres databases using the Benchbase(https://github.com/cmu-db/benchbase)
+framework.
+"""
+
+from typing import Any, Dict, List
+
+from absl import flags
+from perfkitbenchmarker import benchmark_spec as bm_spec
+from perfkitbenchmarker import configs
+from perfkitbenchmarker import sample
+from perfkitbenchmarker.linux_packages import benchbase
+
+BENCHMARK_NAME: str = 'benchbase'
+BENCHMARK_CONFIG: str = """
+benchbase:
+  description: Runs Benchbase benchmark.
+  relational_db:
+    cloud: GCP
+    engine: spanner-postgres
+    spanner_nodes: 3
+    db_spec:
+      GCP:
+        zone: us-central1-f
+    db_disk_spec:
+      GCP:
+        disk_size: 2048
+        disk_type: pd-ssd
+    vm_groups:
+      clients:
+        vm_spec:
+          GCP:
+            machine_type: n2-standard-21
+            zone: us-central1-c
+        disk_spec:
+          GCP:
+            disk_size: 500
+            disk_type: pd-ssd
+"""
+
+FLAGS = flags.FLAGS
+
+
+def GetConfig(user_config: Dict[str, Any]) -> Dict[str, Any]:
+  """Loads and returns benchmark config.
+
+  Args:
+    user_config: user supplied configuration (flags and config file)
+
+  Returns:
+    loaded benchmark configuration
+  """
+  return configs.LoadConfig(BENCHMARK_CONFIG, user_config, BENCHMARK_NAME)
+
+
+# TODO(shuninglin): need to implement auth logic(automatic password gen)
+# for DSQL
+def Prepare(benchmark_spec: bm_spec.BenchmarkSpec) -> None:
+  """Prepares the benchmark by installing BenchBase and loading data.
+
+  Args:
+    benchmark_spec: The benchmark specification. Contains all data that is
+      required to run the benchmark.
+  """
+
+  vms = benchmark_spec.vms
+  client_vm = vms[0]
+
+  # Install BenchBase on the client VM
+  client_vm.Install('benchbase')
+
+  # Create the configuration file on the client VM
+  benchbase.CreateConfigFile(client_vm)
+  # TODO(shuninglin): Implement benchbase data loading
+
+
+def Run(benchmark_spec: bm_spec.BenchmarkSpec) -> List[sample.Sample]:
+  """Runs the BenchBase benchmark.
+
+  Args:
+    benchmark_spec: The benchmark specification.
+
+  Returns:
+    A list of sample.Sample objects.
+  """
+  del benchmark_spec  # Unused for now.
+  # TODO(shuninglin): Implement the run phase
+  # 1. Construct the run command:
+  # 2.1 Send the first run command as warmup
+  # 2.2 Sleep for warmup duration
+  # 2.3 Send the second run command as the main workload
+  # 3. Parse results from the output files
+
+  samples: List[sample.Sample] = []
+  return samples
+
+
+def Cleanup(benchmark_spec: bm_spec.BenchmarkSpec) -> None:
+  """Cleans up the benchmark.
+
+  Args:
+    benchmark_spec: The benchmark specification.
+  """
+  # Nothing to do in cleanup for now.
+  del benchmark_spec  # Unused for now.
+  pass