java 通过聚合查询实现elasticsearch的group by后的数量

通过聚合查询获取group by 后的数量

/**
     * 获取key的个数
     *
     * @param key   要group by的字段名
     * @param index 索引名称
     * @return id的个数
     */
    public static int getKeyCount(String key, String index) {
        int count = 0;
        TransportClient client = null;
        try {
            client = connectionPool.getConnection();
            if (client == null) {
                throw new Exception("没有获取到连接!");
            }
            SearchRequestBuilder search = client.prepareSearch(index);
            //cardinality聚合查询,相当于groupby字段名
            SearchResponse sr = search.addAggregation(AggregationBuilders.cardinality(key + "_count").field(key)).execute().actionGet();
            //从返回数据提取id总数
            Cardinality result = sr.getAggregations().get(key + "_count");
            long value = result.getValue();
            count = (int) value;
        } catch (InterruptedException e) {
        } catch (Exception e) {
            logger.error("getKeyCount错误", e);
        } finally {
            connectionPool.releaseConnection(client);
        }
        return count;
    }

获取group by后的所有key值

/**
     * 获取所有key
     *
     * @param key   被group by的字段名
     * @param index 索引名称
     * @return 所有id
     */
    public static List getAllKey(String key, String index) {
        int keyCount = getKeyCount(key, index);
        List strings = new ArrayList<>();
        TransportClient client = null;
        try {
            client = connectionPool.getConnection();
            if (client == null) {
                throw new Exception("没有获取到数据库连接!");
            }
            SearchRequestBuilder searchRequestBuilder = client.prepareSearch(index);
            //使用聚合,实现去重查询
            SearchResponse searchResponse = searchRequestBuilder.
                    addAggregation(AggregationBuilders.terms("models").field(key).size(keyCount)).execute().actionGet();
            Terms term = searchResponse.getAggregations().get("models");
            List buckets = term.getBuckets();
            //遍历结果,提取出id
            for (Terms.Bucket bucket : buckets) {
                String keyAsString = bucket.getKeyAsString();
                strings.add(keyAsString);
            }
            buckets.clear();
        } catch (InterruptedException e) {

        } catch (Exception e) {
            logger.error("getAllKey错误", e);
        } finally {
            connectionPool.releaseConnection(client);
        }

        return strings;
    }

到此这篇关于Java通过聚合查询获取group by 后的数量的文章就介绍到这了,更多相关Java 聚合查询获取group by数量内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

你可能感兴趣的