Search in sources :

Example 1 with QueryModelFactory

use of org.alfresco.repo.search.impl.querymodel.QueryModelFactory in project SearchServices by Alfresco.

the class AlfrescoSolrDataModel method getFTSQuery.

/**
 * @param searchParametersAndFilter Pair<SearchParameters, Boolean>
 * @param req SolrQueryRequest
 * @return Query
 * @throws ParseException
 */
public Query getFTSQuery(Pair<SearchParameters, Boolean> searchParametersAndFilter, SolrQueryRequest req, FTSQueryParser.RerankPhase rerankPhase) throws ParseException {
    SearchParameters searchParameters = searchParametersAndFilter.getFirst();
    Boolean isFilter = searchParametersAndFilter.getSecond();
    QueryModelFactory factory = new LuceneQueryModelFactory<Query, Sort, SyntaxError>();
    AlfrescoFunctionEvaluationContext functionContext = new AlfrescoSolr4FunctionEvaluationContext(namespaceDAO, getDictionaryService(CMISStrictDictionaryService.DEFAULT), NamespaceService.CONTENT_MODEL_1_0_URI, req.getSchema());
    FTSParser.Mode mode;
    if (searchParameters.getDefaultFTSOperator() == org.alfresco.service.cmr.search.SearchParameters.Operator.AND) {
        mode = FTSParser.Mode.DEFAULT_CONJUNCTION;
    } else {
        mode = FTSParser.Mode.DEFAULT_DISJUNCTION;
    }
    Constraint constraint = FTSQueryParser.buildFTS(searchParameters.getQuery(), factory, functionContext, null, null, mode, searchParameters.getDefaultFTSOperator() == org.alfresco.service.cmr.search.SearchParameters.Operator.OR ? Connective.OR : Connective.AND, searchParameters.getQueryTemplates(), searchParameters.getDefaultFieldName(), rerankPhase);
    org.alfresco.repo.search.impl.querymodel.Query queryModelQuery = factory.createQuery(null, null, constraint, new ArrayList<Ordering>());
    @SuppressWarnings("unchecked") LuceneQueryBuilder<Query, Sort, ParseException> builder = (LuceneQueryBuilder<Query, Sort, ParseException>) queryModelQuery;
    LuceneQueryBuilderContext<Query, Sort, ParseException> luceneContext = getLuceneQueryBuilderContext(searchParameters, req, CMISStrictDictionaryService.DEFAULT, rerankPhase);
    Set<String> selectorGroup = null;
    if (queryModelQuery.getSource() != null) {
        List<Set<String>> selectorGroups = queryModelQuery.getSource().getSelectorGroups(functionContext);
        if (selectorGroups.size() == 0) {
            throw new UnsupportedOperationException("No selectors");
        }
        if (selectorGroups.size() > 1) {
            throw new UnsupportedOperationException("Advanced join is not supported");
        }
        selectorGroup = selectorGroups.get(0);
    }
    Query luceneQuery = builder.buildQuery(selectorGroup, luceneContext, functionContext);
    // query needs some search parameters fro correct caching ....
    ContextAwareQuery contextAwareQuery = new ContextAwareQuery(luceneQuery, Boolean.TRUE.equals(isFilter) ? null : searchParameters);
    return contextAwareQuery;
}
Also used : Set(java.util.Set) HashSet(java.util.HashSet) Query(org.apache.lucene.search.Query) Constraint(org.alfresco.repo.search.impl.querymodel.Constraint) FTSParser(org.alfresco.repo.search.impl.parsers.FTSParser) SearchParameters(org.alfresco.service.cmr.search.SearchParameters) LuceneQueryModelFactory(org.alfresco.repo.search.impl.querymodel.impl.lucene.LuceneQueryModelFactory) LuceneQueryBuilder(org.alfresco.repo.search.impl.querymodel.impl.lucene.LuceneQueryBuilder) Ordering(org.alfresco.repo.search.impl.querymodel.Ordering) Sort(org.apache.lucene.search.Sort) AlfrescoFunctionEvaluationContext(org.alfresco.repo.search.impl.parsers.AlfrescoFunctionEvaluationContext) ParseException(org.apache.lucene.queryparser.classic.ParseException) LuceneQueryModelFactory(org.alfresco.repo.search.impl.querymodel.impl.lucene.LuceneQueryModelFactory) QueryModelFactory(org.alfresco.repo.search.impl.querymodel.QueryModelFactory)

Example 2 with QueryModelFactory

use of org.alfresco.repo.search.impl.querymodel.QueryModelFactory in project alfresco-repository by Alfresco.

the class AbstractAlfrescoFtsQueryLanguage method executeQuery.

public ResultSet executeQuery(SearchParameters searchParameters) {
    String ftsExpression = searchParameters.getQuery();
    QueryModelFactory factory = queryEngine.getQueryModelFactory();
    AlfrescoFunctionEvaluationContext context = new AlfrescoFunctionEvaluationContext(namespaceService, dictionaryService, searchParameters.getNamespace());
    QueryOptions options = QueryOptions.create(searchParameters);
    FTSParser.Mode mode;
    if (options.getDefaultFTSConnective() == Connective.AND) {
        mode = FTSParser.Mode.DEFAULT_CONJUNCTION;
    } else {
        mode = FTSParser.Mode.DEFAULT_DISJUNCTION;
    }
    Constraint constraint = FTSQueryParser.buildFTS(ftsExpression, factory, context, null, null, mode, options.getDefaultFTSFieldConnective(), searchParameters.getQueryTemplates(), options.getDefaultFieldName(), FTSQueryParser.RerankPhase.SINGLE_PASS);
    org.alfresco.repo.search.impl.querymodel.Query query = factory.createQuery(null, null, constraint, buildOrderings(factory, searchParameters));
    QueryEngineResults results = queryEngine.executeQuery(query, options, context);
    ResultSet resultSet = results.getResults().values().iterator().next();
    return resultSet;
}
Also used : QueryEngineResults(org.alfresco.repo.search.impl.querymodel.QueryEngineResults) Constraint(org.alfresco.repo.search.impl.querymodel.Constraint) AlfrescoFunctionEvaluationContext(org.alfresco.repo.search.impl.parsers.AlfrescoFunctionEvaluationContext) FTSParser(org.alfresco.repo.search.impl.parsers.FTSParser) QueryOptions(org.alfresco.repo.search.impl.querymodel.QueryOptions) ResultSet(org.alfresco.service.cmr.search.ResultSet) QueryModelFactory(org.alfresco.repo.search.impl.querymodel.QueryModelFactory)

Aggregations

AlfrescoFunctionEvaluationContext (org.alfresco.repo.search.impl.parsers.AlfrescoFunctionEvaluationContext)2 FTSParser (org.alfresco.repo.search.impl.parsers.FTSParser)2 Constraint (org.alfresco.repo.search.impl.querymodel.Constraint)2 QueryModelFactory (org.alfresco.repo.search.impl.querymodel.QueryModelFactory)2 HashSet (java.util.HashSet)1 Set (java.util.Set)1 Ordering (org.alfresco.repo.search.impl.querymodel.Ordering)1 QueryEngineResults (org.alfresco.repo.search.impl.querymodel.QueryEngineResults)1 QueryOptions (org.alfresco.repo.search.impl.querymodel.QueryOptions)1 LuceneQueryBuilder (org.alfresco.repo.search.impl.querymodel.impl.lucene.LuceneQueryBuilder)1 LuceneQueryModelFactory (org.alfresco.repo.search.impl.querymodel.impl.lucene.LuceneQueryModelFactory)1 ResultSet (org.alfresco.service.cmr.search.ResultSet)1 SearchParameters (org.alfresco.service.cmr.search.SearchParameters)1 ParseException (org.apache.lucene.queryparser.classic.ParseException)1 Query (org.apache.lucene.search.Query)1 Sort (org.apache.lucene.search.Sort)1