如何用DynamoDB V2 Enhanced SDK获取匹配分区键的单条记录

你的核心问题是：调用items().stream().collect(Collectors.toList())时，SDK会自动遍历所有分页结果，哪怕设置了limit(1)也会拉取同分区键下的全部记录。要实现只获取第一条匹配记录，需要直接使用迭代器终止查询，而不是收集所有结果。

正确实现代码 #

import software.amazon.awssdk.enhanced.dynamodb.DynamoDbTable;
import software.amazon.awssdk.enhanced.dynamodb.Key;
import software.amazon.awssdk.enhanced.dynamodb.model.QueryConditional;
import software.amazon.awssdk.enhanced.dynamodb.model.QueryEnhancedRequest;
import java.util.Iterator;

private static DynamoDbTable<MyFooEntity> table;

public MyFooEntity getOneItemByPartitionKey(String partitionKey) {
    // 构建分区键查询条件
    Key key = Key.builder().partitionValue(partitionKey).build();
    QueryEnhancedRequest queryRequest = QueryEnhancedRequest.builder()
            .queryConditional(QueryConditional.keyEqualTo(key))
            .limit(1) // 限制只返回1条记录
            // 可选：指定排序顺序，默认升序；设为false则按排序键降序取第一条
            // .scanIndexForward(false)
            .build();

    // 使用迭代器直接获取第一条记录，不会触发全量拉取
    Iterator<MyFooEntity> itemIterator = table.query(queryRequest).items().iterator();
    return itemIterator.hasNext() ? itemIterator.next() : null;
}

为什么之前的limit设置无效？ #

DynamoDB Enhanced SDK的items()方法返回的是支持分页的Iterable。当你调用stream().collect(Collectors.toList())时，SDK会自动处理分页逻辑——即使设置了limit(1)，它也会持续发起后续请求拉取剩余数据，直到没有更多结果为止，最终把所有匹配记录都收集到List中。

而直接使用迭代器获取第一条记录后就终止，SDK只会发起一次查询请求，获取limit=1的结果后就停止，不会再拉取后续数据，完美解决大数量分区键下的性能问题。

可选优化：明确指定排序顺序 #

如果需要获取同分区键下特定排序的记录（比如最新/最早的一条），可以通过scanIndexForward参数控制：

• scanIndexForward(true)（默认）：按排序键升序排列，返回最小排序键的记录
• scanIndexForward(false)：按排序键降序排列，返回最大排序键的记录

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