com.github.wz2cool.elasticsearch.query.builder.TermsExtQueryBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-data-elasticsearch-extension Show documentation

There is a newer version: 0.1.6

Show newest version

package com.github.wz2cool.elasticsearch.query.builder;

import com.github.wz2cool.elasticsearch.lambda.GetArrayPropertyFunction;
import com.github.wz2cool.elasticsearch.lambda.GetPropertyFunction;
import com.github.wz2cool.elasticsearch.model.ColumnInfo;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.TermsQueryBuilder;

import java.util.Arrays;
import java.util.stream.Collectors;

/**
 * @author Frank
 * @see TermsExtQueryBuilder
 */
public class TermsExtQueryBuilder implements ExtQueryBuilder {

    private TermsQueryBuilder termsQueryBuilder;

    /**
     * @see TermsQueryBuilder#TermsQueryBuilder(String, Iterable)
     */
    @SafeVarargs
    public TermsExtQueryBuilder(GetPropertyFunction getPropertyFunc, R... values) {
        final ColumnInfo columnInfo = getColumnInfo(getPropertyFunc);
        this.termsQueryBuilder = new TermsQueryBuilder(columnInfo.getColumnName(),
                Arrays.stream(values).map(this::getFilterValue).collect(Collectors.toList()));
    }

    /**
     * @see TermsQueryBuilder#TermsQueryBuilder(String, Iterable)
     */
    @SafeVarargs
    public TermsExtQueryBuilder(GetArrayPropertyFunction getPropertyFunc, R... values) {
        final ColumnInfo columnInfo = getColumnInfo(getPropertyFunc);
        this.termsQueryBuilder = new TermsQueryBuilder(columnInfo.getColumnName(),
                Arrays.stream(values).map(this::getFilterValue).collect(Collectors.toList()));
    }

    @Override
    public QueryBuilder build() {
        return this.termsQueryBuilder;
    }
}