Batch Query in DynamoDB Using Java (Guide w/ Code Examples)

DynamoDB has no BatchQuery API. To retrieve item collections for several partition keys, issue one Query per partition and paginate each. BatchGetItem is for known complete primary keys.

This example uses Java 17+ and the software.amazon.awssdk:dynamodb SDK 2.x dependency. Configure an AWS Region and credentials through your usual AWS profile or runtime role. The Orders table has String keys pk and sk; adapt the table and keys to your schema.

import java.util.*;
import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
import software.amazon.awssdk.services.dynamodb.model.*;

public class Example {
    static AttributeValue s(String value) { return AttributeValue.builder().s(value).build(); }
    static Map<String, AttributeValue> key(String order) {
        return Map.of("pk", s("CUSTOMER#123"), "sk", s(order));
    }

    public static void main(String[] args) throws InterruptedException {
        try (DynamoDbClient ddb = DynamoDbClient.create()) {
            for (String customer : List.of("CUSTOMER#123", "CUSTOMER#456")) {
                QueryRequest request = QueryRequest.builder().tableName("Orders")
                    .keyConditionExpression("#pk = :pk")
                    .expressionAttributeNames(Map.of("#pk", "pk"))
                    .expressionAttributeValues(Map.of(":pk", s(customer))).build();
                for (QueryResponse page : ddb.queryPaginator(request)) {
                    page.items().forEach(System.out::println);
                }
            }
        }
    }
}

This example runs sequentially. If you introduce concurrency, bound it to the table capacity and handle throttling.

Reference: AWS Query API. For related access patterns, see DynamoDB query examples.