io.github.stewseo.client.yelpfusion.transform.PivotGroupBy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yelp-fusion-client Show documentation
java client to build api objects, handle http transport, and parse/deserialize/serialize json to/from json
There is a newer version: 1.8.0
package io.github.stewseo.client.yelpfusion.transform;

import io.github.stewseo.client._types.aggregations.DateHistogramAggregation;
import io.github.stewseo.client._types.aggregations.HistogramAggregation;
import io.github.stewseo.client._types.aggregations.TermsAggregation;
import io.github.stewseo.client.json.JsonEnum;
import io.github.stewseo.client.json.JsonpDeserializable;
import io.github.stewseo.client.json.JsonpDeserializer;
import io.github.stewseo.client.json.JsonpMapper;
import io.github.stewseo.client.json.JsonpSerializable;
import io.github.stewseo.client.json.JsonpUtils;
import io.github.stewseo.client.json.ObjectBuilderDeserializer;
import io.github.stewseo.client.json.ObjectDeserializer;
import io.github.stewseo.client.util.ApiTypeHelper;
import io.github.stewseo.client.util.ObjectBuilder;
import io.github.stewseo.client.util.TaggedUnion;
import io.github.stewseo.client.util.TaggedUnionUtils;
import io.github.stewseo.client.util.WithJsonObjectBuilderBase;
import jakarta.json.stream.JsonGenerator;

import java.util.function.Function;

@JsonpDeserializable
public class PivotGroupBy implements TaggedUnion, JsonpSerializable {

    /**
     * {@link PivotGroupBy} variant kinds.
     *
     * @see API
     *      specification
     */

    public enum Kind implements JsonEnum {
        DateHistogram("date_histogram"),

        GeotileGrid("geotile_grid"),

        Histogram("histogram"),

        Terms("terms"),

        ;

        private final String jsonValue;

        Kind(String jsonValue) {
            this.jsonValue = jsonValue;
        }

        public String jsonValue() {
            return this.jsonValue;
        }

    }

    private final Kind _kind;
    private final Object _value;

    @Override
    public final Kind _kind() {
        return _kind;
    }

    @Override
    public final Object _get() {
        return _value;
    }

    public PivotGroupBy(PivotGroupByVariant value) {

        this._kind = ApiTypeHelper.requireNonNull(value._pivotGroupByKind(), this, "");
        this._value = ApiTypeHelper.requireNonNull(value, this, "");

    }

    private PivotGroupBy(Builder builder) {

        this._kind = ApiTypeHelper.requireNonNull(builder._kind, builder, "");
        this._value = ApiTypeHelper.requireNonNull(builder._value, builder, "");

    }

    public static PivotGroupBy of(Function> fn) {
        return fn.apply(new Builder()).build();
    }

    /**
     * Is this variant instance of kind {@code date_histogram}?
     */
    public boolean isDateHistogram() {
        return _kind == Kind.DateHistogram;
    }

    /**
     * Get the {@code date_histogram} variant value.
     *
     * @throws IllegalStateException
     *             if the current variant is not of the {@code date_histogram} kind.
     */
    public DateHistogramAggregation dateHistogram() {
        return TaggedUnionUtils.get(this, Kind.DateHistogram);
    }

    /**
     * Is this variant instance of kind {@code histogram}?
     */
    public boolean isHistogram() {
        return _kind == Kind.Histogram;
    }

    /**
     * Get the {@code histogram} variant value.
     *
     * @throws IllegalStateException
     *             if the current variant is not of the {@code histogram} kind.
     */
    public HistogramAggregation histogram() {
        return TaggedUnionUtils.get(this, Kind.Histogram);
    }

    /**
     * Is this variant instance of kind {@code terms}?
     */
    public boolean isTerms() {
        return _kind == Kind.Terms;
    }

    /**
     * Get the {@code terms} variant value.
     *
     * @throws IllegalStateException
     *             if the current variant is not of the {@code terms} kind.
     */
    public TermsAggregation terms() {
        return TaggedUnionUtils.get(this, Kind.Terms);
    }

    @Override
    @SuppressWarnings("unchecked")
    public void serialize(JsonGenerator generator, JsonpMapper mapper) {

        generator.writeStartObject();

        generator.writeKey(_kind.jsonValue());
        if (_value instanceof JsonpSerializable) {
            ((JsonpSerializable) _value).serialize(generator, mapper);
        }

        generator.writeEnd();

    }

    @Override
    public String toString() {
        return JsonpUtils.toString(this);
    }

    public static class Builder extends WithJsonObjectBuilderBase implements ObjectBuilder {
        private Kind _kind;
        private Object _value;

        @Override
        protected Builder self() {
            return this;
        }
        public ObjectBuilder dateHistogram(DateHistogramAggregation v) {
            this._kind = Kind.DateHistogram;
            this._value = v;
            return this;
        }

        public ObjectBuilder dateHistogram(
                Function> fn) {
            return this.dateHistogram(fn.apply(new DateHistogramAggregation.Builder()).build());
        }

        public ObjectBuilder histogram(HistogramAggregation v) {
            this._kind = Kind.Histogram;
            this._value = v;
            return this;
        }

        public ObjectBuilder histogram(
                Function> fn) {
            return this.histogram(fn.apply(new HistogramAggregation.Builder()).build());
        }

        public ObjectBuilder terms(TermsAggregation v) {
            this._kind = Kind.Terms;
            this._value = v;
            return this;
        }

        public ObjectBuilder terms(
                Function> fn) {
            return this.terms(fn.apply(new TermsAggregation.Builder()).build());
        }

        public PivotGroupBy build() {
            _checkSingleUse();
            return new PivotGroupBy(this);
        }

    }

    protected static void setupPivotGroupByDeserializer(ObjectDeserializer op) {

        op.add(Builder::dateHistogram, DateHistogramAggregation._DESERIALIZER, "date_histogram");
        op.add(Builder::histogram, HistogramAggregation._DESERIALIZER, "histogram");
        op.add(Builder::terms, TermsAggregation._DESERIALIZER, "terms");

    }

    public static final JsonpDeserializer _DESERIALIZER = ObjectBuilderDeserializer.lazy(Builder::new,
            PivotGroupBy::setupPivotGroupByDeserializer, Builder::build);
}