前段时间我偶然发现了同样的问题,不幸的是,java 客户端不能很好地支持聚合响应(如果我错了,请纠正我)。
以下是我经过多轮跟踪错误方法后得出的结论,如果这有帮助,请告诉我。
private Map<String, List<Map.Entry<String, Long>>> buildAggregationsFromResult(SearchResults resultSet, Set aggregationsName) {
Map<String, List<Map.Entry<String, Long>>> aggregations = new LinkedHashMap<String, List<Map.Entry<String, Long>>>();
if (null != resultSet.getAggregations()) {
Map<String, Aggregation> aggregationMap = resultSet.getAggregations().asMap();
for (Map.Entry<String, Aggregation> aggregationEntry : aggregationMap.entrySet()) {
fetchBucketTermsFromAggregations(aggregationEntry,aggregations, aggregationsName);
}
}
return aggregations;
}
private List<Map.Entry<String, Long>> fetchBucketTermsFromAggregations(Map.Entry<String, Aggregation> aggregationEntry, Map<String, List<Map.Entry<String, Long>>> aggregations, Set<String> aggregationsName) {
Terms terms = null;
List<Map.Entry<String, Long>> pairs = new ArrayList<>();
if (aggregationEntry.getValue() instanceof InternalNested) {
for (Map.Entry<String, Aggregation> aggregationEntryNested : ((InternalNested) aggregationEntry.getValue()).getAggregations().asMap().entrySet()) {
pairs = fetchBucketTermsFromAggregations(aggregationEntryNested, aggregations, aggregationsName);
if (CollectionUtils.isNotEmpty(pairs) && aggregationsName.contains(aggregationEntry.getKey()))
aggregations.put(aggregationEntry.getKey(), pairs);
}
} else if (aggregationEntry.getValue() instanceof InternalFilters) {
for (InternalFilters.Bucket buckets : ((InternalFilters) aggregationEntry.getValue()).getBuckets()) {
for (Map.Entry<String, Aggregation> nestedAggregations : buckets.getAggregations().asMap().entrySet()) {
pairs = fetchBucketTermsFromAggregations(nestedAggregations, aggregations, aggregationsName);
if (CollectionUtils.isNotEmpty(pairs) && aggregationsName.contains(aggregationEntry.getKey()) )
aggregations.put(aggregationEntry.getKey(), pairs);
}
}
} else {
terms = (Terms) aggregationEntry.getValue();
if (null != terms ) {
Collection<Terms.Bucket> buckets = terms.getBuckets();
for (Terms.Bucket bucket : buckets) {
if (CollectionUtils.isNotEmpty(bucket.getAggregations().asList()))
pairs.add(new AbstractMap.SimpleEntry<String, Long>(bucket.getKeyAsString(), ((InternalReverseNested)bucket.getAggregations().asList().get(0)).getDocCount()));
else
pairs.add(new AbstractMap.SimpleEntry<String, Long>(bucket.getKeyAsString(), bucket.getDocCount()));
}
}
if (CollectionUtils.isNotEmpty(pairs) && aggregationsName.contains(aggregationEntry.getKey()))
aggregations.put(aggregationEntry.getKey(), pairs);
}
return pairs;
}
我知道这看起来有点笨拙,但可能适合您的情况。否则请告诉我。