Examples with SignificanceHeuristic org.elasticsearch.search.aggregations.bucket.significant.heuristics.SignificanceHeuristic used on opensource projects

Example 6 with SignificanceHeuristic

use of org.elasticsearch.search.aggregations.bucket.significant.heuristics.SignificanceHeuristic in project elasticsearch by elastic.

the class InternalSignificantTerms method doReduce.

@Override
public InternalAggregation doReduce(List<InternalAggregation> aggregations, ReduceContext reduceContext) {
    long globalSubsetSize = 0;
    long globalSupersetSize = 0;
    // top-level Aggregations from each shard
    for (InternalAggregation aggregation : aggregations) {
        @SuppressWarnings("unchecked") InternalSignificantTerms<A, B> terms = (InternalSignificantTerms<A, B>) aggregation;
        globalSubsetSize += terms.getSubsetSize();
        globalSupersetSize += terms.getSupersetSize();
    }
    Map<String, List<B>> buckets = new HashMap<>();
    for (InternalAggregation aggregation : aggregations) {
        @SuppressWarnings("unchecked") InternalSignificantTerms<A, B> terms = (InternalSignificantTerms<A, B>) aggregation;
        for (B bucket : terms.getBucketsInternal()) {
            List<B> existingBuckets = buckets.get(bucket.getKeyAsString());
            if (existingBuckets == null) {
                existingBuckets = new ArrayList<>(aggregations.size());
                buckets.put(bucket.getKeyAsString(), existingBuckets);
            }
            // Adjust the buckets with the global stats representing the
            // total size of the pots from which the stats are drawn
            existingBuckets.add(bucket.newBucket(bucket.getSubsetDf(), globalSubsetSize, bucket.getSupersetDf(), globalSupersetSize, bucket.aggregations));
        }
    }
    SignificanceHeuristic heuristic = getSignificanceHeuristic().rewrite(reduceContext);
    final int size = reduceContext.isFinalReduce() == false ? buckets.size() : Math.min(requiredSize, buckets.size());
    BucketSignificancePriorityQueue<B> ordered = new BucketSignificancePriorityQueue<>(size);
    for (Map.Entry<String, List<B>> entry : buckets.entrySet()) {
        List<B> sameTermBuckets = entry.getValue();
        final B b = sameTermBuckets.get(0).reduce(sameTermBuckets, reduceContext);
        b.updateScore(heuristic);
        if (((b.score > 0) && (b.subsetDf >= minDocCount)) || reduceContext.isFinalReduce() == false) {
            ordered.insertWithOverflow(b);
        }
    }
    B[] list = createBucketsArray(ordered.size());
    for (int i = ordered.size() - 1; i >= 0; i--) {
        list[i] = ordered.pop();
    }
    return create(globalSubsetSize, globalSupersetSize, Arrays.asList(list));
}