Examples with QueryContext org.codelibs.fess.entity.QueryContext used on opensource projects

Example 1 with QueryContext

use of org.codelibs.fess.entity.QueryContext in project fess by codelibs.

the class SearchService method deleteByQuery.

public int deleteByQuery(final HttpServletRequest request, final SearchRequestParams params) {
    final String query = QueryStringBuilder.query(params.getQuery()).extraQueries(params.getExtraQueries()).fields(params.getFields()).build();
    final QueryContext queryContext = queryHelper.build(params.getType(), query, context -> {
        context.skipRoleQuery();
    });
    return fessEsClient.deleteByQuery(fessConfig.getIndexDocumentUpdateIndex(), fessConfig.getIndexDocumentType(), queryContext.getQueryBuilder());
}

Example 2 with QueryContext

use of org.codelibs.fess.entity.QueryContext in project fess by codelibs.

the class QueryHelper method convertFuzzyQuery.

protected QueryBuilder convertFuzzyQuery(final QueryContext context, final FuzzyQuery fuzzyQuery, final float boost) {
    final Term term = fuzzyQuery.getTerm();
    final String field = term.field();
    // TODO fuzzy value
    if (Constants.DEFAULT_FIELD.equals(field)) {
        context.addFieldLog(field, term.text());
        return buildDefaultQueryBuilder((f, b) -> QueryBuilders.fuzzyQuery(f, term.text()).fuzziness(Fuzziness.fromEdits(fuzzyQuery.getMaxEdits())).boost(b * boost));
    } else if (isSearchField(field)) {
        context.addFieldLog(field, term.text());
        return QueryBuilders.fuzzyQuery(field, term.text()).boost(boost).fuzziness(Fuzziness.fromEdits(fuzzyQuery.getMaxEdits()));
    } else {
        final String origQuery = fuzzyQuery.toString();
        context.addFieldLog(Constants.DEFAULT_FIELD, origQuery);
        context.addHighlightedQuery(origQuery);
        return buildDefaultQueryBuilder((f, b) -> QueryBuilders.fuzzyQuery(f, origQuery).fuzziness(Fuzziness.fromEdits(fuzzyQuery.getMaxEdits())).boost(b * boost));
    }
}

Example 3 with QueryContext

use of org.codelibs.fess.entity.QueryContext in project fess by codelibs.

the class QueryHelper method build.

public QueryContext build(final SearchRequestType searchRequestType, final String query, final Consumer<QueryContext> context) {
    String q;
    if (additionalQuery != null && StringUtil.isNotBlank(query)) {
        q = query + " " + additionalQuery;
    } else {
        q = query;
    }
    final QueryContext queryContext = new QueryContext(q, true);
    buildBaseQuery(queryContext, context);
    buildBoostQuery(queryContext);
    buildRoleQuery(queryContext, searchRequestType);
    if (!queryContext.hasSorts() && defaultSortBuilders != null) {
        queryContext.addSorts(defaultSortBuilders);
    }
    return queryContext;
}

Example 4 with QueryContext

use of org.codelibs.fess.entity.QueryContext in project fess by codelibs.

the class QueryHelper method convertTermQuery.

protected QueryBuilder convertTermQuery(final QueryContext context, final TermQuery termQuery, final float boost) {
    final String field = termQuery.getTerm().field();
    final String text = termQuery.getTerm().text();
    if (fessConfig.getQueryReplaceTermWithPrefixQueryAsBoolean() && text.length() > 1 && text.endsWith("*")) {
        return convertPrefixQuery(context, new PrefixQuery(new Term(field, text.substring(0, text.length() - 1))), boost);
    } else if (Constants.DEFAULT_FIELD.equals(field)) {
        context.addFieldLog(field, text);
        context.addHighlightedQuery(text);
        return buildDefaultQueryBuilder((f, b) -> QueryBuilders.matchPhraseQuery(f, text).boost(b * boost));
    } else if ("sort".equals(field)) {
        final String[] values = text.split("\\.");
        if (values.length > 2) {
            throw new InvalidQueryException(messages -> messages.addErrorsInvalidQuerySortValue(UserMessages.GLOBAL_PROPERTY_KEY, text), "Invalid sort field: " + termQuery);
        }
        final String sortField = values[0];
        if (!isSortField(sortField)) {
            throw new InvalidQueryException(messages -> messages.addErrorsInvalidQueryUnsupportedSortField(UserMessages.GLOBAL_PROPERTY_KEY, sortField), "Unsupported sort field: " + termQuery);
        }
        SortOrder sortOrder;
        if (values.length == 2) {
            sortOrder = SortOrder.DESC.toString().equalsIgnoreCase(values[1]) ? SortOrder.DESC : SortOrder.ASC;
            if (sortOrder == null) {
                throw new InvalidQueryException(messages -> messages.addErrorsInvalidQueryUnsupportedSortOrder(UserMessages.GLOBAL_PROPERTY_KEY, values[1]), "Invalid sort order: " + termQuery);
            }
        } else {
            sortOrder = SortOrder.ASC;
        }
        context.addSorts(createFieldSortBuilder(sortField, sortOrder));
        return null;
    } else if (INURL_FIELD.equals(field)) {
        return QueryBuilders.wildcardQuery(fessConfig.getIndexFieldUrl(), "*" + text + "*").boost(boost);
    } else if (isSearchField(field)) {
        context.addFieldLog(field, text);
        context.addHighlightedQuery(text);
        if (notAnalyzedFieldSet.contains(field)) {
            return QueryBuilders.termQuery(field, text).boost(boost);
        } else {
            return QueryBuilders.matchPhraseQuery(field, text).boost(boost);
        }
    } else {
        final String origQuery = termQuery.toString();
        context.addFieldLog(Constants.DEFAULT_FIELD, origQuery);
        context.addHighlightedQuery(origQuery);
        return buildDefaultQueryBuilder((f, b) -> QueryBuilders.matchPhraseQuery(f, origQuery).boost(b * boost));
    }
}

Example 5 with QueryContext

use of org.codelibs.fess.entity.QueryContext in project fess by codelibs.

the class QueryHelper method buildBaseQuery.

public void buildBaseQuery(final QueryContext queryContext, final Consumer<QueryContext> context) {
    try {
        final Query query = getQueryParser().parse(queryContext.getQueryString());
        final QueryBuilder queryBuilder = convertQuery(queryContext, query, 1.0f);
        if (queryBuilder != null) {
            queryContext.setQueryBuilder(queryBuilder);
        } else {
            queryContext.setQueryBuilder(QueryBuilders.matchAllQuery());
        }
        // TODO options query
        context.accept(queryContext);
    } catch (final ParseException e) {
        throw new InvalidQueryException(messages -> messages.addErrorsInvalidQueryParseError(UserMessages.GLOBAL_PROPERTY_KEY), "Invalid query: " + queryContext.getQueryString(), e);
    }
}