Java SDK 通过迭代器读取主键范围内的多行数据,并自动处理底层分批请求和翻页。
前提条件
安装 Tablestore Java SDK并初始化客户端。
功能说明
调用 createRangeIterator 方法创建指定主键范围的行数据迭代器。当前批次的数据耗尽后,迭代器自动发起下一批请求,无需记录 nextStartPrimaryKey 或循环调用 getRange。
public Iterator<Row> createRangeIterator(RangeIteratorParameter rangeIteratorParameter) throws TableStoreException, ClientException
以下示例从数据表 range_iter_demo 中主键值 row1 开始正向迭代到表末尾,逐行输出。
String tableName = "range_iter_demo";
RangeIteratorParameter parameter = new RangeIteratorParameter(tableName);
// 起始主键(含)
PrimaryKeyBuilder startPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
startPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.fromString("row1"));
parameter.setInclusiveStartPrimaryKey(startPkBuilder.build());
// 结束主键(不含),INF_MAX 表示读取到表末尾
PrimaryKeyBuilder endPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
endPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MAX);
parameter.setExclusiveEndPrimaryKey(endPkBuilder.build());
parameter.setMaxVersions(1);
Iterator<Row> iterator = client.createRangeIterator(parameter);
int count = 0;
while (iterator.hasNext()) {
Row row = iterator.next();
System.out.println(row);
count++;
}
System.out.println("Total rows iterated: " + count);
参数说明
通过 RangeIteratorParameter 携带迭代查询条件,各参数说明如下。
|
名称 |
类型 |
说明 |
|
tableName(必选) |
String |
读取的数据表名称。 |
|
inclusiveStartPrimaryKey(必选) |
PrimaryKey |
起始主键信息,包括主键列名称和主键值。
|
|
exclusiveEndPrimaryKey(必选) |
PrimaryKey |
结束主键信息,包括主键列名称和主键值。
|
|
direction(可选) |
Direction |
读取方向。
|
|
maxVersions(可选) |
OptionalValue<Integer> |
每个属性列返回的最大版本数。 如果符合查询条件的数据版本数量超过设置的最大版本数,按从新到旧的顺序返回指定版本数量的数据。 |
|
timeRange(可选) |
OptionalValue<TimeRange> |
数据版本范围。 数据表的每个属性列可以有不同的数据版本,设置版本范围后仅返回该范围内的数据。 |
|
maxCount(可选) |
int |
迭代器返回的最大总行数,必须大于 0。 |
|
bufferSize(可选) |
int |
每次底层请求拉取的行数,必须大于 0。 适当减小此值可降低大数据量场景下的内存峰值;当前批次数据耗尽后,迭代器自动从上次断点发起下一次请求。 |
|
columnsToGet(可选) |
Set<String> |
指定读取的数据列,可以是主键列或属性列。
|
|
filter(可选) |
OptionalValue<Filter> |
过滤条件,详情请参见过滤器。 如果同时设置 |
返回值
createRangeIterator 返回 Iterator<Row>。调用 hasNext() 判断是否仍有数据,调用 next() 获取下一行 Row 对象。
场景示例
反向迭代
通过 setDirection(Direction.BACKWARD) 反向迭代。反向迭代时起始主键必须大于结束主键。
String tableName = "range_iter_demo";
RangeIteratorParameter parameter = new RangeIteratorParameter(tableName);
parameter.setDirection(Direction.BACKWARD);
// 反向迭代时,起始主键必须大于结束主键
PrimaryKeyBuilder startPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
startPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MAX);
parameter.setInclusiveStartPrimaryKey(startPkBuilder.build());
PrimaryKeyBuilder endPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
endPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.fromString("row1"));
parameter.setExclusiveEndPrimaryKey(endPkBuilder.build());
parameter.setMaxVersions(1);
Iterator<Row> iterator = client.createRangeIterator(parameter);
while (iterator.hasNext()) {
Row row = iterator.next();
System.out.println(row);
}
限制迭代总行数
通过 setMaxCount 限制迭代器最多返回的总行数。适用于按需取 Top N 的场景,避免无意义扫描。
String tableName = "range_iter_demo";
RangeIteratorParameter parameter = new RangeIteratorParameter(tableName);
PrimaryKeyBuilder startPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
startPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MIN);
parameter.setInclusiveStartPrimaryKey(startPkBuilder.build());
PrimaryKeyBuilder endPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
endPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MAX);
parameter.setExclusiveEndPrimaryKey(endPkBuilder.build());
parameter.setMaxVersions(1);
// 最多返回 3 行
parameter.setMaxCount(3);
Iterator<Row> iterator = client.createRangeIterator(parameter);
while (iterator.hasNext()) {
Row row = iterator.next();
System.out.println(row);
}
内存优化分批
通过 setBufferSize 控制每次底层请求拉取的行数。适当减小可降低大数据量场景下的内存峰值。
String tableName = "range_iter_demo";
RangeIteratorParameter parameter = new RangeIteratorParameter(tableName);
PrimaryKeyBuilder startPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
startPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MIN);
parameter.setInclusiveStartPrimaryKey(startPkBuilder.build());
PrimaryKeyBuilder endPkBuilder = PrimaryKeyBuilder.createPrimaryKeyBuilder();
endPkBuilder.addPrimaryKeyColumn("id", PrimaryKeyValue.INF_MAX);
parameter.setExclusiveEndPrimaryKey(endPkBuilder.build());
parameter.setMaxVersions(1);
// 每次底层请求拉取 2 行
parameter.setBufferSize(2);
Iterator<Row> iterator = client.createRangeIterator(parameter);
while (iterator.hasNext()) {
Row row = iterator.next();
System.out.println(row);
}