Search in sources :

Example 16 with AtlasVertex

use of org.apache.atlas.repository.graphdb.AtlasVertex in project incubator-atlas by apache.

the class GraphHelperTest method testGetOutgoingEdgesByLabel.

@Test
public void testGetOutgoingEdgesByLabel() throws Exception {
    AtlasGraph graph = TestUtils.getGraph();
    AtlasVertex v1 = graph.addVertex();
    AtlasVertex v2 = graph.addVertex();
    graph.addEdge(v1, v2, "l1");
    graph.addEdge(v1, v2, "l2");
    Iterator<AtlasEdge> iterator = GraphHelper.getInstance().getOutGoingEdgesByLabel(v1, "l1");
    assertTrue(iterator.hasNext());
    assertTrue(iterator.hasNext());
    assertNotNull(iterator.next());
    assertNull(iterator.next());
    assertFalse(iterator.hasNext());
    assertFalse(iterator.hasNext());
}
Also used : AtlasVertex(org.apache.atlas.repository.graphdb.AtlasVertex) AtlasGraph(org.apache.atlas.repository.graphdb.AtlasGraph) AtlasEdge(org.apache.atlas.repository.graphdb.AtlasEdge) Test(org.testng.annotations.Test)

Example 17 with AtlasVertex

use of org.apache.atlas.repository.graphdb.AtlasVertex in project incubator-atlas by apache.

the class GraphHelperTest method testGetVerticesForGUIDSWithDuplicates.

@Test
public void testGetVerticesForGUIDSWithDuplicates() throws Exception {
    ITypedReferenceableInstance hrDept = TestUtils.createDeptEg1(TypeSystem.getInstance());
    List<String> result = repositoryService.createEntities(hrDept).getCreatedEntities();
    String guid = result.get(0);
    Map<String, AtlasVertex> verticesForGUIDs = GraphHelper.getInstance().getVerticesForGUIDs(Arrays.asList(guid, guid));
    Assert.assertEquals(verticesForGUIDs.size(), 1);
    Assert.assertTrue(verticesForGUIDs.containsKey(guid));
}
Also used : AtlasVertex(org.apache.atlas.repository.graphdb.AtlasVertex) ITypedReferenceableInstance(org.apache.atlas.typesystem.ITypedReferenceableInstance) Test(org.testng.annotations.Test)

Example 18 with AtlasVertex

use of org.apache.atlas.repository.graphdb.AtlasVertex in project incubator-atlas by apache.

the class AbstractGraphDatabaseTest method removeVertices.

@AfterMethod
public void removeVertices() {
    for (AtlasVertex vertex : newVertices) {
        if (vertex.exists()) {
            getGraph().removeVertex(vertex);
        }
    }
    getGraph().commit();
    newVertices.clear();
}
Also used : AtlasVertex(org.apache.atlas.repository.graphdb.AtlasVertex) AfterMethod(org.testng.annotations.AfterMethod)

Example 19 with AtlasVertex

use of org.apache.atlas.repository.graphdb.AtlasVertex in project incubator-atlas by apache.

the class Titan1Graph method getVertex.

@Override
public AtlasVertex<Titan1Vertex, Titan1Edge> getVertex(String vertexId) {
    Iterator<Vertex> it = getGraph().vertices(vertexId);
    Vertex vertex = getSingleElement(it, vertexId);
    return GraphDbObjectFactory.createVertex(this, vertex);
}
Also used : Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) AtlasVertex(org.apache.atlas.repository.graphdb.AtlasVertex)

Example 20 with AtlasVertex

use of org.apache.atlas.repository.graphdb.AtlasVertex in project incubator-atlas by apache.

the class Titan1Graph method removeVertex.

@Override
public void removeVertex(AtlasVertex<Titan1Vertex, Titan1Edge> vertex) {
    Vertex wrapped = vertex.getV().getWrappedElement();
    wrapped.remove();
}
Also used : Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) AtlasVertex(org.apache.atlas.repository.graphdb.AtlasVertex)

Aggregations

AtlasVertex (org.apache.atlas.repository.graphdb.AtlasVertex)164 AtlasBaseException (org.apache.atlas.exception.AtlasBaseException)53 AtlasEdge (org.apache.atlas.repository.graphdb.AtlasEdge)26 ITypedReferenceableInstance (org.apache.atlas.typesystem.ITypedReferenceableInstance)21 ArrayList (java.util.ArrayList)20 Test (org.testng.annotations.Test)19 HashMap (java.util.HashMap)16 Id (org.apache.atlas.typesystem.persistence.Id)14 Map (java.util.Map)13 List (java.util.List)12 AtlasEntity (org.apache.atlas.model.instance.AtlasEntity)12 AtlasObjectId (org.apache.atlas.model.instance.AtlasObjectId)12 AtlasEntityType (org.apache.atlas.type.AtlasEntityType)12 AtlasException (org.apache.atlas.AtlasException)11 GraphTransaction (org.apache.atlas.annotation.GraphTransaction)11 RepositoryException (org.apache.atlas.repository.RepositoryException)11 AtlasGraphQuery (org.apache.atlas.repository.graphdb.AtlasGraphQuery)11 AtlasType (org.apache.atlas.type.AtlasType)11 HashSet (java.util.HashSet)8 AtlasEntityHeader (org.apache.atlas.model.instance.AtlasEntityHeader)8