Examples with YqlParser com.yahoo.search.yql.YqlParser used on opensource projects

Example 1 with YqlParser

use of com.yahoo.search.yql.YqlParser in project vespa by vespa-engine.

the class GroupingParserTestCase method assertYqlParsable.

private static void assertYqlParsable(String request, String... expectedOperations) {
    YqlParser parser = new YqlParser(new ParserEnvironment());
    parser.parse(new Parsable().setQuery("select foo from bar where baz contains 'baz' | " + request + ";"));
    List<VespaGroupingStep> steps = parser.getGroupingSteps();
    List<String> actual = new ArrayList<>(steps.size());
    for (VespaGroupingStep step : steps) {
        actual.add(step.getOperation().toString());
    }
    if (expectedOperations.length > 0) {
        assertEquals(Arrays.asList(expectedOperations), actual);
    }
}

Example 2 with YqlParser

use of com.yahoo.search.yql.YqlParser in project vespa by vespa-engine.

the class ValidatePredicateSearcherTestCase method doSearch.

private static Result doSearch(ValidatePredicateSearcher searcher, String yqlQuery, String command) {
    QueryTree queryTree = new YqlParser(new ParserEnvironment()).parse(new Parsable().setQuery(yqlQuery));
    Query query = new Query();
    query.getModel().getQueryTree().setRoot(queryTree.getRoot());
    TreeMap<String, List<String>> masterClusters = new TreeMap<>();
    masterClusters.put("cluster", Arrays.asList("document"));
    SearchDefinition searchDefinition = new SearchDefinition("document");
    Index index = new Index("predicate_field");
    index.addCommand(command);
    searchDefinition.addIndex(index);
    Map<String, SearchDefinition> searchDefinitionMap = new HashMap<>();
    searchDefinitionMap.put("document", searchDefinition);
    IndexFacts indexFacts = new IndexFacts(new IndexModel(masterClusters, searchDefinitionMap, searchDefinition));
    Execution.Context context = new Execution.Context(null, indexFacts, null, new RendererRegistry(MoreExecutors.directExecutor()), new SimpleLinguistics());
    return new Execution(searcher, context).search(query);
}