Examples with RangeFilterPipe com.tinkerpop.pipes.filter.RangeFilterPipe used on opensource projects

Example 1 with RangeFilterPipe

use of com.tinkerpop.pipes.filter.RangeFilterPipe in project gremlin by tinkerpop.

the class GremlinFluentUtility method optimizePipelineForVertexQuery.

public static boolean optimizePipelineForVertexQuery(final GremlinPipeline pipeline, final Pipe pipe) {
    VertexQueryPipe queryPipe = null;
    for (int i = pipeline.size() - 1; i > 0; i--) {
        final Pipe temp = pipeline.get(i);
        if (temp instanceof VertexQueryPipe) {
            queryPipe = (VertexQueryPipe) temp;
            break;
        } else if (!(temp instanceof IdentityPipe))
            break;
    }
    if (null != queryPipe) {
        if (pipe instanceof EdgesVerticesPipe) {
            if (queryPipe.getResultElementClass().equals(Vertex.class))
                return false;
            queryPipe.setResultingElementClass(Vertex.class);
        } else if (pipe instanceof VerticesVerticesPipe) {
            if (queryPipe.getResultElementClass().equals(Vertex.class))
                return false;
            queryPipe.setDirection(((VerticesVerticesPipe) pipe).getDirection());
            queryPipe.setLabels(((VerticesVerticesPipe) pipe).getLabels());
            queryPipe.setBranchFactor(((VerticesVerticesPipe) pipe).getBranchFactor());
        } else if (pipe instanceof VerticesEdgesPipe) {
            if (queryPipe.getResultElementClass().equals(Vertex.class))
                return false;
            queryPipe.setResultingElementClass(Edge.class);
            queryPipe.setDirection(((VerticesEdgesPipe) pipe).getDirection());
            queryPipe.setLabels(((VerticesEdgesPipe) pipe).getLabels());
            queryPipe.setBranchFactor(((VerticesEdgesPipe) pipe).getBranchFactor());
        } else if (pipe instanceof PropertyFilterPipe) {
            if (queryPipe.getResultElementClass().equals(Vertex.class))
                return false;
            final PropertyFilterPipe temp = (PropertyFilterPipe) pipe;
            queryPipe.addHasContainer(new QueryPipe.HasContainer(temp.getKey(), temp.getPredicate(), temp.getValue()));
        } else if (pipe instanceof IntervalFilterPipe) {
            if (queryPipe.getResultElementClass().equals(Vertex.class))
                return false;
            final IntervalFilterPipe temp = (IntervalFilterPipe) pipe;
            queryPipe.addIntervalContainer(new QueryPipe.IntervalContainer(temp.getKey(), temp.getStartValue(), temp.getEndValue()));
        } else if (pipe instanceof RangeFilterPipe) {
            queryPipe.setLowRange(((RangeFilterPipe) pipe).getLowRange());
            queryPipe.setHighRange(((RangeFilterPipe) pipe).getHighRange());
        }
        pipeline.addPipe(new IdentityPipe());
        return true;
    } else {
        return false;
    }
}

Example 2 with RangeFilterPipe

use of com.tinkerpop.pipes.filter.RangeFilterPipe in project gremlin by tinkerpop.

the class GremlinFluentUtility method optimizePipelineForGraphQuery.

private static boolean optimizePipelineForGraphQuery(final GremlinPipeline pipeline, final Pipe pipe) {
    GraphQueryPipe queryPipe = null;
    for (int i = pipeline.size() - 1; i > 0; i--) {
        final Pipe temp = pipeline.get(i);
        if (temp instanceof GraphQueryPipe) {
            queryPipe = (GraphQueryPipe) temp;
            break;
        } else if (!(temp instanceof IdentityPipe))
            break;
    }
    if (null != queryPipe) {
        if (pipe instanceof PropertyFilterPipe) {
            final PropertyFilterPipe temp = (PropertyFilterPipe) pipe;
            queryPipe.addHasContainer(new QueryPipe.HasContainer(temp.getKey(), temp.getPredicate(), temp.getValue()));
        } else if (pipe instanceof IntervalFilterPipe) {
            final IntervalFilterPipe temp = (IntervalFilterPipe) pipe;
            queryPipe.addIntervalContainer(new QueryPipe.IntervalContainer(temp.getKey(), temp.getStartValue(), temp.getEndValue()));
        } else if (pipe instanceof RangeFilterPipe) {
            queryPipe.setLowRange(((RangeFilterPipe) pipe).getLowRange());
            queryPipe.setHighRange(((RangeFilterPipe) pipe).getHighRange());
        }
        pipeline.addPipe(new IdentityPipe());
        return true;
    } else {
        return false;
    }
}

Example 3 with RangeFilterPipe

use of com.tinkerpop.pipes.filter.RangeFilterPipe in project gremlin by tinkerpop.

the class GremlinFluentUtilityTest method testVertexRangeOptimization.

public void testVertexRangeOptimization() {
    Graph graph = TinkerGraphFactory.createTinkerGraph();
    GremlinPipeline pipeline = new GremlinPipeline(graph.getVertex(1)).out().out();
    //System.out.println(pipeline);
    assertEquals(pipeline.size(), 3);
    assertTrue(pipeline.get(0) instanceof GremlinStartPipe);
    assertTrue(pipeline.get(1) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(2) instanceof VertexQueryPipe);
    pipeline = new GremlinPipeline(graph.getVertex(1)).out().out().range(0, 10);
    //System.out.println(pipeline);
    assertEquals(pipeline.size(), 4);
    assertTrue(pipeline.get(0) instanceof GremlinStartPipe);
    assertTrue(pipeline.get(1) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(2) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(3) instanceof IdentityPipe);
    pipeline = new GremlinPipeline(graph.getVertex(1)).out("knows").out().range(5, 10);
    //System.out.println(pipeline);
    assertEquals(pipeline.size(), 4);
    assertTrue(pipeline.get(0) instanceof GremlinStartPipe);
    assertTrue(pipeline.get(1) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(2) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(3) instanceof IdentityPipe);
    // no optimization should occur
    pipeline = new GremlinPipeline(graph.getVertex(1)).out("knows").out().has("name", "marko").range(0, 10);
    // System.out.println(pipeline);
    assertEquals(pipeline.size(), 5);
    assertTrue(pipeline.get(0) instanceof GremlinStartPipe);
    assertTrue(pipeline.get(1) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(2) instanceof VertexQueryPipe);
    assertTrue(pipeline.get(3) instanceof PropertyFilterPipe);
    assertTrue(pipeline.get(4) instanceof RangeFilterPipe);
}