org.apache.cassandra.thrift.TokenRange Java Examples

/**
 * The same as {@code describeRing(String)} but converts TokenRange to the String for JMX compatibility
 *
 * @param keyspace The keyspace to fetch information about
 *
 * @return a List of TokenRange(s) converted to String for the given keyspace
 */
public List<String> describeRingJMX(String keyspace) throws IOException
{
    List<TokenRange> tokenRanges;
    try
    {
        tokenRanges = describeRing(keyspace);
    }
    catch (InvalidRequestException e)
    {
        throw new IOException(e.getMessage());
    }
    List<String> result = new ArrayList<>(tokenRanges.size());

    for (TokenRange tokenRange : tokenRanges)
        result.add(tokenRange.toString());

    return result;
}
 

private List<String> getAllServers(List<TokenRange> input) {
  HashMap<String, Integer> map = new HashMap<String, Integer>(input.size());
  for (TokenRange thisRange : input) {
    List<String> servers = thisRange.rpc_endpoints;
    for (String newServer : servers) {
      map.put(newServer, new Integer(1));
    }
  }

  return new ArrayList<String>(map.keySet());
}
 

public SplitCallable(TokenRange tr, Configuration conf)
{
    this.range = tr;
    this.conf = conf;
}
 

/**
 * The same as {@code describeRing(String)} but considers only the part of the ring formed by nodes in the local DC.
 */
public List<TokenRange> describeLocalRing(String keyspace) throws InvalidRequestException
{
    return describeRing(keyspace, true);
}
 

private List<TokenRange> describeRing(String keyspace, boolean includeOnlyLocalDC) throws InvalidRequestException
{
    if (!Schema.instance.getKeyspaces().contains(keyspace))
        throw new InvalidRequestException("No such keyspace: " + keyspace);

    if (keyspace == null || Keyspace.open(keyspace).getReplicationStrategy() instanceof LocalStrategy)
        throw new InvalidRequestException("There is no ring for the keyspace: " + keyspace);

    List<TokenRange> ranges = new ArrayList<>();
    Token.TokenFactory tf = getPartitioner().getTokenFactory();

    Map<Range<Token>, List<InetAddress>> rangeToAddressMap =
            includeOnlyLocalDC
                    ? getRangeToAddressMapInLocalDC(keyspace)
                    : getRangeToAddressMap(keyspace);

    for (Map.Entry<Range<Token>, List<InetAddress>> entry : rangeToAddressMap.entrySet())
    {
        Range range = entry.getKey();
        List<InetAddress> addresses = entry.getValue();
        List<String> endpoints = new ArrayList<>(addresses.size());
        List<String> rpc_endpoints = new ArrayList<>(addresses.size());
        List<EndpointDetails> epDetails = new ArrayList<>(addresses.size());

        for (InetAddress endpoint : addresses)
        {
            EndpointDetails details = new EndpointDetails();
            details.host = endpoint.getHostAddress();
            details.datacenter = DatabaseDescriptor.getEndpointSnitch().getDatacenter(endpoint);
            details.rack = DatabaseDescriptor.getEndpointSnitch().getRack(endpoint);

            endpoints.add(details.host);
            rpc_endpoints.add(getRpcaddress(endpoint));

            epDetails.add(details);
        }

        TokenRange tr = new TokenRange(tf.toString(range.left.getToken()), tf.toString(range.right.getToken()), endpoints)
                                .setEndpoint_details(epDetails)
                                .setRpc_endpoints(rpc_endpoints);

        ranges.add(tr);
    }

    return ranges;
}
 

protected RingConnOption(List<TokenRange> servers) {
  this.servers = getAllServers(servers);
}
 

/**
 * Reset the servers in the ring.
 */
public void resetRing(List<TokenRange> servers) {
  this.servers = getAllServers(servers);
}
 

public RandomizerOption(List<TokenRange> rings) {
  super(rings);
  generator = new Random();
}
 

public RoundRobinOption(List<TokenRange> rings) {
  super(rings);
  lastUsedIndex = 0;
}
 

/**
 * The TokenRange for a given keyspace.
 *
 * @param keyspace The keyspace to fetch information about
 *
 * @return a List of TokenRange(s) for the given keyspace
 *
 * @throws InvalidRequestException if there is no ring information available about keyspace
 */
public List<TokenRange> describeRing(String keyspace) throws InvalidRequestException
{
    return describeRing(keyspace, false);
}