elasticsearch(二)：java操作es实现分组聚合查询和基本查询

港控/mmm° 2021-12-10 15:43 3785阅读 1赞

es查询还有一个很常用的一种叫聚合查询，相当于mysql中的分组group by 后拿各组数量进行统计 ，实现起来也是很简单。

# 基本查询包含聚合： #

具体直接看代码，代码中部分有注释

/**
         * person查询
         * 继承baseService的list返回的query对象  
         * @return
         */
       public Result personList(PersonInfo personInfo) {
            Result result = new Result();
    		//如果参数有值，放这个map中
            Map<String, Object> queryParams = new HashMap<>();
    		//接收参数，如果没有则按没条件查询，下方有判空
            String query_EQ_personName = personInfo.getPersonName();
            String query_EQ_workOrgId = personInfo.getWorkOrgId()+"";
            String query_EQ_filed = personInfo.getFiled();
            if( StringUtils.isNotEmpty(query_EQ_personName) ){
                queryParams.put("EQ_person_name", query_EQ_personName);
            }
    
            if( StringUtils.isNotEmpty(query_EQ_workOrgId) && !"null".equals(query_EQ_workOrgId)){
                queryParams.put("EQ_work_org_id", Integer.parseInt(query_EQ_workOrgId));
            }
    
            if( StringUtils.isNotEmpty(query_EQ_filed) ){
                queryParams.put("EQ_filed", query_EQ_filed);
            }
            QueryBuilder queryBuilder = createQueryBuilder(create(queryParams));
    		//左侧聚合 条件：人员类型，职称
            SearchQuery searchQuery = creatSearchQuery(queryBuilder,
    				//指定要聚合的字段
    				AggregationBuilders.terms("military_rank").field("military_rank.keyword"),//种类
                    AggregationBuilders.terms("personType").field("personType.keyword"),//人员类型
                    AggregationBuilders.terms("title").field("title.keyword"));//职称
    
    		//查询操作,这里用的另外一种查询，不包括分页效果
            AggregatedPage<PersonInfoEs> page = (AggregatedPage<PersonInfoEs>) personRepository.search(searchQuery);
    		//创建map用来放聚合的数据
            Map<String,Object> title = new HashMap<>();
            Map<String,Object> personType = new HashMap<>();
    		//从结果中根据名称来拿聚合出来的数据
            Terms termTitle = (Terms) page.getAggregations().getAsMap().get("title");
            Terms termPersonType = (Terms) page.getAggregations().getAsMap().get("personType");
    		Terms termMilitary_rank = (Terms) page.getAggregations().getAsMap().get("military_rank");
    		//遍历方式拿数据，放入map中，map为数据内容，value为数量
            for (Terms.Bucket bucket : termPersonType.getBuckets()) {
                personType.put(bucket.getKey().toString().isEmpty() ? "其他" : bucket.getKey().toString(), bucket.getDocCount());
            }
            for (Terms.Bucket bucket : termTitle.getBuckets()) {
                title.put(bucket.getKey().toString().isEmpty() ? "其他" : bucket.getKey().toString(), bucket.getDocCount());
            }
    		 for (Terms.Bucket bucket : termMilitary_rank.getBuckets()) {
                military_rank.put(bucket.getKey().toString().isEmpty() || null== bucket.getKey() ? "其他" : bucket.getKey().toString(), bucket.getDocCount());
            }
           
    		//最后在放入map中一并返回
            Map<String, Object> totalMap = new LinkedHashMap<>();
            totalMap.put("title", title);
            totalMap.put("personType",personType );
    		totalMap.put("military_rank", military_rank);
    		
            result.setCode(200);
            Map<String, Object> contentMap = new HashMap<>();
            contentMap.put("content", page.getContent());
            result.setData(contentMap);
            result.setCount(String.valueOf(page.getTotalElements()));
            result.setPageSize(page.getSize());
            result.setPageNumber(page.getNumber());
            HashMap<String, List<GroupInfo>> groupFilterMap = builAggregationVo(totalMap);
    
            result.setGroupFilter(groupFilterMap);
            return result;
        }

返回结果如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1hpYW9kb25nZ2U1MjE_size_16_color_FFFFFF_t_70][]

如有问题请留言探讨；

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1hpYW9kb25nZ2U1MjE_size_16_color_FFFFFF_t_70]: /images/20211210/15f41ef2573c4f5ea411d92cb7b1aa83.png