com.github.wz2cool.elasticsearch.operator.RangeOperator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-dynamic-query Show documentation
There is a newer version: 4.2.6
Show newest version
package com.github.wz2cool.elasticsearch.operator;

import com.github.wz2cool.elasticsearch.model.FilterMode;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.RangeQueryBuilder;

import java.util.Objects;

public class RangeOperator implements IFilterOperator {

    private R gt;
    private R gte;
    private R lt;
    private R lte;
    private String timezone;
    private String relation;

    /**
     * @see RangeQueryBuilder#gt(Object)
     */
    public RangeOperator gt(boolean enable, R from) {
        if (enable) {
            this.gt = from;
        }
        return this;
    }

    /**
     * @see RangeQueryBuilder#gt(Object)
     */
    public RangeOperator gt(R from) {
        this.gt = from;
        return this;
    }

    /**
     * @see RangeQueryBuilder#gte(Object)
     */
    public RangeOperator gte(boolean enable, R from) {
        if (enable) {
            this.gte = from;
        }
        return this;
    }

    /**
     * @see RangeQueryBuilder#gte(Object)
     */
    public RangeOperator gte(R from) {
        this.gte = from;
        return this;
    }

    /**
     * @see RangeQueryBuilder#lt(Object)
     */
    public RangeOperator lt(boolean enable, R to) {
        if (enable) {
            this.lt = to;
        }
        return this;
    }

    /**
     * @see RangeQueryBuilder#lt(Object)
     */
    public RangeOperator lt(R to) {
        this.lt = to;
        return this;
    }

    /**
     * @see RangeQueryBuilder#lte(Object)
     */
    public RangeOperator lte(boolean enable, R to) {
        if (enable) {
            this.lte = to;
        }
        return this;
    }

    /**
     * @see RangeQueryBuilder#lte(Object)
     */
    public RangeOperator lte(R to) {
        this.lte = to;
        return this;
    }

    /**
     * @see RangeQueryBuilder#timeZone(String)
     */
    public RangeOperator timezone(String timeZone) {
        this.timezone = timeZone;
        return this;
    }

    /**
     * @see RangeQueryBuilder#relation(String)
     */
    public RangeOperator relation(String relation) {
        this.relation = relation;
        return this;
    }

    @Override
    public FilterMode getDefaultFilterMode() {
        return FilterMode.FILTER;
    }

    @Override
    public QueryBuilder buildQuery(String columnName) {
        RangeQueryBuilder rangeQueryBuilder = new RangeQueryBuilder(columnName);
        if (Objects.nonNull(gt)) {
            rangeQueryBuilder.gt(getFilterValue(gt));
        }
        if (Objects.nonNull(gte)) {
            rangeQueryBuilder.gte(getFilterValue(gte));
        }
        if (Objects.nonNull(lt)) {
            rangeQueryBuilder.lt(getFilterValue(lt));
        }
        if (Objects.nonNull(lte)) {
            rangeQueryBuilder.lte(getFilterValue(lte));
        }
        if (Objects.nonNull(timezone)) {
            rangeQueryBuilder.timeZone(timezone);
        }
        if (Objects.nonNull(relation)) {
            rangeQueryBuilder.relation(relation);
        }
        return rangeQueryBuilder;
    }
}