com.github.noconnor.junitperf.statements.EvaluationTask Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of junitperf Show documentation

API performance testing framework built using JUnit

There is a newer version: 1.36.0

package com.github.noconnor.junitperf.statements;

import com.github.noconnor.junitperf.statistics.StatisticsCalculator;
import com.google.common.util.concurrent.RateLimiter;
import lombok.Builder;
import org.junit.runners.model.Statement;

import java.util.function.Supplier;

import static java.lang.System.nanoTime;
import static java.util.Objects.nonNull;
import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static java.util.concurrent.TimeUnit.NANOSECONDS;

final class EvaluationTask implements Runnable {

  private final Statement statement;
  private final RateLimiter rateLimiter;
  private final Supplier terminator;
  private final StatisticsCalculator stats;
  private final long warmUpPeriodNs;

  @Builder
  EvaluationTask(Statement statement, RateLimiter rateLimiter, StatisticsCalculator stats, int warmUpPeriodMs) {
    this(statement, rateLimiter, () -> Thread.currentThread().isInterrupted(), stats, warmUpPeriodMs);
  }

  // Test only
  EvaluationTask(Statement statement,
    RateLimiter rateLimiter,
    Supplier terminator,
    StatisticsCalculator stats,
    int warmUpPeriodMs) {
    this.statement = statement;
    this.rateLimiter = rateLimiter;
    this.terminator = terminator;
    this.stats = stats;
    this.warmUpPeriodNs = NANOSECONDS.convert(warmUpPeriodMs > 0 ? warmUpPeriodMs : 0, MILLISECONDS);
  }

  @Override
  public void run() {
    long startTimeNs = nanoTime();
    long startMeasurements = startTimeNs + warmUpPeriodNs;
    while (!terminator.get()) {
      waitForPermit();
      evaluateStatement(startMeasurements);
    }
  }

  private void evaluateStatement(long startMeasurements) {
    if (nanoTime() < startMeasurements) {
      try {
        statement.evaluate();
      } catch (Throwable throwable) {
        // IGNORE
      }
    } else {
      long startTimeNs = nanoTime();
      try {
        statement.evaluate();
        stats.addLatencyMeasurement(nanoTime() - startTimeNs);
        stats.incrementEvaluationCount();
      } catch (InterruptedException e) { // NOSONAR
        // IGNORE - no metrics
      } catch (Throwable throwable) {
        stats.incrementEvaluationCount();
        stats.incrementErrorCount();
        stats.addLatencyMeasurement(nanoTime() - startTimeNs);
      }
    }
  }

  private void waitForPermit() {
    if (nonNull(rateLimiter)) {
      rateLimiter.acquire();
    }
  }

}