@Override public Iterator<Vertex> vertices(Object... vertexIds) { List<Vertex> vertices = new ArrayList<>(); if (vertexIds.length > 0) { for (Object vertexId : vertexIds) { DuctileDBVertex baseVertex = baseGraph.getVertex(convertId(vertexId)); if (baseVertex != null) { vertices.add(new DuctileVertex(baseVertex, this)); } } } else { baseGraph.getVertices().forEach(vertex -> vertices.add(new DuctileVertex(vertex, this))); } return vertices.iterator(); }
@Override public void clear(Graph graph, Configuration configuration) throws Exception { try (DuctileDBGraph ductileGraph = ((DuctileGraph) graph).getBaseGraph()) { logger.info("Delete ductile graph..."); for (DuctileDBEdge edge : ductileGraph.getEdges()) { edge.remove(); } for (DuctileDBVertex vertex : ductileGraph.getVertices()) { vertex.remove(); } ductileGraph.commit(); logger.info("Ductile graph deleted."); } }
public static Consumer<DuctileDBGraph> assertVertexEdgeCounts(final int expectedVertexCount, final int expectedEdgeCount) { return (g) -> { assertEquals(expectedVertexCount, DuctileDBTestHelper.count(g.getVertices())); assertEquals(expectedEdgeCount, DuctileDBTestHelper.count(g.getEdges())); }; }
public static void removeGraph(DuctileDBGraph graph) throws IOException { logger.info("Delete ductile graph..."); for (DuctileDBEdge edge : graph.getEdges()) { edge.remove(); } for (DuctileDBVertex vertex : graph.getVertices()) { vertex.remove(); } graph.commit(); DuctileDBGraphManager graphManager = graph.createGraphManager(); for (String variableName : graphManager.getVariableNames()) { graphManager.removeVariable(variableName); } DuctileDBSchemaManager schemaManager = graph.createSchemaManager(); for (String typeName : schemaManager.getDefinedTypes()) { for (ElementType elementType : ElementType.values()) { schemaManager.removeTypeDefinition(elementType, typeName); } } for (String propertyName : schemaManager.getDefinedProperties()) { for (ElementType elementType : ElementType.values()) { schemaManager.removePropertyDefinition(elementType, propertyName); } } assertEquals(DuctileDBGraphImpl.class, graph.getClass()); new DuctileDBHealthCheck((DuctileDBGraphImpl) graph).runCheck(); logger.info("Ductile graph deleted."); } }