如何用ElasticSearch单查询实现按类别交替返回结果？

需要从ElasticSearch（ES）索引中查询30万条数据，数据分布差异明显：部分类别数据量超10万，部分仅1千条以内。要求返回结果按类别交替展示，例如每次返回每个类别2条结果（若该类别还有剩余数据），示例如下：

- cat 1 result
- cat 1 result
- cat 2 result
- cat 2 result
- cat 3 result # only one result in that category
- cat 1 result
- cat 1 result
- cat 2 result
- etc.

注：实际类别更多，部分类别当前查询可能无结果。

单类别逐个查询虽能实现该逻辑，但效率低且分页难以处理。目前仅找到通过multi-search查询后合并结果的方案，希望找到单查询实现方式；若ES不支持，想了解其他搜索引擎是否具备该能力。

ElasticSearch单查询实现思路 #

ES本身没有原生支持这种「按类别批量交替返回」的查询语法，但可以通过两种间接方式实现：

1. 用script_score结合分页逻辑模拟 #

• 核心逻辑：给每个文档计算排序分数，让不同类别的文档按「批次」交替排列
• 操作步骤：
  • 先通过terms聚合统计每个类别的总文档数
  • 对每个文档，计算其在所属类别内的「批次号」：floor(文档在类别内的偏移量 / 每批次返回条数)，比如每批次取2条，第1-2条批次号为0，第3-4条为1，以此类推
  • 用script_score设置排序规则：先按批次号升序，再按类别排序，最后叠加业务需求的排序规则
  • 分页时，需保证from/size参数覆盖所有类别的对应批次数据，同时过滤已返回的文档
• 局限性：依赖提前获取每个类别的文档数，分页逻辑复杂，类别多、数据差异大时维护成本高

2. 用collapse结合inner_hits批量获取后合并 #

• 核心逻辑：通过字段折叠按类别分组，批量获取每个类别指定数量的文档，再在应用层交替合并结果
• 操作步骤：
  • 查询时用collapse指定类别字段，通过inner_hits设置每个类别每次返回的条数（比如2条），同时将size设为类别总数（确保所有有数据的类别都被获取）
  • 拿到结果后，在应用层循环遍历每个类别的inner_hits，按批次交替取出文档，直到所有数据处理完毕
• 优势：单查询即可获取所有类别的批次数据，应用层合并逻辑简单；分页时只需维护每个类别已返回的偏移量，下次查询在inner_hits的from参数中设置对应值即可

其他搜索引擎支持情况 #

• Apache Solr：无原生交替返回语法，但可通过function query计算排序分数，或用Result Grouping（对应ES的collapse）获取分组数据后在应用层合并
• Meilisearch：支持group by分组查询，但仍需在应用层实现交替合并逻辑，无原生功能
• Typesense：提供group_by分组能力，获取每组指定数量文档后，在应用层完成交替排列

总结：目前主流搜索引擎都没有原生支持这种「按类别批量交替返回」的能力，最可行的方案是通过单查询获取分组数据后，在应用层实现交替合并逻辑——相比multi-search效率更高，分页也更容易处理。

内容的提问来源于stack exchange，提问作者François Constant