public Iterator getPartitionKeys()

in spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java [110:122]

• 13 lines of code
• 2 McCabe index (conditional complexity)

    public Iterator<PartitionKey> getPartitionKeys(KeyspaceTablePair table, final BigInteger prevToken, final BigInteger token) {
        try {
            RetryStrategy retryStrategy = retryStrategyProvider.get();
            return retryStrategy.retry(
                () -> Uninterruptibles.getUninterruptibly(fetchPartitionKeys(table, prevToken, token))
            );
        }
        catch (Exception ex) {
            throw new RuntimeException(String.format("Unable to get partition keys (%s, %s] in table (%s) from cluster (%s)",
                                                     prevToken, token, table, clusterId.name()),
                                       ex);
        }
    }