Elasticsearch聚合查询

一、AggregationBuilder

RestHighLevelClient中分组需要用到AggregationBuilder作为分组条件的构建。我们可以看到AggregationBuilder接口有很多实现类,比如TermsAggregationBuilder、AvgAggregationBuilder、MaxAggregationBuilder、SumAggregationBuilder等常用的统计函数。

二、样例代码

    SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();

    // 以batchId为分组条件,terms为分组后的字段名称,field为将被分组的字段名称
    TermsAggregationBuilder aggregation = AggregationBuilders.terms("batchId").field("batchId.keyword")
            // 分组求和integral字段,并将求和后的字段名改为score
            // subAggregation为子聚合,即在batchId分组后的小组内聚合
            .subAggregation(AggregationBuilders.sum("score").field("integral"))
            .subAggregation(AggregationBuilders.topHits("details").size(1));

    sourceBuilder.aggregation(aggregation);

    BoolQueryBuilder boolBuilder = QueryBuilders.boolQuery();
    sourceBuilder.query(boolBuilder);

    SearchRequest searchRequest = new SearchRequest("table");
    searchRequest.source(sourceBuilder);

    SearchResponse search = client.search(searchRequest);

    // 和之前不同的是这里需要getAggregations获取聚合后的数据
    Aggregations aggregations = search.getAggregations();

    // 从分组后的数据中拿到batchId的数据,这里以batchId分组,则分组后的数据都在batchId里
    ParsedStringTerms terms = aggregations.get("batchId");

    // 获取到分组后的所有bucket
    List<? extends Terms.Bucket> buckets = terms.getBuckets();

    for (Terms.Bucket bucket : buckets) {
        // 解析bucket 因为一级聚合为以batchId分组,二级聚合为求和,所以这里还需要getAggregations获取求和的数据
        Aggregations bucketAggregations = bucket.getAggregations();

        // 这里我是通过debug才找到返回的参数类型的,我不知道在哪找得到这个东西,所以我们拿到了ParsedTopHits,这里我们是取了一个,所以这个值的数组长度为1
        ParsedTopHits topHits = bucketAggregations.get("details");

        // 因为求和和下面的topHits都是AggregationBuilders.terms("batchId").field("batchId.keyword")的subAggreation,所以都属于batchId组内
        // 获取到求和的数据信息
        ParsedSum sum = bucketAggregations.get("integral");

        // 因为topHits中命中的hits肯定至少有一个,要不然也不会成组,所以这里直接获取第一个,并解析成map
        Map<String, Object> sourceAsMap = topHits.getHits().getHits()[0].getSourceAsMap();

        // 将求和后的integral覆盖到原数据中
        sourceAsMap.put("integral", sum.value());

        // 打印出统计后的数据
        System.out.println(sourceAsMap);
    }

三、说明

3.1 subAggregation

为什么要用subAggregation(AggregationBuilders.topHits("details").size(1))

这里有个坑,如果只用batchId分组并用intrgal求和,那么返回值中只会返回batchId与integral求和后的值这两个字段。这里举个例子,现在统计全国所有学校的总人数,那么我们以school分组,求和了人数。但是我们仍然想知道每个学校属于哪个地区的,这里聚合后就不会返回这些信息,让人很难受。

此时我们就可以使用subAggregation(AggregationBuilders.topHits("details").size(1))来获取一个通用的数据,拿到学校所属的地区名称或者其他学校的上级属性。

3.2 AggregationBuilders层级

我们注意到subAggregation方法是AggregationBuilder的,所以在构建AggregationBuilders的时候,每一个子AggregationBuilder都可以有子AggregationBuilder,如下

TermsAggregationBuilder aggregation = AggregationBuilders.terms("batchId").field("batchId.keyword")
  .subAggregation(
      AggregationBuilders.terms("folder").field("folder")
          .subAggregation(AggregationBuilders.sum("integral").field("integral"))
);

这种就是组内再聚合套组内,但是得注意字段的问题,即若你以batchId分组后,那么组内只有batchId与其他的字段,就和第一个注意点一样,导致组内聚合拿不到相关的字段,导致报错。

资料来源:
RestHighLevelClient进阶-聚合操作

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350

推荐阅读更多精彩内容