com.tinkerpop.gremlin.giraph.structure.GiraphHelper Maven / Gradle / Ivy

Go to download
package com.tinkerpop.gremlin.giraph.structure;

import com.tinkerpop.gremlin.structure.Direction;
import com.tinkerpop.gremlin.structure.Edge;
import com.tinkerpop.gremlin.structure.Vertex;
import com.tinkerpop.gremlin.tinkergraph.structure.TinkerEdge;
import com.tinkerpop.gremlin.util.StreamFactory;

import java.util.Iterator;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class GiraphHelper {

    public static Iterator getVertices(final GiraphGraph graph, final Vertex vertex, final Direction direction, final int branchFactor, final String... labels) {
        return vertex instanceof GiraphVertex ?
                StreamFactory.stream(((GiraphVertex) vertex).getBaseVertex().vertices(direction, branchFactor, labels)).map(v -> graph.v(v.id())).iterator() :
                (Iterator) vertex.vertices(direction, branchFactor, labels);
    }

    public static Iterator getEdges(final GiraphGraph graph, final Vertex vertex, final Direction direction, final int branchFactor, final String... labels) {
        return vertex instanceof GiraphVertex ?
                (Iterator) StreamFactory.stream(((GiraphVertex) vertex).getBaseVertex().edges(direction, branchFactor, labels)).map(e -> new GiraphEdge((TinkerEdge) e, graph)).iterator() :
                (Iterator) vertex.edges(direction, branchFactor, labels);

    }

    public static Iterator getVertices(final GiraphGraph graph, final Edge edge, final Direction direction) {
        return edge instanceof GiraphEdge ?
                ((GiraphEdge) edge).getBaseEdge().flatMap(e -> e.get().vertices(direction)).map(v -> graph.v(v.get().id())) :
                (Iterator) edge.toV(direction);
    }
}