Examples with QueryBuilder org.apache.lucene.queryparser.flexible.core.builders.QueryBuilder used on opensource projects

Example 1 with QueryBuilder

use of org.apache.lucene.queryparser.flexible.core.builders.QueryBuilder in project datawave by NationalSecurityAgency.

the class LuceneToJexlQueryParser method convertToJexlNode.

public JexlNode convertToJexlNode(String query) throws ParseException {
    // replace open smart quote 147
    query = query.replaceAll("\\u0093", "\"");
    // replace close smart quote 148
    query = query.replaceAll("\\u0094", "\"");
    // replace open left double quote
    query = query.replaceAll("\\u201c", "\"");
    // replace close right double quote
    query = query.replaceAll("\\u201d", "\"");
    JexlNode parsedQuery = null;
    try {
        Locale.setDefault(Locale.US);
        AccumuloSyntaxParser syntaxParser = new AccumuloSyntaxParser();
        syntaxParser.enable_tracing();
        QueryNodeProcessor processor = getQueryNodeProcessor();
        QueryBuilder builder = new JexlTreeBuilder(allowedFunctions);
        org.apache.lucene.queryparser.flexible.core.nodes.QueryNode queryTree = syntaxParser.parse(query, "");
        queryTree = processor.process(queryTree);
        parsedQuery = (JexlNode) builder.build(queryTree);
    } catch (Exception e) {
        throw new ParseException(e);
    }
    return parsedQuery;
}

Example 2 with QueryBuilder

use of org.apache.lucene.queryparser.flexible.core.builders.QueryBuilder in project datawave by NationalSecurityAgency.

the class LuceneQueryParser method parse.

@Override
public datawave.query.language.tree.QueryNode parse(String query) throws ParseException {
    // replace open smart quote 147
    query = query.replaceAll("\\u0093", "\"");
    // replace close smart quote 148
    query = query.replaceAll("\\u0094", "\"");
    datawave.query.language.tree.QueryNode parsedQuery = null;
    try {
        Locale.setDefault(Locale.US);
        AccumuloSyntaxParser syntaxParser = new AccumuloSyntaxParser();
        syntaxParser.enable_tracing();
        org.apache.lucene.queryparser.flexible.core.config.QueryConfigHandler queryConfigHandler = new QueryConfigHandler();
        QueryNodeProcessor processor = new CustomQueryNodeProcessorPipeline(queryConfigHandler);
        QueryBuilder builder = null;
        if (allowedFunctions == null) {
            builder = new AccumuloQueryTreeBuilder();
        } else {
            builder = new AccumuloQueryTreeBuilder(allowedFunctions);
        }
        QueryNode queryTree = syntaxParser.parse(query, "");
        queryTree = processor.process(queryTree);
        parsedQuery = (datawave.query.language.tree.QueryNode) builder.build(queryTree);
        Set<FieldedTerm> positiveFilters = new TreeSet<>();
        if (log.isTraceEnabled()) {
            log.trace("Query before filters extracted: " + parsedQuery.getContents());
        }
        extractFilters(parsedQuery, positiveFilters);
        parsedQuery.setPositiveFilters(positiveFilters);
        if (log.isTraceEnabled()) {
            log.trace("Query after filters extracted: " + parsedQuery.getContents());
        }
    } catch (QueryNodeException | RuntimeException e) {
        throw new ParseException(e);
    }
    return parsedQuery;
}