Example usage for org.springframework.data.elasticsearch.core.query Criteria isOr

List of usage examples for org.springframework.data.elasticsearch.core.query Criteria isOr

Introduction

In this page you can find the example usage for org.springframework.data.elasticsearch.core.query Criteria isOr.

Prototype

public boolean isOr() 

Source Link

Usage

From source file:com.github.vanroy.springdata.jest.CriteriaFilterProcessor.java

QueryBuilder createFilterFromCriteria(Criteria criteria) {
    List<QueryBuilder> fbList = new LinkedList<QueryBuilder>();
    QueryBuilder filter = null;//from   ww  w . ja  v  a  2s  .  c  o m

    ListIterator<Criteria> chainIterator = criteria.getCriteriaChain().listIterator();

    while (chainIterator.hasNext()) {
        QueryBuilder fb = null;
        Criteria chainedCriteria = chainIterator.next();
        if (chainedCriteria.isOr()) {
            fb = QueryBuilders.boolQuery();
            for (QueryBuilder f : createFilterFragmentForCriteria(chainedCriteria)) {
                ((BoolQueryBuilder) fb).should(f);
            }
            fbList.add(fb);
        } else if (chainedCriteria.isNegating()) {
            List<QueryBuilder> negationFilters = buildNegationFilter(criteria.getField().getName(),
                    criteria.getFilterCriteriaEntries().iterator());

            if (!negationFilters.isEmpty()) {
                fbList.addAll(negationFilters);
            }
        } else {
            fbList.addAll(createFilterFragmentForCriteria(chainedCriteria));
        }
    }

    if (!fbList.isEmpty()) {
        if (fbList.size() == 1) {
            filter = fbList.get(0);
        } else {
            filter = QueryBuilders.boolQuery();
            for (QueryBuilder f : fbList) {
                ((BoolQueryBuilder) filter).must(f);
            }
        }
    }
    return filter;
}

From source file:com.github.vanroy.springdata.jest.CriteriaQueryProcessor.java

QueryBuilder createQueryFromCriteria(Criteria criteria) {
    if (criteria == null)
        return null;

    List<QueryBuilder> shouldQueryBuilderList = new LinkedList<QueryBuilder>();
    List<QueryBuilder> mustNotQueryBuilderList = new LinkedList<QueryBuilder>();
    List<QueryBuilder> mustQueryBuilderList = new LinkedList<QueryBuilder>();

    ListIterator<Criteria> chainIterator = criteria.getCriteriaChain().listIterator();

    QueryBuilder firstQuery = null;// w  w w .  ja  v  a  2  s  .c  o m
    boolean negateFirstQuery = false;

    while (chainIterator.hasNext()) {
        Criteria chainedCriteria = chainIterator.next();
        QueryBuilder queryFragmentForCriteria = createQueryFragmentForCriteria(chainedCriteria);
        if (queryFragmentForCriteria != null) {
            if (firstQuery == null) {
                firstQuery = queryFragmentForCriteria;
                negateFirstQuery = chainedCriteria.isNegating();
                continue;
            }
            if (chainedCriteria.isOr()) {
                shouldQueryBuilderList.add(queryFragmentForCriteria);
            } else if (chainedCriteria.isNegating()) {
                mustNotQueryBuilderList.add(queryFragmentForCriteria);
            } else {
                mustQueryBuilderList.add(queryFragmentForCriteria);
            }
        }
    }

    if (firstQuery != null) {
        if (!shouldQueryBuilderList.isEmpty() && mustNotQueryBuilderList.isEmpty()
                && mustQueryBuilderList.isEmpty()) {
            shouldQueryBuilderList.add(0, firstQuery);
        } else {
            if (negateFirstQuery) {
                mustNotQueryBuilderList.add(0, firstQuery);
            } else {
                mustQueryBuilderList.add(0, firstQuery);
            }
        }
    }

    BoolQueryBuilder query = null;

    if (!shouldQueryBuilderList.isEmpty() || !mustNotQueryBuilderList.isEmpty()
            || !mustQueryBuilderList.isEmpty()) {

        query = boolQuery();

        for (QueryBuilder qb : shouldQueryBuilderList) {
            query.should(qb);
        }
        for (QueryBuilder qb : mustNotQueryBuilderList) {
            query.mustNot(qb);
        }
        for (QueryBuilder qb : mustQueryBuilderList) {
            query.must(qb);
        }
    }

    return query;
}