Java GremlinPipeline类代码示例

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
public GremlinPipeline<?, Instruction> exits()
{
	final int MAX_LOOPS = 10000;
	final String[] EDGES = {Traversals.INSTR_CFLOW_EDGE, Traversals.INSTR_CFLOW_TRANSITIVE_EDGE};
	if (!this.getVertices(Direction.OUT, EDGES).iterator().hasNext())
	{
		return new GremlinPipeline<>(this);
	} else
	{
		return new GremlinPipeline<>(this.getBaseVertex()).as("start")
				.out(EDGES).dedup().loop("start",
						arg -> arg.getLoops() < MAX_LOOPS,
						arg -> arg.getLoops() < MAX_LOOPS
								&& !arg.getObject().getEdges(Direction.OUT, EDGES).iterator().hasNext())
				.dedup()
				.transform(Instruction::new);
	}
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
public static Set<ReachingDefinitionAnalyser.Definition> killedDefinitions(
		final Vertex vertex) {
	Set<ReachingDefinitionAnalyser.Definition> killSet = new HashSet<>();
	GremlinPipeline<Vertex, Edge> pipe = new GremlinPipeline<>();
	pipe.start(vertex)
	    .out("WRITE")
	    .out("BELONGS_TO")
	    .in("BELONGS_TO")
	    .inE("WRITE");
	for (Edge writeEdge : pipe) {
		Vertex genVertex = writeEdge.getVertex(Direction.OUT);
		Vertex aloc = writeEdge.getVertex(Direction.IN);
		GremlinPipeline<Vertex, Object> pipe2 = new
				GremlinPipeline<>();
		pipe2.start(aloc)
		     .out("BELONGS_TO")
		     .in("BELONGS_TO")
		     .property("name");
		for (Object identifier : pipe2) {
			killSet.add(new ReachingDefinitionAnalyser.Definition(
					genVertex, identifier));
		}
	}
	return killSet;
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Override
protected Pipe getQueryPipe() {
    GremlinPipeline p;
    if (guid.equals("*")) {
        p = new GremlinPipeline().has(Constants.ENTITY_TEXT_PROPERTY_KEY).
                hasNot(Constants.ENTITY_TYPE_PROPERTY_KEY, "Taxonomy").outE();
    } else {
        p = new GremlinPipeline().has(Constants.GUID_PROPERTY_KEY, guid).outE();
    }
    //todo: this is basically the same pipeline used in TagRelation.asPipe()
    p.add(new FilterFunctionPipe<>(new PipeFunction<Edge, Boolean>() {
        @Override
        public Boolean compute(Edge edge) {
            String type = edge.getVertex(Direction.OUT).getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY);
            VertexWrapper v = new TermVertexWrapper(edge.getVertex(Direction.IN));
            return edge.getLabel().startsWith(type) && v.getPropertyKeys().contains("available_as_tag");
        }
    }));

    return p.inV();
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
public TestAtlasEntityQuery(QueryExpression queryExpression,
                            ResourceDefinition resourceDefinition,
                            Request request,
                            GremlinPipeline initialPipeline,
                            Pipe queryPipe,
                            Pipe notDeletedPipe,
                            AtlasGraph graph,
                            VertexWrapper vWrapper) {

    super(queryExpression, resourceDefinition, request);
    this.initialPipeline = initialPipeline;
    this.queryPipe = queryPipe;
    this.notDeletedPipe = notDeletedPipe;
    this.graph = graph;
    this.vWrapper = vWrapper;
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testBulkLoading() throws Exception {
    bulkLoadGraphOfTheGods(f1);
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(17, new GremlinPipeline(g).E().count());
    new GremlinPipeline(g).V().sideEffect(new PipeFunction<Vertex, Vertex>() {
        @Override
        public Vertex compute(Vertex vertex) {
            assertEquals(2, vertex.getPropertyKeys().size());
            assertNotNull(vertex.getProperty("name"));
            return vertex;
        }
    }).iterate();
    assertEquals("saturn", new GremlinPipeline(g).V("name", "hercules").out("father").out("father").property("name").next());
    List names = new GremlinPipeline(g).V("name", "hercules").outE("battled").sideEffect(new PipeFunction<Edge, Edge>() {
        @Override
        public Edge compute(Edge edge) {
            assertNotNull(edge.getProperty("time"));
            return edge;
        }
    }).inV().property("name").toList();
    assertTrue(names.contains("nemean"));
    assertTrue(names.contains("hydra"));
    assertTrue(names.contains("cerberus"));
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testBulkElementDeletions() throws Exception {
    bulkLoadGraphOfTheGods(f1);
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(17, new GremlinPipeline(g).E().count());
    new HadoopPipeline(f2).V().drop().submit();
    clopen();
    assertEquals(0, new GremlinPipeline(g).V().count());
    assertEquals(0, new GremlinPipeline(g).E().count());
    bulkLoadGraphOfTheGods(f1);
    new HadoopPipeline(f2).E().drop().submit();
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(0, new GremlinPipeline(g).E().count());
    new HadoopPipeline(f2).V().drop().submit();
    clopen();
    assertEquals(0, new GremlinPipeline(g).V().count());
    assertEquals(0, new GremlinPipeline(g).E().count());
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testFewElementDeletions() throws Exception {
    bulkLoadGraphOfTheGods(f1);
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(17, new GremlinPipeline(g).E().count());

    new HadoopPipeline(f2).E().has("label", "battled").drop().submit();
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(14, new GremlinPipeline(g).E().count());
    assertEquals(0, new GremlinPipeline(g).E().has("label", "battled").count());
    assertEquals(1, new GremlinPipeline(g).E().has("label", "mother").count());
    assertEquals(2, new GremlinPipeline(g).E().has("label", "father").count());

    new HadoopPipeline(f2).V().has("name", "hercules").drop().submit();
    clopen();
    assertEquals(11, new GremlinPipeline(g).V().count());
    assertEquals(12, new GremlinPipeline(g).E().count());
    assertEquals(0, new GremlinPipeline(g).E().has("label", "battled").count());
    assertEquals(0, new GremlinPipeline(g).E().has("label", "mother").count());
    assertEquals(1, new GremlinPipeline(g).E().has("label", "father").count());
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testBulkVertexPropertyDeletions() throws Exception {
    bulkLoadGraphOfTheGods(f1);
    new HadoopPipeline(f2).V().sideEffect("{it.removeProperty('name')}").submit();
    clopen();
    assertEquals(12, new GremlinPipeline(g).V().count());
    assertEquals(17, new GremlinPipeline(g).E().count());

    for (Vertex v : g.getVertices()) {
        assertNull(v.getProperty("name"));
        assertEquals(1, v.getPropertyKeys().size());
    }
    new GremlinPipeline(g).V("name", "hercules").outE("battled").sideEffect(new PipeFunction<Edge, Edge>() {
        @Override
        public Edge compute(Edge edge) {
            assertNotNull(edge.getProperty("time"));
            return edge;
        }
    }).iterate();
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
    public void testUnidirectionEdges() throws Exception {
        // Declare schema in Titan
        TitanManagement mgmt = g.getManagementSystem();
        mgmt.makeEdgeLabel("father").unidirected().make();
        mgmt.commit();

//        // Reload schema from Titan into Faunus's Type Manager
//        FaunusTypeManager.getTypeManager(null).clear();
//        SchemaProvider titanSchemaProvider = new SchemaContainer(g);
//        FaunusTypeManager typeManager = FaunusTypeManager.getTypeManager(null); //argument is ignored
//        typeManager.setSchemaProvider(titanSchemaProvider);

        bulkLoadGraphOfTheGods(f1);
        clopen();
        assertEquals(12, new GremlinPipeline(g).V().count());
        assertEquals(17, new GremlinPipeline(g).E().count());
        assertEquals(new GremlinPipeline(g).V("name", "hercules").out("father").count(), 1);
        assertEquals(new GremlinPipeline(g).V("name", "jupiter").in("father").count(), 0);

//        // Reset/clear types to avoid interference with subsequent tests
//        typeManager.clear();
//        typeManager.setSchemaProvider(DefaultSchemaProvider.INSTANCE);
    }
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testApplyLayout() throws Exception {
  horizontalLayout.applyLayout( graph, 1000, 1000 );

  assertThat( verticesByProperty( "degree", 0 ), containsInAnyOrder( "S0", "S1" ) );
  assertThat( verticesByProperty( "degree", 1 ), contains( "A" ) );
  assertThat( verticesByProperty( "degree", 2 ), contains( "B" ) );
  assertThat( verticesByProperty( "degree", 3 ), containsInAnyOrder( "C", "D" ) );

  List<String> orderedX = new GremlinPipeline<>( graph ).V()
    .order( compareProperty( "x" ) )
    .id().cast( String.class )
    .toList();

  assertThat( orderedX, containsInRelativeOrder( "S1", "A", "B", "D" ) );
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
public static void main(String[] args) {
	if (args.length != 1) {
		System.err.println("Please enter a graph name.");
		System.err.println("graphs: memory-beta, simpsons, wookieepedia");
		System.err
				.println("Graph path should be set in the graphs.poperties file.");
		System.exit(1);
	}
	

	TinkerGraph graph;
	try {
		graph = SingletonGraph.getInstance().getGraphs().get(args[0]);
		
		Index<Vertex> index = graph.getIndex("verb-idx", Vertex.class);
		GremlinPipeline<Iterable<Vertex>, Vertex> pipeline = new GremlinPipeline<Iterable<Vertex>, Vertex>();

		pipeline.start(index.get("verbIndex", null)).hasNot(NerdleGraphTransformer.PROPERTY_ISSYNONYM, true);
	} catch (ConfigurationException e) {
		e.printStackTrace();
	}
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
@Test
public void testTraversalPerformance() {

	TinkerGraph t = TinkerGraphFactory.createTinkerGraph();

	FramedGraph f = new FramedGraph(t);

	Timer timer = metrics.timer("gremlin");

	Context time = timer.time();
	for (int count = 0; count < iterations; count++) {
		GremlinPipeline g = new GremlinPipeline(t);
		g.V().both().both().both().toList();
	}
	long nanoseconds = time.stop();
	System.out.println("Iterate over all GremlinPipeline " + nanoseconds / 1000000);
	time = timer.time();
	for (int count = 0; count < iterations; count++) {
		f.V().both().both().both().toList();
	}
	nanoseconds = time.stop();
	System.out.println("Iterate over all Totorom " + nanoseconds / 1000000);
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
/**
 * Calculates the weight of the comment path as defined by Q14 between a and b
 * (directional)
 *
 * @param a Source (Person) vertex
 * @param b Target (Person) vertex
 * @return weight of comment-reply path between a and b as defined by Q14
 */
private double calcWeight(final Vertex a, final Vertex b) {

    double w = 0.0;
    GremlinPipeline<Vertex, Vertex> pipe = (new GremlinPipeline<>(a));
    Iterable<Row> res1 = pipe.in("hasCreator").as("postComment").in("replyOf").out("hasCreator")
            .filter(new PipesFunction<Vertex, Boolean>() {
                @Override
                public Boolean compute(Vertex commentCreator) {
                    return commentCreator.equals(b);
                }
            }).select();

    for (Row r : res1) {
        w += (((Vertex) r.getColumn(0)).getProperty("label").equals("Post") ? 1 : 0.5);
    }
    return w;
}
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
/**
     * Given a vertex representing a person collects information about
     * the universities this person has gone to using a table.
     *
     * @param v Person vertex for which to collect universities
     * @return list of lists, internal list is
     * (uniCity:CITY)<-[:IS_LOCATED_IN]-(uni:UNIVERSITY)<-[studyAt:STUDY_AT]-(person)
     */
    public static List<List<Object>> collectUnis(Vertex v) {

        List<List<Object>> res = new ArrayList<>();
        Table table = new Table();
        GremlinPipeline<Vertex, List<Vertex>> gp = new GremlinPipeline<>(v);
//        List<Edge> studyEdges= (new GremlinPipeline<Vertex,List<Edge>>(v).outE("studyAt").aggregate()).toList();
//        List<Vertex> uniVertices = (new GremlinPipeline<Vertex,List<Vertex>>(v).outE("studyAt").outV().aggregate()).toList();
//        List<Vertex> uniCountries = (new GremlinPipeline<Vertex,List<Vertex>>(v).outE("studyAt").outV().out("isLocatedIn").aggregate()).toList();
        gp.outE("studyAt").as("e").inV().as("u").out("isLocatedIn").as("c").table(table).iterate();
        for (Row r : table) {
            ArrayList<Object> a = new ArrayList<>();
            a.add(((Vertex) r.get(1)).getProperty("name")); //University->name
            a.add(((Edge) r.get(0)).getProperty("classYear")); //studyAt->classYear
            a.add(((Vertex) r.get(2)).getProperty("name")); //City->name
            res.add(a);
        }
        return res;
    }
 

import com.tinkerpop.gremlin.java.GremlinPipeline; //导入依赖的package包/类
/**
 * Given a vertex representing a person collects information about
 * the companies this person has worked at using a table.
 *
 * @param v Person for which to collect companies
 * @return list of lists, internal list is
 * (companyCountry:PLACE:COUNTRY)<-[:IS_LOCATED_IN]-(company:COMPANY)<-[worksAt:WORKS_AT]-(person)
 */
public static List<List<Object>> collectComps(Vertex v) {
    //TODO combine with collectUnis
    List<List<Object>> res = new ArrayList<>();
    Table table = new Table();
    GremlinPipeline<Vertex, List<Vertex>> gp = new GremlinPipeline<>(v);
    gp.outE("workAt").as("e").inV().as("c").out("isLocatedIn").as("co").table(table).iterate();
    for (Row r : table) {
        ArrayList<Object> a = new ArrayList<>();
        a.add(((Vertex) r.get(1)).getProperty("name")); //Company->name
        a.add(((Edge) r.get(0)).getProperty("workFrom")); //workAt->workFrom
        a.add(((Vertex) r.get(2)).getProperty("name")); //Country->name
        res.add(a);
    }
    return res;
}