如何通过Spring Data Elasticsearch响应式模板检索聚合结果？

嘿，我之前也踩过这个坑！你现在用的reactiveElasticsearchTemplate.search()方法返回的Flux<SearchHit<T>>确实只包含匹配的文档数据，聚合结果并不在这个流里。要拿到聚合信息，你需要改用另一个重载方法，它会返回包含完整搜索上下文的Mono<SearchHits<T>>。

具体步骤： #

• 替换查询方法：把search()换成searchForPage()，这个方法会返回Mono<SearchHits<T>>，里面包含了聚合结果、分页信息以及命中的文档。
• 解析聚合结果：从SearchHits对象中通过getAggregations()获取聚合容器，再根据你定义的聚合名称提取对应的聚合数据。

修改后的代码示例： #

Query searchQuery = new NativeSearchQueryBuilder()
    .withQuery(queryBuilder)
    .withPageable(PageRequest.of(0, 10))
    .addAggregation(AggregationBuilders.terms("categories").field("category"))
    .build();

// 使用searchForPage()获取完整的搜索结果
Mono<SearchHits<T>> searchResultMono = reactiveElasticsearchTemplate.searchForPage(
    searchQuery, 
    documentType, 
    IndexCoordinates.of(indexName)
);

// 订阅并处理结果
searchResultMono.subscribe(searchHits -> {
    // 提取聚合结果
    Aggregations aggregations = searchHits.getAggregations();
    if (aggregations != null) {
        // 获取名为"categories"的terms聚合
        Terms categoriesAggregation = aggregations.get("categories");
        // 遍历每个桶，输出分类和对应的文档数量
        for (Terms.Bucket bucket : categoriesAggregation.getBuckets()) {
            String category = bucket.getKeyAsString();
            long docCount = bucket.getDocCount();
            System.out.printf("分类：%s，文档数量：%d%n", category, docCount);
        }
    }

    // 同时你也可以获取命中的文档流
    Flux<SearchHit<T>> documentFlux = Flux.fromIterable(searchHits.getSearchHits());
    // 处理文档数据...
});

为什么之前的方法拿不到聚合？ #

reactiveElasticsearchTemplate.search()的设计是将匹配的文档拆分成Flux<SearchHit<T>>流式返回，方便直接处理文档数据，但这种方式会丢弃聚合、分页元数据等额外信息。而searchForPage()会保留完整的搜索响应结构，所以能拿到你需要的聚合结果。

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