@Override public Iterator<Edge> search(SearchVertexQuery uniQuery) { SelectCollector<JdbcSchema<Edge>, Select, Edge> collector = new SelectCollector<>( schema -> schema.getSearch(uniQuery, ((JdbcEdgeSchema) schema).toPredicates(uniQuery.getVertices(), uniQuery.getDirection(), uniQuery.getPredicates())), (schema, results) -> schema.parseResults(results, uniQuery) ); Map<JdbcSchema<Edge>, Select> selects = edgeSchemas.stream() .filter(schema -> this.traversalFilter.filter(schema, uniQuery.getTraversal())).collect(collector); return this.search(uniQuery, selects, collector); }
@Override public BaseRequest getSearch(SearchVertexQuery query) { int limit = query.getOrders() == null || query.getOrders().size() > 0 ? -1 : query.getLimit(); PredicatesHolder edgePredicates = this.toPredicates(query.getPredicates()); PredicatesHolder vertexPredicates = this.getVertexPredicates(query.getVertices(), query.getDirection()); PredicatesHolder predicatesHolder = PredicatesHolderFactory.and(edgePredicates, vertexPredicates); return createSearch(predicatesHolder, limit); }
@Override public QueryBuilder getSearch(SearchVertexQuery query) { PredicatesHolder edgePredicates = this.toPredicates(query.getPredicates()); PredicatesHolder vertexPredicates = this.getVertexPredicates(query.getVertices(), query.getDirection()); PredicatesHolder predicatesHolder = PredicatesHolderFactory.and(edgePredicates, vertexPredicates); if (predicatesHolder.isAborted()) return null; QueryBuilder queryBuilder = createQueryBuilder(predicatesHolder); return queryBuilder; // return createSearch(query, queryBuilder); }
public QueryBuilder createQueryBuilder(SearchVertexQuery query) { PredicatesHolder edgePredicates = this.toPredicates(query.getPredicates()); if(edgePredicates.isAborted()) return null; PredicatesHolder childPredicates = childVertexSchema.toPredicates(query.getVertices()); childPredicates = PredicatesHolderFactory.and(edgePredicates, childPredicates); QueryBuilder childQuery = createNestedQueryBuilder(childPredicates); if(query.getDirection().equals(parentDirection.opposite())) { if (childPredicates.isAborted()) return null; return childQuery; } else if (!query.getDirection().equals(Direction.BOTH)) childQuery = null; PredicatesHolder parentPredicates = parentVertexSchema.toPredicates(query.getVertices()); QueryBuilder parentQuery = createQueryBuilder(parentPredicates); if(parentQuery != null) { // if (parentPredicates.isAborted()) return null; QueryBuilder edgeQuery = createNestedQueryBuilder(edgePredicates); if (edgeQuery != null) { parentQuery = QueryBuilders.boolQuery().must(parentQuery).must(edgeQuery); } } if(query.getDirection().equals(parentDirection) && parentPredicates.notAborted()) return parentQuery; else if(childQuery == null && parentPredicates.notAborted()) return parentQuery; else if(parentQuery == null && childPredicates.notAborted()) return childQuery; else if(parentPredicates.isAborted() && childPredicates.isAborted()) return null; else return QueryBuilders.boolQuery().should(parentQuery).should(childQuery); }