当前位置：首页 > news >正文

深入浅出Java RestClient与Elasticsearch：一次简单的聚合查询实现

news 来源：原创 2025/5/12 19:20:53

文章目录

- - - 一、什么是聚合查询？
    - 二、开始编写代码
    - 三、代码解析
    - 四、总结

在当今的大数据时代，Elasticsearch 作为一个分布式搜索和分析引擎，被广泛应用于各类数据密集型应用中。而使用Java与Elasticsearch 交互进行数据操作时，Java High Level RestClient 提供了一种方便且高效的方式。今天，我们将通过一个简单的例子，向大家展示如何使用 Java RestClient 在 Elasticsearch 中进行一次聚合查询。

一、什么是聚合查询？

在开始之前，先简单介绍一下聚合查询的概念。聚合（Aggregation）是Elasticsearch中非常强大的功能之一，它允许我们对文档进行复杂的数据统计和分析。比如，我们可以使用聚合来统计某个字段的值的分布情况，类似SQL中的GROUP BY操作。

二、开始编写代码

下面，我们将通过一个具体的代码示例来演示如何使用 Java RestClient 实现一个聚合查询。代码如下：

@Test
void testAgg() throws IOException {
    // 1.创建request对象
    SearchRequest request = new SearchRequest("items");
    
    // 2.组织DSL参数
    request.source().size(0); // 不返回实际文档，只返回聚合结果
    
    // 2.2.聚合条件
    String brandAggName = "brandAgg";
    request.source().aggregation(
            AggregationBuilders.terms(brandAggName).field("brand").size(10)
    );
    
    // 3.发送请求
    SearchResponse response = client.search(request, RequestOptions.DEFAULT);
    
    // 4.解析响应
    Aggregations aggregations = response.getAggregations();
    
    // 4.1.根据聚合条件名称获取聚合结果
    Terms brandTerms = aggregations.get(brandAggName);
    
    // 4.2.获取buckets
    List<? extends Terms.Bucket> buckets = brandTerms.getBuckets();
    
    // 4.3.遍历获取每一个bucket
    for (Terms.Bucket bucket : buckets) {
        System.out.println("brand=" + bucket.getKeyAsString() + ",count=" + bucket.getDocCount());
    }
}

三、代码解析

创建SearchRequest对象
我们首先创建一个SearchRequest对象，并指定要查询的索引名称为“items”。这个索引名称可以根据实际情况进行调整。
组织DSL查询
在这一步中，我们使用request.source().size(0)方法设置不返回实际文档，仅返回聚合结果。接着，我们定义了一个聚合条件，即对“brand”字段进行聚合，并指定聚合结果中最多返回10个不同的品牌。
发送请求并获取响应
使用client.search(request, RequestOptions.DEFAULT)方法发送请求，并获取查询响应。
解析聚合结果
在收到响应后，我们通过response.getAggregations()方法获取聚合结果。然后，根据我们之前定义的聚合条件名称“brandAgg”获取对应的聚合结果。最后，我们遍历每一个bucket，打印出品牌名称和文档数量。