EasyEs复杂聚合使用案例

最新推荐文章于 2024-09-29 11:10:35 发布

xia_codings

最新推荐文章于 2024-09-29 11:10:35 发布

阅读量542

点赞数 3

文章标签：开发语言 EasyEs es java

本文链接：https://blog.csdn.net/xiasiyu123456/article/details/139339903

版权

EasyEs 聚合简单使用案例，这里面有几个注意点：
1.wrapper 条件必须前置，后获取聚合构造器
2.COUNT 聚合必须在某个聚合的子聚合
3.每个聚合结果映射均不一样，
4.分组的概念：就是父子聚合的概念，父就是分组

/**
* 构建聚合查询
*/
public CashOrderReport sumByDate(Long storeId, LocalDate startTime, LocalDate endTime) {
        if (endTime == null) {
            endTime = startTime;
        }
        LambdaEsQueryWrapper<CashOrder> wrapper = new LambdaEsQueryWrapper<>();
        //查询条件  必须在前
        wrapper.eq(CashOrder::getStoreId, storeId);
        wrapper.ge(CashOrder::getCreateTime, startTime);
        wrapper.le(CashOrder::getCreateTime, endTime);

        SearchRequest searchRequest = new SearchRequest();
        //指定索引
        searchRequest.indices("cash_order");
		//terms 查询条件
        TermQueryBuilder orderNumFilter = QueryBuilders.termQuery("orderStatus", "settled");
        //filter + COUNT 聚合
        AggregationBuilder orderNumAggr = AggregationBuilders.filter("orderNum", orderNumFilter)
                .subAggregation(AggregationBuilders.count("orderNum").field("orderStatus"));
		//获取构造器 注意：这里入参是 wrapper,（条件续前置，后置条件无效）
        SearchSourceBuilder builder = getEsMapper().getSearchSourceBuilder(wrapper);
        //SUM 聚合
        builder.aggregation(AggregationBuilders.sum("total").field("total"));
        //AVG 聚合
        builder.aggregation(AggregationBuilders.avg("aveTotal").field("total"));
        //COUNT 聚合
        builder.aggregation(orderNumAggr);
        searchRequest.source(builder);
		//聚合请求
        wrapper.setSearchSourceBuilder(builder);
        SearchResponse response = getEsMapper().search(wrapper);
        System.out.println("response = " + response);
        return convertResponse(response);
    }

/**
* 聚合查询结果转化
*/

    /**
     * 将返回结果转换为对象
     */
    private List<CashOrderReport> convertResponseToList(SearchResponse response) {
        List<CashOrderReport> orderReports = new ArrayList<>();
        Aggregations aggregations = response.getAggregations();
        aggregations.asList().forEach(value -> {
            ParsedDateHistogram histogram = (ParsedDateHistogram) value;
            histogram.getBuckets().forEach(bucket -> {
                orderReports.add(covertBucket(bucket));
            });
        });
        return orderReports;
    }

    private CashOrderReport covertBucket(Histogram.Bucket bucket) {
        CashOrderReport orderReport = new CashOrderReport();
        orderReport.setIncome(new BigDecimal("0"));
        orderReport.setTurnover(new BigDecimal("0"));
        orderReport.setAcCoupon(new BigDecimal("0"));
        orderReport.setRealPay(new BigDecimal("0"));
        orderReport.setValidOrderNum(0);
        orderReport.setAveOrderAmount(new BigDecimal("0"));
        orderReport.setReportDate(LocalDate.parse(bucket.getKeyAsString()));
        Map<String, Aggregation> aggregationMap = bucket.getAggregations().asMap();
        return covertReport(orderReport, aggregationMap);
    }

    CashOrderReport covertReport(CashOrderReport orderReport, Map<String, Aggregation> aggregationMap) {
        ParsedSum total = (ParsedSum) aggregationMap.get("total");
        ParsedSum totalPrice = (ParsedSum) aggregationMap.get("totalPrice");
        ParsedSum discountPrice = (ParsedSum) aggregationMap.get("discountPrice");
        ParsedAvg aveTotal = (ParsedAvg) aggregationMap.get("aveTotal");
        ParsedFilter orderNum = (ParsedFilter) aggregationMap.get("orderNum");
        orderReport.setRealPay(BigDecimal.valueOf(total.getValue()).setScale(2, RoundingMode.HALF_UP));
        orderReport.setTurnover(BigDecimal.valueOf(totalPrice.getValue()).setScale(2, RoundingMode.HALF_UP));
        orderReport.setAcCoupon(BigDecimal.valueOf(discountPrice.getValue()).setScale(2, RoundingMode.HALF_UP));
        orderReport.setIncome(orderReport.getTurnover().subtract(orderReport.getAcCoupon()));
        orderReport.setAveOrderAmount(BigDecimal.valueOf(aveTotal.getValue()).setScale(2, RoundingMode.HALF_UP));
        orderNum.getAggregations().asList().forEach(e -> {
            ParsedValueCount count = (ParsedValueCount) e;
            orderReport.setValidOrderNum((int) count.getValue());
        });
        return orderReport;
    }

    /**
     * 将返回结果转换为对象
     */
    private CashOrderReport convertResponse(SearchResponse response) {
        CashOrderReport orderReport = new CashOrderReport();
        orderReport.setIncome(new BigDecimal("0"));
        orderReport.setTurnover(new BigDecimal("0"));
        orderReport.setAcCoupon(new BigDecimal("0"));
        orderReport.setRealPay(new BigDecimal("0"));
        orderReport.setValidOrderNum(0);
        orderReport.setAveOrderAmount(new BigDecimal("0"));

        Map<String, Aggregation> aggregationMap = response.getAggregations().asMap();
        return covertReport(orderReport, aggregationMap);
    }