protected PredicatesHolder getVertexPredicates(List<Vertex> vertices, Direction direction) { PredicatesHolder outPredicates = this.outVertexSchema.toPredicates(vertices); PredicatesHolder inPredicates = this.inVertexSchema.toPredicates(vertices); if(direction.equals(Direction.OUT)) return outPredicates; if(direction.equals(Direction.IN)) return inPredicates; return PredicatesHolderFactory.or(inPredicates, outPredicates); }
/** * Receives a list of vertices and converts them to predicates * @param vertices A list of vertices * @return Predicate holder */ default PredicatesHolder toPredicates(List<? extends Vertex> vertices) { if(vertices == null || vertices.size() == 0) return PredicatesHolderFactory.abort(); HashSet<PredicatesHolder> predicates = new HashSet<>(); vertices.stream().collect(Collectors.groupingBy(Vertex::label)).forEach((label, labelVertices) -> { HasContainer labelPredicate = new HasContainer(T.label.getAccessor(), P.eq(label)); HasContainer ids = new HasContainer(T.id.getAccessor(), P.within(labelVertices.stream().map(Vertex::id).collect(Collectors.toSet()))); PredicatesHolder labelPredicates = PredicatesHolderFactory.and(ids, labelPredicate); PredicatesHolder toPredicates = toPredicates(labelPredicates); predicates.add(toPredicates); }); return PredicatesHolderFactory.or(predicates); } }
protected PredicatesHolder getVertexPredicates(List<Vertex> vertices, Direction direction) { PredicatesHolder outPredicates = this.outVertexSchema.toPredicates(vertices); PredicatesHolder inPredicates = this.inVertexSchema.toPredicates(vertices); if(direction.equals(Direction.OUT) && outPredicates.notAborted()) return outPredicates; if(direction.equals(Direction.IN) && inPredicates.notAborted()) return inPredicates; if (outPredicates.notAborted() && inPredicates.notAborted()) return PredicatesHolderFactory.or(inPredicates, outPredicates); else if (outPredicates.isAborted()) return inPredicates; else if (inPredicates.isAborted()) return outPredicates; else return PredicatesHolderFactory.abort(); }
protected PredicatesHolder getVertexPredicates(List<Vertex> vertices, Direction direction) { PredicatesHolder outPredicates = this.outVertexSchema.toPredicates(vertices); PredicatesHolder inPredicates = this.inVertexSchema.toPredicates(vertices); if (direction.equals(Direction.OUT) && outPredicates.notAborted()) return outPredicates; if (direction.equals(Direction.IN) && inPredicates.notAborted()) return inPredicates; if (outPredicates.notAborted() && inPredicates.notAborted()) return PredicatesHolderFactory.or(inPredicates, outPredicates); else if (outPredicates.isAborted()) return inPredicates; else if (inPredicates.isAborted()) return outPredicates; else return PredicatesHolderFactory.abort(); } }
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); }