com.datastax.driver.core.SpeculativeExecutionInfoFinder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dse-java-driver-core Show documentation

A driver for DataStax Enterprise (DSE) and Apache Cassandra 1.2+ clusters that works exclusively with the Cassandra Query Language version 3 (CQL3) and Cassandra's binary protocol, supporting DSE-specific features such as geospatial types, DSE Graph and DSE authentication.

There is a newer version: 2.4.0

Show newest version

/*
 * Copyright DataStax, Inc.
 *
 * This software can be used solely with DataStax Enterprise. Please consult the license at
 * http://www.datastax.com/terms/datastax-dse-driver-license-terms
 */
package com.datastax.driver.core;

import static com.datastax.driver.core.InsightsSchema.*;
import static com.datastax.driver.core.PackageUtil.getNamespace;

import com.datastax.driver.core.policies.ConstantSpeculativeExecutionPolicy;
import com.datastax.driver.core.policies.PercentileSpeculativeExecutionPolicy;
import com.datastax.driver.core.policies.SpeculativeExecutionPolicy;
import java.util.HashMap;
import java.util.Map;

class SpeculativeExecutionInfoFinder {
  SpeculativeExecutionInfo getSpeculativeExecutionInfo(
      SpeculativeExecutionPolicy speculativeExecutionPolicy) {
    String type = speculativeExecutionPolicy.getClass().getSimpleName();
    String namespace = getNamespace(speculativeExecutionPolicy.getClass());

    Map options = new HashMap();

    if (speculativeExecutionPolicy instanceof ConstantSpeculativeExecutionPolicy) {
      ConstantSpeculativeExecutionPolicy constantSpeculativeExecutionPolicy =
          (ConstantSpeculativeExecutionPolicy) speculativeExecutionPolicy;
      options.put(
          "constantDelayMillis", constantSpeculativeExecutionPolicy.getConstantDelayMillis());
      options.put(
          "maxSpeculativeExecutions",
          constantSpeculativeExecutionPolicy.getMaxSpeculativeExecutions());
    } else if (speculativeExecutionPolicy instanceof PercentileSpeculativeExecutionPolicy) {
      PercentileSpeculativeExecutionPolicy percentileSpeculativeExecutionPolicy =
          (PercentileSpeculativeExecutionPolicy) speculativeExecutionPolicy;
      options.put(
          "maxSpeculativeExecutions",
          percentileSpeculativeExecutionPolicy.getMaxSpeculativeExecutions());
      options.put("percentile", percentileSpeculativeExecutionPolicy.getPercentile());
    }
    return new SpeculativeExecutionInfo(type, namespace, options);
  }
}