com.icthh.xm.commons.search.query.dto.NativeSearchQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xm-commons-search Show documentation
xm-commons-search
The newest version!
/*
 * Original version of this file is located at:
 * https://github.com/spring-projects/spring-data-elasticsearch/blob/3.1.12.RELEASE/src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQuery.java
 *
 * Copyright 2013-2019 the original author or authors.
 *
 * 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
 *
 *      https://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 com.icthh.xm.commons.search.query.dto;

import com.icthh.xm.commons.search.builder.QueryBuilder;
import com.icthh.xm.commons.search.builder.aggregation.AbstractAggregationBuilder;
import com.icthh.xm.commons.search.query.AbstractQuery;
import com.icthh.xm.commons.search.query.SearchQuery;
import com.icthh.xm.commons.search.sort.SortBuilder;

import java.util.ArrayList;
import java.util.List;

public class NativeSearchQuery extends AbstractQuery implements SearchQuery {

    private QueryBuilder query;
    private QueryBuilder filter;
    private List sorts;
    private List aggregations;

    public NativeSearchQuery(QueryBuilder query) {
        this.query = query;
    }

    public NativeSearchQuery(QueryBuilder query, QueryBuilder filter) {
        this.query = query;
        this.filter = filter;
    }

    @Override
    public QueryBuilder getQuery() {
        return query;
    }

    @Override
    public QueryBuilder getFilter() {
        return filter;
    }

    public List getElasticsearchSorts() {
        return sorts;
    }

    @Override
    public List getAggregations() {
        return aggregations;
    }

    public void setAggregations(List aggregations) {
        this.aggregations = aggregations;
    }

    public void addAggregation(AbstractAggregationBuilder aggregationBuilder) {
        if (aggregations == null) {
            aggregations = new ArrayList<>();
        }
        aggregations.add(aggregationBuilder);
    }

    public void setSorts(List sortBuilders) {
        this.sorts = sortBuilders;
    }
}