All Downloads are FREE. Search and download functionalities are using the official Maven repository.

querqy.elasticsearch.aggregation.InternalDecorationAggregation Maven / Gradle / Ivy

Go to download 

package querqy.elasticsearch.aggregation;

import org.elasticsearch.Version;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
import org.elasticsearch.search.aggregations.InternalAggregation;
import org.elasticsearch.search.aggregations.metrics.ScriptedMetric;
import org.elasticsearch.xcontent.XContentBuilder;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import static java.util.Collections.singletonList;

public class InternalDecorationAggregation extends InternalAggregation implements ScriptedMetric {

    private final List aggregations;

    InternalDecorationAggregation(final String name, final List aggregations, final Map metadata) {
        super(name, metadata);
        this.aggregations = aggregations;
    }

    public InternalDecorationAggregation(final StreamInput in) throws IOException {
        super(in);
        aggregations = in.readList(StreamInput::readGenericValue);
    }

    @Override
    protected void doWriteTo(final StreamOutput out) throws IOException {
        out.writeCollection(aggregations, StreamOutput::writeGenericValue);
    }

    @Override
    public String getWriteableName() {
        return QuerqyDecorationAggregationBuilder.NAME;
    }

    @Override
    public Object aggregation() {
        if (aggregations.size() != 1) {
            throw new IllegalStateException("aggregation was not reduced");
        }
        return aggregations.get(0);
    }

    List aggregationsList() {
        return aggregations;
    }

    @Override
    public InternalAggregation reduce(final List aggregations, final ReduceContext reduceContext) {
        final List aggregationObjects = new ArrayList<>();
        for (final InternalAggregation aggregation : aggregations) {
            final InternalDecorationAggregation mapReduceAggregation = (InternalDecorationAggregation) aggregation;
            aggregationObjects.addAll(mapReduceAggregation.aggregations);
        }
        final InternalDecorationAggregation firstAggregation = ((InternalDecorationAggregation) aggregations.get(0));
        final List aggregation;
        if (reduceContext.isFinalReduce()) {
            aggregation = Collections.singletonList(aggregationObjects);
        } else {
            // if we are not an final reduce we have to maintain all the aggs from all the incoming one
            // until we hit the final reduce phase.
            aggregation = aggregationObjects;
        }
        return new InternalDecorationAggregation(firstAggregation.getName(), aggregation, getMetadata());
    }

    @Override
    protected boolean mustReduceOnSingleInternalAgg() {
        return true;
    }

    @Override
    public Object getProperty(final List path) {
        if (path.isEmpty()) {
            return this;
        } else if (path.size() == 1 && "value".equals(path.get(0))) {
            return aggregation();
        } else {
            throw new IllegalArgumentException("path not supported for [" + getName() + "]: " + path);
        }
    }

    @Override
    public XContentBuilder doXContentBody(final XContentBuilder builder, final Params params) throws IOException {
        return builder.field(CommonFields.VALUE.getPreferredName(), aggregation());
    }

    @Override
    public boolean equals(final Object obj) {
        if (this == obj) return true;
        if (obj == null || getClass() != obj.getClass()) return false;
        if (!super.equals(obj)) return false;

        final InternalDecorationAggregation other = (InternalDecorationAggregation) obj;
        return Objects.equals(aggregations, other.aggregations);
    }

    @Override
    public int hashCode() {
        return Objects.hash(super.hashCode(), aggregations);
    }

}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy