使用 Tablestore Python SDK 可按指定多元索引字段折叠查询结果,使每个不同字段值只返回一行数据。
前提条件
安装Tablestore Python SDK并初始化客户端。
功能说明
折叠按照指定多元索引字段的值对匹配行分组,每个不同字段值只返回一行代表数据。分组内返回哪一行由查询排序方式决定,可通过排序和翻页设置。折叠只改变查询结果,不修改数据表。
折叠字段必须启用排序与统计聚合,只支持 Keyword、Long 和 Double 类型的非数组字段。折叠查询只能使用 limit 和 offset 翻页,不能使用 next_token;limit + offset 不能超过 100000。统计聚合、分组和匹配总行数均基于折叠前的结果,无法通过响应获取折叠后的总分组数。
以下示例按 category 折叠结果,并按 price 降序排序,因此每个类别返回价格最高的一行。
query = MatchAllQuery()
search_query = SearchQuery(
query,
sort=Sort([FieldSort("price", SortOrder.DESC)]),
collapse_field=Collapse("category"),
limit=10,
get_total_count=True,
)
response = client.search(
"example_table",
"example_index",
search_query,
ColumnsToGet(
["category", "price"],
ColumnReturnType.SPECIFIED,
),
)
print(response.total_count)
print(response.rows)
参数说明
查询请求
search 方法包含以下参数。
|
名称 |
类型 |
说明 |
|
table_name(必选) |
|
数据表名称。 |
|
index_name(必选) |
|
多元索引名称。 |
|
search_query(必选) |
|
查询条件和折叠配置。 |
|
columns_to_get(可选) |
|
返回列配置。未设置时只返回主键列。 |
|
routing_keys(可选) |
|
自定义路由字段对应的主键值列表。未配置自定义路由时无需设置。 |
|
timeout_s(可选) |
|
请求级超时时间,单位为秒。未设置时使用客户端默认超时时间。 |
查询配置
search_query 的类型为 SearchQuery,包含以下参数。
|
名称 |
类型 |
说明 |
|
query(必选) |
|
查询条件。根据数据匹配需求设置具体 Query 类型。 |
|
sort(可选) |
|
返回结果的排序方式。有关配置方法,请参见排序和翻页。 |
|
get_total_count(可选) |
|
是否返回匹配总行数。默认值为 |
|
next_token(可选) |
|
使用折叠功能时不能设置。 |
|
offset(可选) |
|
起始分组位置,默认值为 |
|
limit(可选) |
|
本次查询返回的最大行数。设置为 |
|
aggs(可选) |
|
指标聚合配置。有关配置方法,请参见统计聚合。 |
|
group_bys(可选) |
|
分组配置。有关配置方法,请参见统计聚合。 |
|
collapse_field(必选) |
|
结果折叠配置。 |
|
highlight(可选) |
|
|
折叠配置
search_query.collapse_field 的类型为 Collapse,包含以下参数。
|
名称 |
类型 |
说明 |
|
field_name(必选) |
|
折叠字段名称。字段必须启用排序与统计聚合,且为 |
返回列
columns_to_get 的类型为 ColumnsToGet,包含以下参数。
|
名称 |
类型 |
说明 |
|
column_names(可选) |
|
要返回的属性列名称。仅 |
|
return_type(可选) |
|
返回列模式。 |
返回值
search 方法返回 SearchResponse。核心字段如下。
|
字段 |
类型 |
说明 |
|
rows |
|
折叠后的行数据,每个不同折叠字段值最多返回一行。 |
|
next_token |
|
下一页凭证。值为空时表示没有更多数据。 |
|
total_count |
|
折叠前的匹配行数,取决于 |
|
is_all_succeed |
|
是否已成功查询全部索引分区。值为 |
|
agg_results |
|
指标聚合结果。未配置 |
|
group_by_results |
|
分组结果。未配置 |
|
search_hits |
|
查询命中结果,包含行数据、相关性得分和高亮结果等扩展信息。 |
兼容 Tuple 返回格式
Tablestore Python SDK 5.2.0 开始将查询接口的返回值由 Tuple 调整为响应对象,5.1.0 及以下版本直接返回 Tuple。5.2.1 及以上版本可调用 SearchResponse.v1_response() 获取与旧版本兼容的 Tuple。新代码建议直接访问 SearchResponse 的属性,避免返回字段扩展后解包数量不匹配。
(
rows,
next_token,
total_count,
is_all_succeed,
agg_results,
group_by_results,
search_hits,
) = response.v1_response()