io.polyglotted.eswrapper.query.response.Aggregation Maven / Gradle / Ivy

Go to download
package io.polyglotted.eswrapper.query.response;

import com.google.common.collect.ImmutableMap;
import io.polyglotted.eswrapper.query.AggregationType;
import lombok.*;
import lombok.experimental.Accessors;

import java.util.AbstractMap.SimpleEntry;
import java.util.*;
import java.util.Map.Entry;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
import static com.google.common.collect.Iterables.transform;

@RequiredArgsConstructor(access = AccessLevel.PRIVATE)
@ToString(includeFieldNames = false, doNotUseGetters = true, of = {"label", "type", "value"})
public final class Aggregation {
    public final String label;
    public final String type;
    public final Object value;
    public final ImmutableMap params;

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Aggregation that = (Aggregation) o;
        return label.equals(that.label) && type.equals(that.type) &&
           value.equals(that.value) && params.equals(that.params);
    }

    @Override
    public int hashCode() {
        return Objects.hash(label, type, value, params);
    }

    public boolean hasBuckets() {
        return AggregationType.valueOf(type).hasBuckets;
    }

    @SuppressWarnings("unchecked")
    public List buckets() {
        checkState(hasBuckets(), type + " does not support buckets");
        return (List) value;
    }

    public  T param(String name, Class tClass) {
        return tClass.cast(params.get(name));
    }

    public long longValue(String name) {
        return value(name, Long.class);
    }

    public double doubleValue(String name) {
        return value(name, Double.class);
    }

    @SuppressWarnings("unchecked")
    public  T value(String name, Class tClass) {
        return (value instanceof Map) ? tClass.cast(((Map) value).get(name)) : tClass.cast(value);
    }

    @SuppressWarnings("unchecked")
    public Iterable> valueIterable() {
        return (value instanceof Map) ? ((Map) value).entrySet()
           : Collections.singletonList(new SimpleEntry("value", value));
    }

    public static Builder aggregationBuilder() {
        return new Builder();
    }

    @Setter
    @Accessors(fluent = true, chain = true)
    @NoArgsConstructor(access = AccessLevel.PRIVATE)
    public static class Builder {
        private String label;
        private AggregationType type;
        private final java.util.Map valueMap = new TreeMap<>();
        private final java.util.Map paramsMap = new TreeMap<>();
        private final java.util.List builders = new ArrayList<>();

        public Builder value(String key, Object value) {
            valueMap.put(key, value);
            return this;
        }

        public Builder param(String key, Object value) {
            paramsMap.put(key, value);
            return this;
        }

        public Builder bucket(Bucket.Builder builder) {
            this.builders.add(builder);
            return this;
        }

        public Bucket.Builder bucketBuilder() {
            Bucket.Builder builder = Bucket.bucketBuilder();
            this.builders.add(builder);
            return builder;
        }

        public Aggregation build() {
            Iterable buckets = transform(builders, Bucket.Builder::build);
            return new Aggregation(checkNotNull(label, "label cannot be null"), checkNotNull(type,
               "type cannot be null").name(), type.valueFrom(valueMap, buckets), ImmutableMap.copyOf(paramsMap));
        }
    }
}