Examples with BooleanClause org.apache.lucene.search.BooleanClause used on opensource projects

Example 51 with BooleanClause

use of org.apache.lucene.search.BooleanClause in project elasticsearch by elastic.

the class MapperQueryParser method getBooleanQueryCoordDisabled.

/**
     * @deprecated review all use of this, don't rely on coord
     */
@Deprecated
protected Query getBooleanQueryCoordDisabled(List<BooleanClause> clauses) throws ParseException {
    BooleanQuery.Builder builder = new BooleanQuery.Builder();
    builder.setDisableCoord(true);
    for (BooleanClause clause : clauses) {
        builder.add(clause);
    }
    return fixNegativeQueryIfNeeded(builder.build());
}

Example 52 with BooleanClause

use of org.apache.lucene.search.BooleanClause in project elasticsearch by elastic.

the class MapperQueryParser method getWildcardQuery.

@Override
protected Query getWildcardQuery(String field, String termStr) throws ParseException {
    if (termStr.equals("*") && field != null) {
        if ("*".equals(field)) {
            return newMatchAllDocsQuery();
        }
        String actualField = field;
        if (actualField == null) {
            actualField = this.field;
        }
        // effectively, we check if a field exists or not
        return FIELD_QUERY_EXTENSIONS.get(ExistsFieldQueryExtension.NAME).query(context, actualField);
    }
    Collection<String> fields = extractMultiFields(field);
    if (fields != null) {
        if (fields.size() == 1) {
            return getWildcardQuerySingle(fields.iterator().next(), termStr);
        }
        if (settings.useDisMax()) {
            List<Query> queries = new ArrayList<>();
            boolean added = false;
            for (String mField : fields) {
                Query q = getWildcardQuerySingle(mField, termStr);
                if (q != null) {
                    added = true;
                    queries.add(applyBoost(mField, q));
                }
            }
            if (!added) {
                return null;
            }
            return new DisjunctionMaxQuery(queries, settings.tieBreaker());
        } else {
            List<BooleanClause> clauses = new ArrayList<>();
            for (String mField : fields) {
                Query q = getWildcardQuerySingle(mField, termStr);
                if (q != null) {
                    clauses.add(new BooleanClause(applyBoost(mField, q), BooleanClause.Occur.SHOULD));
                }
            }
            // happens for stopwords
            if (clauses.isEmpty())
                return null;
            return getBooleanQueryCoordDisabled(clauses);
        }
    } else {
        return getWildcardQuerySingle(field, termStr);
    }
}

Example 53 with BooleanClause

use of org.apache.lucene.search.BooleanClause in project elasticsearch by elastic.

the class MapperQueryParser method getPrefixQuery.

@Override
protected Query getPrefixQuery(String field, String termStr) throws ParseException {
    Collection<String> fields = extractMultiFields(field);
    if (fields != null) {
        if (fields.size() == 1) {
            return getPrefixQuerySingle(fields.iterator().next(), termStr);
        }
        if (settings.useDisMax()) {
            List<Query> queries = new ArrayList<>();
            boolean added = false;
            for (String mField : fields) {
                Query q = getPrefixQuerySingle(mField, termStr);
                if (q != null) {
                    added = true;
                    queries.add(applyBoost(mField, q));
                }
            }
            if (!added) {
                return null;
            }
            return new DisjunctionMaxQuery(queries, settings.tieBreaker());
        } else {
            List<BooleanClause> clauses = new ArrayList<>();
            for (String mField : fields) {
                Query q = getPrefixQuerySingle(mField, termStr);
                if (q != null) {
                    clauses.add(new BooleanClause(applyBoost(mField, q), BooleanClause.Occur.SHOULD));
                }
            }
            // happens for stopwords
            if (clauses.isEmpty())
                return null;
            return getBooleanQueryCoordDisabled(clauses);
        }
    } else {
        return getPrefixQuerySingle(field, termStr);
    }
}

Example 54 with BooleanClause

use of org.apache.lucene.search.BooleanClause in project elasticsearch by elastic.

the class MapperQueryParser method getRangeQuery.

@Override
protected Query getRangeQuery(String field, String part1, String part2, boolean startInclusive, boolean endInclusive) throws ParseException {
    if ("*".equals(part1)) {
        part1 = null;
    }
    if ("*".equals(part2)) {
        part2 = null;
    }
    Collection<String> fields = extractMultiFields(field);
    if (fields == null) {
        return getRangeQuerySingle(field, part1, part2, startInclusive, endInclusive, context);
    }
    if (fields.size() == 1) {
        return getRangeQuerySingle(fields.iterator().next(), part1, part2, startInclusive, endInclusive, context);
    }
    if (settings.useDisMax()) {
        List<Query> queries = new ArrayList<>();
        boolean added = false;
        for (String mField : fields) {
            Query q = getRangeQuerySingle(mField, part1, part2, startInclusive, endInclusive, context);
            if (q != null) {
                added = true;
                queries.add(applyBoost(mField, q));
            }
        }
        if (!added) {
            return null;
        }
        return new DisjunctionMaxQuery(queries, settings.tieBreaker());
    } else {
        List<BooleanClause> clauses = new ArrayList<>();
        for (String mField : fields) {
            Query q = getRangeQuerySingle(mField, part1, part2, startInclusive, endInclusive, context);
            if (q != null) {
                clauses.add(new BooleanClause(applyBoost(mField, q), BooleanClause.Occur.SHOULD));
            }
        }
        // happens for stopwords
        if (clauses.isEmpty())
            return null;
        return getBooleanQueryCoordDisabled(clauses);
    }
}

Example 55 with BooleanClause

use of org.apache.lucene.search.BooleanClause in project elasticsearch by elastic.

the class BoolQueryBuilder method addBooleanClauses.

private static void addBooleanClauses(QueryShardContext context, BooleanQuery.Builder booleanQueryBuilder, List<QueryBuilder> clauses, Occur occurs) throws IOException {
    for (QueryBuilder query : clauses) {
        Query luceneQuery = null;
        switch(occurs) {
            case MUST:
            case SHOULD:
                luceneQuery = query.toQuery(context);
                break;
            case FILTER:
            case MUST_NOT:
                luceneQuery = query.toFilter(context);
                break;
        }
        booleanQueryBuilder.add(new BooleanClause(luceneQuery, occurs));
    }
}