コード例 #1
0
  private Predicates getPredicates(Step step, Traversal.Admin traversal) {
    Predicates predicates = new Predicates();
    Step<?, ?> nextStep = step.getNextStep();

    while (true) {
      if (nextStep instanceof HasContainerHolder) {
        HasContainerHolder hasContainerHolder = (HasContainerHolder) nextStep;
        boolean skip = false;
        for (HasContainer has : hasContainerHolder.getHasContainers())
          if (has.getPredicate().getTraversals().size() > 0) skip = true;

        if (!skip) {
          hasContainerHolder.getHasContainers().forEach((has) -> predicates.hasContainers.add(has));
          nextStep.getLabels().forEach(label -> predicates.labels.add(label.toString()));
          traversal.removeStep(nextStep);
        }
      } else if (nextStep instanceof RangeGlobalStep) {
        RangeGlobalStep rangeGlobalStep = (RangeGlobalStep) nextStep;
        predicates.limitLow = rangeGlobalStep.getLowRange();
        predicates.limitHigh = rangeGlobalStep.getHighRange();
        traversal.removeStep(nextStep);
      } else return predicates;

      nextStep = nextStep.getNextStep();
    }
  }
コード例 #2
0
  @Override
  public void apply(Traversal.Admin<?, ?> traversal) {
    if (traversal.getEngine().isComputer()) return;
    Graph graph = traversal.getGraph().get();
    if (!(graph instanceof ElasticGraph)) return;
    ElasticGraph elasticGraph = (ElasticGraph) graph;

    TraversalHelper.getStepsOfClass(GraphStep.class, traversal)
        .forEach(
            graphStep -> {
              if (graphStep.getIds().length == 0) {
                Predicates predicates = getPredicates(graphStep, traversal);
                final ElasticGraphStep<?> elasticGraphStep =
                    new ElasticGraphStep<>(graphStep, predicates, elasticGraph.elasticService);
                TraversalHelper.replaceStep(graphStep, (Step) elasticGraphStep, traversal);
              }
            });

    TraversalHelper.getStepsOfClass(VertexStep.class, traversal)
        .forEach(
            vertexStep -> {
              boolean returnVertex = vertexStep.getReturnClass().equals(Vertex.class);
              Predicates predicates =
                  returnVertex ? new Predicates() : getPredicates(vertexStep, traversal);

              ElasticVertexStep elasticVertexStep =
                  new ElasticVertexStep(vertexStep, predicates, elasticGraph.elasticService);
              TraversalHelper.replaceStep(vertexStep, elasticVertexStep, traversal);
            });
  }