diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AbstractDynamoDbStore.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AbstractDynamoDbStore.java index 8ecf06fba..4d48b2f5f 100644 --- a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AbstractDynamoDbStore.java +++ b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AbstractDynamoDbStore.java @@ -68,6 +68,15 @@ public class AbstractDynamoDbStore { } } + protected List> scan(ScanRequest scanRequest, int max) { + + return db().scanPaginator(scanRequest) + .items() + .stream() + .limit(max) + .collect(Collectors.toList()); + } + static void writeInBatches(final Iterable items, final Consumer> action) { final List batch = new ArrayList<>(DYNAMO_DB_MAX_BATCH_SIZE);