me.phoboslabs.illuminati.elasticsearch.infra.param.RequestEsParam Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017 Phoboslabs.me
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package me.phoboslabs.illuminati.elasticsearch.infra.param;

import com.google.gson.annotations.Expose;
import com.google.gson.annotations.SerializedName;
import me.phoboslabs.illuminati.common.constant.IlluminatiConstant;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class RequestEsParam {

    @Expose
    private int size = 10;
    @Expose
    private int from = 0;
    @Expose
    private Map query;
    @Expose
    private Map sort = new HashMap<>();
    @Expose @SerializedName("_source")
    private List source;
    @Expose
    private Map aggs;

    public static RequestEsParam Builder () {
        return new RequestEsParam();
    }
    public static RequestEsParam Builder (Map query) {
        return new RequestEsParam(query);
    }
    public static RequestEsParam Builder (Map query, List source) {
        return new RequestEsParam(query, source);
    }

    private RequestEsParam () {

    }

    private RequestEsParam (Map query) {
        this.query = query;
    }

    private RequestEsParam (Map query, List source) {
        this.query = query;
        this.source = source;
    }

    public RequestEsParam setSize (int size) {
        this.size = size;
        return this;
    }

    public RequestEsParam setFrom (int from) {
        this.from = from;
        return this;
    }

    public RequestEsParam setSource (List source) {
        this.source = source;
        return this;
    }

    public RequestEsParam setSort (Map sort) {
        this.sort = sort;
        return this;
    }

    public RequestEsParam setGroupBy (Map groupBy) {
        this.aggs = groupBy;
        return this;
    }

    public RequestEsParam setQuery (Map query) {
        this.query = query;
        return this;
    }

    private void resetFieldsWithoutAggregation () {
        this.size = 0;
        this.from = 0;

        if (this.source != null) {
            this.source.clear();
        } else {
            this.source = new ArrayList<>();
        }
        this.resetSort();
    }

    private void resetAggregationField () {
        this.aggs = new HashMap<>();
    }

    private void resetSort () {
        if (this.sort != null) {
            this.sort.clear();
        } else {
            this.sort = new HashMap<>();
        }
    }

    public String build () {
        if (this.aggs != null) {
            this.resetFieldsWithoutAggregation();
        } else {
            this.resetAggregationField();
        }
        return IlluminatiConstant.ILLUMINATI_GSON_OBJ.toJson(this);
    }
}