elasticsearch对聚合后的结果进行范围筛选

小半 • 2023年1月20日上午10:03 • 后端开发 • 阅读 342

这一篇写对聚合后的范围筛选。

一般的我们可以在query或filter里使用range进行范围筛选查询，但是我们在聚合里怎么对聚合后的结果筛选出符合范围的聚合结果呢？

dsl语句如下：

  "aggregations": {
    "statistics_assets": {
      "terms": {
        "field": "one_account.one_account_no",
        "size": 10
      },
      "aggregations": {
        "assets": {
          "sum": {
            "field": "assets.merge"
          }
        },
        "assets_sum_range": {
          "bucket_selector": {
            "buckets_path": {
              "key": "assets"
            },
            "script": {
              "source": "0<=params.key&&params.key<=100000",
              "lang": "painless"
            }
          }
        }
      }
    }
  }

assets_sum_range中使用bucket_selector对sum聚合后的”assets.merge”结果进行筛选，筛选的条件范围是0<=聚合结<=100000

Java代码：

SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();

TermsAggregationBuilder termsAggregationBuilder = AggregationBuilders.terms("statistics_assets")
                   .field("one_account.one_account_no")
                   .size(10);       
AggregationBuilder assetsAggregation = AggregationBuilders.sum("assets").field("assets.merge");
termsAggregationBuilder.subAggregation(assetsAggregation);

 //对聚合后的总资产进行范围选择
   Map<String, String> bucketsPathsMap = new HashMap<>();
   bucketsPathsMap.put("key", "assets");
   Script script = new Script( "0<=params.key&&params.key<=100000" );
   BucketSelectorPipelineAggregationBuilder bs = PipelineAggregatorBuilders.bucketSelector("assets_sum_range", bucketsPathsMap, script);
   termsAggregationBuilder.subAggregation(bs);

searchSourceBuilder.aggregation(termsAggregationBuilder);

最后输出的结果便是0-100000范围的聚合结果：

  "aggregations" : {
    "statistics_assets" : {
      "doc_count_error_upper_bound" : 0,
      "sum_other_doc_count" : 4,
      "buckets" : [
        {
          "key" : "110600000973",
          "doc_count" : 2,
          "assets" : {
            "value" : 0.0
          }
        },
        {
          "key" : "105100015207",
          "doc_count" : 1,
          "assets" : {
            "value" : 9.999
          }
        }
      ]
    }
  }