コード例 #1
0
ファイル: Benchmark.java プロジェクト: JingitLabs/voltdb
  /**
   * Constructor for benchmark instance. Configures VoltDB client and prints configuration.
   *
   * @param cliConfig Parsed & validated CLI options.
   */
  public Benchmark(BenchmarkConfig cliConfig) {

    this.cliConfig = cliConfig;

    ClientConfig clientConfig = new ClientConfig("", "", new StatusListener());
    // Throttle so that ad hoc queries don't get rejected with "planner not available".
    if (cliConfig.querythrottle > 0) {
      System.out.printf(
          "Throttling maximum outstanding transactions to %d\n", cliConfig.querythrottle);
      clientConfig.setMaxOutstandingTxns(cliConfig.querythrottle);
    }
    client = ClientFactory.createClient(clientConfig);

    periodicStatsContext = client.createStatsContext();
    fullStatsContext = client.createStatsContext();

    tracer = new QueryTracer(cliConfig.querytracefile);

    System.out.print(HORIZONTAL_RULE);
    System.out.println(" Command Line Configuration");
    System.out.println(HORIZONTAL_RULE);
    System.out.println(cliConfig.getConfigDumpString());
  }
コード例 #2
0
ファイル: ClientMain.java プロジェクト: pettermahlen/voltdb
  /**
   * Constructor that initializes the framework portions of the client. Creates a Volt client and
   * connects it to all the hosts provided on the command line with the specified username and
   * password
   *
   * @param args
   */
  public ClientMain(String args[]) {
    /*
     * Input parameters: HOST=host:port (may occur multiple times)
     * USER=username PASSWORD=password
     */

    // default values
    String username = "";
    String password = "";
    ControlState state = ControlState.PREPARING; // starting state
    String reason = ""; // and error string
    int transactionRate = -1;
    int id = 0;
    boolean exitOnCompletion = true;
    float checkTransaction = 0;
    boolean checkTables = false;
    String statsDatabaseURL = null;
    String deploymentFilePath = null;
    int statsPollInterval = 10000;
    Integer maxOutstanding = null;

    // scan the inputs once to read everything but host names
    for (final String arg : args) {
      final String[] parts = arg.split("=", 2);
      if (parts.length == 1) {
        state = ControlState.ERROR;
        reason = "Invalid parameter: " + arg;
        break;
      } else if (parts[1].startsWith("${")) {
        continue;
      } else if (parts[0].equals("USER")) {
        username = parts[1];
      } else if (parts[0].equals("PASSWORD")) {
        password = parts[1];
      } else if (parts[0].equals("EXITONCOMPLETION")) {
        exitOnCompletion = Boolean.parseBoolean(parts[1]);
      } else if (parts[0].equals("TXNRATE")) {
        transactionRate = Integer.parseInt(parts[1]);
      } else if (parts[0].equals("ID")) {
        id = Integer.parseInt(parts[1]);
      } else if (parts[0].equals("CHECKTRANSACTION")) {
        checkTransaction = Float.parseFloat(parts[1]);
      } else if (parts[0].equals("CHECKTABLES")) {
        checkTables = Boolean.parseBoolean(parts[1]);
      } else if (parts[0].equals("STATSDATABASEURL")) {
        statsDatabaseURL = parts[1];
      } else if (parts[0].equals("STATSPOLLINTERVAL")) {
        statsPollInterval = Integer.parseInt(parts[1]);
      } else if (parts[0].equals("DEPLOYMENTFILEPATH")) {
        deploymentFilePath = parts[1];
      } else if (parts[0].equals("MAXOUTSTANDING")) {
        maxOutstanding = Integer.parseInt(parts[1]);
      }
    }
    StatsUploaderSettings statsSettings = null;
    if (statsDatabaseURL != null) {
      try {
        statsSettings =
            new StatsUploaderSettings(
                statsDatabaseURL, getApplicationName(), getSubApplicationName(), statsPollInterval);
      } catch (Exception e) {
        System.err.println(e.getMessage());
        // e.printStackTrace();
        statsSettings = null;
      }
    }
    ClientConfig clientConfig = new ClientConfig(username, password);
    clientConfig.setExpectedOutgoingMessageSize(getExpectedOutgoingMessageSize());
    clientConfig.setHeavyweight(useHeavyweightClient());
    clientConfig.setStatsUploaderSettings(statsSettings);
    if (maxOutstanding != null) {
      clientConfig.setMaxOutstandingTxns(maxOutstanding);
    }

    m_voltClient = ClientFactory.createClient(clientConfig);

    m_id = id;
    m_exitOnCompletion = exitOnCompletion;
    m_username = username;
    m_password = password;
    m_txnRate = transactionRate;
    m_txnsPerMillisecond = transactionRate / 1000.0;
    m_deploymentFilePath = deploymentFilePath;

    // report any errors that occurred before the client was instantiated
    if (state != ControlState.PREPARING) setState(state, reason);

    // scan the inputs again looking for host connections
    boolean atLeastOneConnection = false;
    for (final String arg : args) {
      final String[] parts = arg.split("=", 2);
      if (parts.length == 1) {
        continue;
      } else if (parts[0].equals("HOST")) {
        final String hostnport[] = parts[1].split("\\:", 2);
        m_host = hostnport[0];
        String host = hostnport[0];
        int port;
        if (hostnport.length < 2) {
          port = Client.VOLTDB_SERVER_PORT;
        } else {
          port = Integer.valueOf(hostnport[1]);
        }
        try {
          System.err.println("Creating connection to  " + host + ":" + port);
          createConnection(host, port);
          System.err.println("Created connection.");
          atLeastOneConnection = true;
        } catch (final Exception ex) {
          setState(
              ControlState.ERROR, "createConnection to " + arg + " failed: " + ex.getMessage());
        }
      }
    }
    if (!atLeastOneConnection) setState(ControlState.ERROR, "No HOSTS specified on command line.");
    m_checkTransaction = checkTransaction;
    m_checkTables = checkTables;
    m_constraints = new LinkedHashMap<Pair<String, Integer>, Expression>();

    m_countDisplayNames = getTransactionDisplayNames();
    m_counts = new AtomicLong[m_countDisplayNames.length];
    for (int ii = 0; ii < m_counts.length; ii++) {
      m_counts[ii] = new AtomicLong(0);
    }
  }