Class BenchmarkState

src/main/java/com/oltpbenchmark/BenchmarkState.java:26–116

import org.slf4j.LoggerFactory;

public final class BenchmarkState {

  private static final Logger LOG = LoggerFactory.getLogger(BenchmarkState.class);

  private final long testStartNs;
  private final CountDownLatch startBarrier;
  private final AtomicInteger notDoneCount;
  private volatile State state = State.WARMUP;

  /**
   * @param numThreads number of threads involved in the test: including the master thread.
   */
  public BenchmarkState(int numThreads) {
    startBarrier = new CountDownLatch(numThreads);
    notDoneCount = new AtomicInteger(numThreads);

    testStartNs = System.nanoTime();
  }

  // Protected by this

  public long getTestStartNs() {
    return testStartNs;
  }

  public State getState() {
    synchronized (this) {
      return state;
    }
  }

  /** Wait for all threads to call this. Returns once all the threads have entered. */
  public void blockForStart() {

    startBarrier.countDown();
    try {
      startBarrier.await();
    } catch (InterruptedException e) {
      throw new RuntimeException(e);
    }
  }

  public void startMeasure() {
    state = State.MEASURE;
  }

  public void startColdQuery() {
    state = State.COLD_QUERY;
  }

  public void startHotQuery() {
    state = State.MEASURE;
  }

  public void signalLatencyComplete() {
    state = State.LATENCY_COMPLETE;
  }