Class DBWorkload

src/main/java/com/oltpbenchmark/DBWorkload.java:46–860

import org.slf4j.LoggerFactory;

public class DBWorkload {
  private static final Logger LOG = LoggerFactory.getLogger(DBWorkload.class);

  private static final String SINGLE_LINE = StringUtil.repeat("=", 70);

  private static final String RATE_DISABLED = "disabled";
  private static final String RATE_UNLIMITED = "unlimited";

  /**
   * @param args
   * @throws Exception
   */
  public static void main(String[] args) throws Exception {

    // create the command line parser
    CommandLineParser parser = new DefaultParser();

    XMLConfiguration pluginConfig = buildConfiguration("config/plugin.xml");

    Options options = buildOptions(pluginConfig);

    CommandLine argsLine = parser.parse(options, args);

    if (argsLine.hasOption("h")) {
      printUsage(options);
      return;
    } else if (!argsLine.hasOption("c")) {
      LOG.error("Missing Configuration file");
      printUsage(options);
      return;
    } else if (!argsLine.hasOption("b")) {
      LOG.error("Missing Benchmark Class to load");
      printUsage(options);
      return;
    }

    // Monitoring setup.
    ImmutableMonitorInfo.Builder builder = ImmutableMonitorInfo.builder();
    if (argsLine.hasOption("im")) {
      builder.monitoringInterval(Integer.parseInt(argsLine.getOptionValue("im")));
    }
    if (argsLine.hasOption("mt")) {
      switch (argsLine.getOptionValue("mt")) {
        case "advanced":
          builder.monitoringType(MonitorInfo.MonitoringType.ADVANCED);
          break;
        case "throughput":
          builder.monitoringType(MonitorInfo.MonitoringType.THROUGHPUT);
          break;
        default:
          throw new ParseException(
              "Monitoring type '"
                  + argsLine.getOptionValue("mt")
                  + "' is undefined, allowed values are: advanced/throughput");
      }
    }
    MonitorInfo monitorInfo = builder.build();