Java 21 virtual threads

src/main/java/com/oltpbenchmark/DBWorkload.java

@@ -85,6 +85,11 @@ public static void main(String[] args) throws Exception {
  intervalMonitor = Integer.parseInt(argsLine.getOptionValue("im"));
  }
 
+ Boolean useVirtualThreads = false;
+ if (argsLine.hasOption("vt")) {
+ useVirtualThreads = Boolean.parseBoolean(argsLine.getOptionValue("vt"));
+ }
+
  // -------------------------------------------------------------------
  // GET PLUGIN LIST
  // -------------------------------------------------------------------
@@ -518,7 +523,7 @@ public static void main(String[] args) throws Exception {
  if (isBooleanOptionSet(argsLine, "execute")) {
  // Bombs away!
  try {
- Results r = runWorkload(benchList, intervalMonitor);
+ Results r = runWorkload(benchList, intervalMonitor, useVirtualThreads);
  writeOutputs(r, activeTXTypes, argsLine, xmlConfig);
  writeHistograms(r);
 
@@ -567,6 +572,7 @@ private static Options buildOptions(XMLConfiguration pluginConfig) {
  "Base directory for the result files, default is current directory");
  options.addOption(null, "dialects-export", true, "Export benchmark SQL to a dialects file");
  options.addOption("jh", "json-histograms", true, "Export histograms to JSON file");
+ options.addOption("vt", "virtual-threads", true, "Use virtual threads instead of real threads");
  return options;
  }
 
@@ -733,7 +739,8 @@ private static void runLoader(BenchmarkModule bench)
  bench.loadDatabase();
  }
 
- private static Results runWorkload(List<BenchmarkModule> benchList, int intervalMonitor)
+ private static Results runWorkload(
+ List<BenchmarkModule> benchList, int intervalMonitor, Boolean useVirtualThreads)
  throws IOException {
  List<Worker<?>> workers = new ArrayList<>();
  List<WorkloadConfiguration> workConfs = new ArrayList<>();
@@ -748,7 +755,8 @@ private static Results runWorkload(List<BenchmarkModule> benchList, int interval
  bench.getBenchmarkName().toUpperCase(), num_phases, (num_phases > 1 ? "s" : "")));
  workConfs.add(bench.getWorkloadConfiguration());
  }
- Results r = ThreadBench.runRateLimitedBenchmark(workers, workConfs, intervalMonitor);
+ Results r =
+ ThreadBench.runRateLimitedBenchmark(workers, workConfs, intervalMonitor, useVirtualThreads);
  LOG.info(SINGLE_LINE);
  LOG.info("Rate limited reqs/s: {}", r);
  return r;

src/main/java/com/oltpbenchmark/ThreadBench.java

@@ -37,31 +37,42 @@ public class ThreadBench implements Thread.UncaughtExceptionHandler {
  private final List<WorkloadConfiguration> workConfs;
  private final ArrayList<LatencyRecord.Sample> samples = new ArrayList<>();
  private final int intervalMonitor;
+ private final Boolean useVirtualThreads;
 
  private ThreadBench(
  List<? extends Worker<? extends BenchmarkModule>> workers,
  List<WorkloadConfiguration> workConfs,
- int intervalMonitoring) {
+ int intervalMonitoring,
+ Boolean useVirtualThreads) {
  this.workers = workers;
  this.workConfs = workConfs;
  this.workerThreads = new ArrayList<>(workers.size());
  this.intervalMonitor = intervalMonitoring;
  this.testState = new BenchmarkState(workers.size() + 1);
+ this.useVirtualThreads = useVirtualThreads;
  }
 
  public static Results runRateLimitedBenchmark(
  List<Worker<? extends BenchmarkModule>> workers,
  List<WorkloadConfiguration> workConfs,
- int intervalMonitoring) {
- ThreadBench bench = new ThreadBench(workers, workConfs, intervalMonitoring);
+ int intervalMonitoring,
+ Boolean useVirtualThreads) {
+ ThreadBench bench = new ThreadBench(workers, workConfs, intervalMonitoring, useVirtualThreads);
  return bench.runRateLimitedMultiPhase();
  }
 
  private void createWorkerThreads() {
 
  for (Worker<?> worker : workers) {
  worker.initializeState();
- Thread thread = new Thread(worker);
+
+ Thread thread;
+ if (useVirtualThreads) {
+ thread = Thread.ofVirtual().unstarted(worker);
+ } else {
+ thread = new Thread(worker);
+ }
+
  thread.setUncaughtExceptionHandler(this);
  thread.start();
  this.workerThreads.add(thread);