从es聚合统计中获取统计结果封装到LinkedHashMap

最新推荐文章于 2022-10-27 17:35:23 发布

量天尺VS流彩虹

最新推荐文章于 2022-10-27 17:35:23 发布

阅读量414

点赞数

文章标签： elasticsearch 数据库

本文链接：https://blog.csdn.net/weixin_38861291/article/details/117367262

版权

从es聚合统计中获取统计结果封装到LinkedHashMap

/**
     * 根据聚合名 将es查询返回聚合统计解析为map
     * 统计结果在桶内，则键为桶内key
     * 统计结果不在桶内，则键为 统计名
     * @param aggsMap 聚合对象map
     * @param name 名称
     * @param isBucket 是否桶内
     * @param key 桶内键
     * lzm
     * 2021/04/22
     * @return HashMap<String,String>
     */
    public  static LinkedHashMap<String,String> getValFromAggs( Map<String,Aggregation> aggsMap,String name,boolean isBucket,String key){
        LinkedHashMap<String,String> resultMap =new LinkedHashMap<>();
        if(aggsMap != null){
            Iterator<Map.Entry<String,Aggregation>> it=aggsMap.entrySet().iterator();
            while(it.hasNext()){
                Map.Entry<String,Aggregation> entry=it.next();
                Aggregation aggregation = entry.getValue();
                try{
                    if(aggregation instanceof  ParsedNested  || aggregation instanceof   ParsedFilter || aggregation instanceof ParsedReverseNested) {
                        Class catClass = aggregation.getClass();
                        String aggName = catClass.getMethod("getName").invoke(aggregation, null).toString(); // 调用指定方法
                        if(aggName.equals(name)){
                            String value = String.valueOf(catClass.getMethod("getDocCount").invoke(aggregation, null));
                            resultMap =  new LinkedHashMap<>();
                            if(isBucket){
                                resultMap.put(key,value);
                            }else{
                                resultMap.put(name,value);
                            }
                            return resultMap;
                        }else{
                            Aggregations aggregations = (Aggregations)catClass.getMethod("getAggregations").invoke(aggregation, null);
                            resultMap = getValFromAggs(aggregations.asMap(), name,false,null);
                            if(resultMap != null && !resultMap.isEmpty()){
                                return resultMap;
                            }
                        }
                    }else if(aggregation instanceof ParsedStringTerms || aggregation instanceof ParsedLongTerms){
                        Class catClass = aggregation.getClass();
                        String aggName = catClass.getMethod("getName").invoke(aggregation, null).toString(); // 调用指定方法
                        if(aggName.equals(name)){
                           //return aggs.getBuckets();
                        }else{
                            LinkedHashMap<String,String> tempMap = new LinkedHashMap<>();
                            List<? extends Bucket> buckets = ( List<? extends Bucket>)catClass.getMethod("getBuckets").invoke(aggregation, null);
                            for (Terms.Bucket bucket : buckets) {
                                resultMap =getValFromAggs(bucket.getAggregations().asMap(), name,true,bucket.getKeyAsString());
                                if(resultMap != null && !resultMap.isEmpty()){
                                    tempMap.putAll(resultMap);
                                }
                            }
                            if(tempMap != null && !tempMap.isEmpty()){
                                return tempMap;
                            }
                        }
                    }else {
                        logger.info(aggregation.getClass());
                    }
                }catch (Exception e){
                    e.printStackTrace();
                    logger.error(ExceptionUtil.getStackTraceString(e));
                }
            }
        }
        return null;
    }