Class BenchmarkModule

src/main/java/com/oltpbenchmark/api/BenchmarkModule.java:38–438

Base class for all benchmark implementations

/** Base class for all benchmark implementations */
public abstract class BenchmarkModule {
  private static final Logger LOG = LoggerFactory.getLogger(BenchmarkModule.class);

  /** The workload configuration for this benchmark invocation */
  protected final WorkloadConfiguration workConf;

  /** Class loader variable for this benchmark */
  protected ClassLoader classLoader;

  /** These are the variations of the Procedure's Statement SQL */
  protected final StatementDialects dialects;

  /** Supplemental Procedures */
  private final Set<Class<? extends Procedure>> supplementalProcedures = new HashSet<>();

  /** A single Random object that should be re-used by all a benchmark's components */
  private static final ThreadLocal<Random> rng = new ThreadLocal<>();

  private AbstractCatalog catalog = null;

  /**
   * Constructor!
   *
   * @param workConf
   */
  public BenchmarkModule(WorkloadConfiguration workConf) {
    this.workConf = workConf;
    this.dialects = new StatementDialects(workConf);
    // setClassLoader();
    this.classLoader = Thread.currentThread().getContextClassLoader();
  }

  /**
   * Instantiates the classLoader variable, needs to be overwritten if benchmark uses a custom
   * implementation.
   */
  protected void setClassLoader() {
    this.classLoader = Thread.currentThread().getContextClassLoader();
  }

  // --------------------------------------------------------------------------
  // DATABASE CONNECTION
  // --------------------------------------------------------------------------

  public final Connection makeConnection() throws SQLException {

    if (StringUtils.isEmpty(workConf.getUsername())) {
      return DriverManager.getConnection(workConf.getUrl());
    } else {
      return DriverManager.getConnection(
          workConf.getUrl(), workConf.getUsername(), workConf.getPassword());
    }
  }

  private String afterLoadScriptPath = null;

  public final void setAfterLoadScriptPath(String scriptPath) throws FileNotFoundException {
    if (scriptPath != null) scriptPath = scriptPath.trim();