如何在AWS SDK v2的DynamoDB增强客户端中查询多个partitionKey

DynamoDB的Query API本身仅支持针对单个PartitionKey发起查询，无法直接通过一次Query操作获取多个不同PartitionKey的记录。要实现你所需的需求，有两种常用且推荐的方案：

方案一：多次Query后合并结果 #

针对每个目标PartitionKey单独执行Query请求，再在客户端将所有结果合并。这种方式能充分利用DynamoDB的索引优化，性能远优于全表扫描，是生产环境的首选方案。

示例代码：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import software.amazon.awssdk.enhanced.dynamodb.QueryConditional;
import software.amazon.awssdk.enhanced.dynamodb.PageIterable;

List<String> targetPartitionKeys = Arrays.asList("movie01", "movie02", "movie04");
List<MovieActor> allResults = new ArrayList<>();

for (String partitionKey : targetPartitionKeys) {
    QueryConditional keyEqual = QueryConditional.keyEqualTo(b -> b.partitionValue(partitionKey));
    PageIterable<MovieActor> pagedResults = movieActorTable.query(keyEqual);
    
    // 遍历分页结果并收集所有记录
    pagedResults.stream()
                .flatMap(page -> page.items().stream())
                .forEach(allResults::add);
}

// allResults 即为所有目标PartitionKey对应的记录集合

方案二：使用BatchGetItem API #

如果明确知道每个PartitionKey对应的完整主键（PartitionKey + SortKey），可以用BatchGetItem批量获取指定主键的记录。但如果SortKey不固定，这种方式无法覆盖所有场景。

示例代码（假设已知每个PartitionKey对应的SortKey）：

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import software.amazon.awssdk.services.dynamodb.model.BatchGetItemRequest;
import software.amazon.awssdk.services.dynamodb.model.BatchGetItemResponse;
import software.amazon.awssdk.services.dynamodb.model.Key;
import software.amazon.awssdk.enhanced.dynamodb.DynamoDbEnhancedClient;
import software.amazon.awssdk.enhanced.dynamodb.TableSchema;

List<Key> targetKeys = Arrays.asList(
    Key.builder().partitionValue("movie01").sortValue("actor01").build(),
    Key.builder().partitionValue("movie02").sortValue("actor02").build(),
    Key.builder().partitionValue("movie04").sortValue("actor04").build()
);

BatchGetItemResponse response = dynamoDbClient.batchGetItem(b -> b
    .requestItems(Map.of(
        "MovieActorTable", BatchGetItemRequest.builder()
            .keys(targetKeys)
            .build()
    ))
);

// 将返回的DynamoDB条目转换为自定义实体类
List<MovieActor> results = response.responses().get("MovieActorTable")
    .stream()
    .map(item -> DynamoDbEnhancedClient.create()
        .table("MovieActorTable", TableSchema.fromBean(MovieActor.class))
        .itemToBean(item))
    .collect(Collectors.toList());

不推荐方案：Scan操作 #

虽然可以通过Scan配合FilterExpression筛选多个PartitionKey，但Scan会遍历整个表，性能极低且消耗大量读取容量单位（RCU），仅适合数据量极小的测试场景，生产环境严禁使用。

内容的提问来源于stack exchange，提问作者csujiabin