Example usage for com.amazonaws.services.dynamodbv2.model QueryResult getConsumedCapacity

Introduction

In this page you can find the example usage for com.amazonaws.services.dynamodbv2.model QueryResult getConsumedCapacity.

Prototype

public ConsumedCapacity getConsumedCapacity() 

Source Link

Document

The capacity units consumed by the Query operation.

Usage

From source file:com.amazon.janusgraph.diskstorage.dynamodb.DynamoDbDelegate.java

License:Open Source License

public QueryResult query(final QueryRequest request, final int permitsToConsume) throws BackendException {
    setUserAgent(request);//  w w w . j a v  a 2  s  .c  om
    QueryResult result;
    timedReadThrottle(QUERY, request.getTableName(), permitsToConsume);
    final Timer.Context apiTimerContext = getTimerContext(QUERY, request.getTableName());
    try {
        result = client.query(request);
    } catch (Exception e) {
        throw processDynamoDbApiException(e, QUERY, request.getTableName());
    } finally {
        apiTimerContext.stop();
    }
    meterConsumedCapacity(QUERY, result.getConsumedCapacity());
    measureItemCount(QUERY, request.getTableName(), result.getCount());
    return result;
}

---

From source file:com.amazon.janusgraph.diskstorage.dynamodb.QueryWorker.java

License:Open Source License

@Override
public QueryResultWrapper next() throws BackendException {
    final Query backoff = new ExponentialBackoff.Query(request, delegate, permitsToConsume);
    final QueryResult result = backoff.runWithBackoff();
    final ConsumedCapacity consumedCapacity = result.getConsumedCapacity();
    if (null != consumedCapacity) {
        permitsToConsume = Math.max((int) (consumedCapacity.getCapacityUnits() - 1.0), 1);
        totalCapacityUnits += consumedCapacity.getCapacityUnits();
    }//from   ww w  . j a v  a2 s.com

    if (result.getLastEvaluatedKey() != null && !result.getLastEvaluatedKey().isEmpty()) {
        request.setExclusiveStartKey(result.getLastEvaluatedKey());
    } else {
        markComplete();
    }
    // a update returned count
    returnedCount += result.getCount();

    // b update scanned count
    scannedCount += result.getScannedCount();
    // c add scanned finalItemList
    finalItemList.addAll(result.getItems());
    return new QueryResultWrapper(titanKey, result);
}

---

From source file:com.rapid7.diskstorage.dynamodb.DynamoDBDelegate.java

License:Open Source License

public QueryResult query(QueryRequest request, int permitsToConsume) throws BackendException {
    setUserAgent(request);/*  w  ww.ja v  a2  s .  co m*/
    QueryResult result;
    timedReadThrottle(QUERY, request.getTableName(), permitsToConsume);
    final Timer.Context apiTimerContext = getTimerContext(QUERY, request.getTableName());
    try {
        result = client.query(request);
    } catch (Exception e) {
        throw processDynamoDBAPIException(e, QUERY, request.getTableName());
    } finally {
        apiTimerContext.stop();
    }
    meterConsumedCapacity(QUERY, result.getConsumedCapacity());
    measureItemCount(QUERY, request.getTableName(), result.getCount());
    return result;
}

---

From source file:com.rapid7.diskstorage.dynamodb.QueryWorker.java

License:Open Source License

@Override
public QueryResultWrapper next() throws BackendException {
    ExponentialBackoff.Query backoff = new ExponentialBackoff.Query(request, delegate, permitsToConsume);
    QueryResult result = backoff.runWithBackoff();
    ConsumedCapacity consumedCapacity = result.getConsumedCapacity();
    if (null != consumedCapacity) {
        permitsToConsume = Math.max((int) (consumedCapacity.getCapacityUnits() - 1.0), 1);
        totalCapacityUnits += consumedCapacity.getCapacityUnits();
    }/*from  ww  w .  j  a  va  2s  .  c  o  m*/

    if (result.getLastEvaluatedKey() != null && !result.getLastEvaluatedKey().isEmpty()) {
        request.setExclusiveStartKey(result.getLastEvaluatedKey());
    } else {
        markComplete();
    }
    // a update returned count
    returnedCount += result.getCount();

    // b update scanned count
    scannedCount += result.getScannedCount();
    // c add scanned items
    items.addAll(result.getItems());
    return new QueryResultWrapper(titanKey, result);
}

---

From source file:org.apache.hadoop.dynamodb.preader.QueryRecordReadRequest.java

License:Open Source License

@Override
protected PageResults<Map<String, AttributeValue>> fetchPage(RequestLimit lim) {
    // Read from DynamoDB
    RetryResult<QueryResult> retryResult = context.getClient().queryTable(tableName,
            context.getSplit().getFilterPushdown(), lastEvaluatedKey, lim.items, context.getReporter());

    QueryResult result = retryResult.result;
    int retries = retryResult.retries;

    return new PageResults<>(result.getItems(), result.getLastEvaluatedKey(),
            result.getConsumedCapacity().getCapacityUnits(), retries);
}

---

From source file:org.iternine.jeppetto.dao.dynamodb.iterable.QueryIterable.java

License:Apache License

@Override
protected Iterator<Map<String, AttributeValue>> fetchItems() {
    QueryResult currentQueryResult = getDynamoDB().query(queryRequest);
    Iterator<Map<String, AttributeValue>> iterator = currentQueryResult.getItems().iterator();

    queryRequest.setExclusiveStartKey(currentQueryResult.getLastEvaluatedKey()); // Prepare for next query

    if (logger.isDebugEnabled()) {
        logger.debug(/*from   w  w  w  .j  a v a  2 s.  com*/
                "Queried {} using {}.  Took {} read capacity units, retrieved {} items, more items {} available.",
                getEnhancer().getBaseClass().getSimpleName(), queryRequest,
                currentQueryResult.getConsumedCapacity(), currentQueryResult.getCount(),
                currentQueryResult.getLastEvaluatedKey() == null ? "are not" : "are");
    }

    return iterator;
}

---