Examples with Admin org.apache.tinkerpop.gremlin.process.traversal.Traversal.Admin used on opensource projects

Example 1 with Admin

use of org.apache.tinkerpop.gremlin.process.traversal.Traversal.Admin in project janusgraph by JanusGraph.

the class JanusGraphLocalQueryOptimizerStrategy method inspectLocalTraversals.

private void inspectLocalTraversals(final Admin<?, ?> traversal) {
    TraversalHelper.getStepsOfClass(LocalStep.class, traversal).forEach(localStep -> {
        final Admin localTraversal = ((LocalStep<?, ?>) localStep).getLocalChildren().get(0);
        final Step localStart = localTraversal.getStartStep();
        if (localStart instanceof VertexStep) {
            final JanusGraphVertexStep vertexStep = new JanusGraphVertexStep((VertexStep) localStart);
            TraversalHelper.replaceStep(localStart, vertexStep, localTraversal);
            if (JanusGraphTraversalUtil.isEdgeReturnStep(vertexStep)) {
                HasStepFolder.foldInHasContainer(vertexStep, localTraversal, traversal);
                HasStepFolder.foldInOrder(vertexStep, vertexStep.getNextStep(), localTraversal, traversal, false, null);
            }
            HasStepFolder.foldInRange(vertexStep, JanusGraphTraversalUtil.getNextNonIdentityStep(vertexStep), localTraversal, null);
            unfoldLocalTraversal(traversal, localStep, localTraversal, vertexStep);
        }
        if (localStart instanceof PropertiesStep) {
            final JanusGraphPropertiesStep propertiesStep = new JanusGraphPropertiesStep((PropertiesStep) localStart);
            TraversalHelper.replaceStep(localStart, propertiesStep, localTraversal);
            if (propertiesStep.getReturnType().forProperties()) {
                HasStepFolder.foldInHasContainer(propertiesStep, localTraversal, traversal);
                HasStepFolder.foldInOrder(propertiesStep, propertiesStep.getNextStep(), localTraversal, traversal, false, null);
            }
            HasStepFolder.foldInRange(propertiesStep, JanusGraphTraversalUtil.getNextNonIdentityStep(propertiesStep), localTraversal, null);
            unfoldLocalTraversal(traversal, localStep, localTraversal, propertiesStep);
        }
    });
}