@Test
  public void removeNodesWithARelation() {
    int originalNodeCount = countNodesOf(graphDb);
    int removedNodeCount = 0;

    Transaction tx = graphDb.beginTx();
    try {
      Set<Node> toRemove = new HashSet<Node>();
      for (Node node : graphDb.getAllNodes()) {
        for (Relationship relationship : node.getRelationships(RelationType.ON)) {
          toRemove.add(relationship.getStartNode());
          toRemove.add(relationship.getEndNode());
          relationship.delete();
        }
      }
      for (Node node : toRemove) {
        node.delete();
        removedNodeCount++;
      }
      tx.success();
    } finally {
      tx.finish();
    }

    int finalNodeCount = countNodesOf(graphDb);
    assertEquals(
        Integer.valueOf(originalNodeCount), Integer.valueOf(finalNodeCount + removedNodeCount));
  }
 private int countNodesOf(GraphDatabaseService aGraphDb) {
   int nodeCount = 0;
   for (@SuppressWarnings("unused") Node node : aGraphDb.getAllNodes()) {
     nodeCount++;
   }
   return nodeCount;
 }
Exemplo n.º 3
0
 public static void destroy(Map<String, Node> nodes) {
   if (nodes.isEmpty()) return;
   GraphDatabaseService db = nodes.values().iterator().next().getGraphDatabase();
   Transaction tx = db.beginTx();
   try {
     for (Node node : db.getAllNodes()) {
       for (Relationship rel : node.getRelationships()) rel.delete();
       node.delete();
     }
     tx.success();
   } finally {
     tx.finish();
   }
 }
  public void testNeo4jRaw() throws Exception {
    double totalTime = 0.0d;
    Graph graph = graphTest.getGraphInstance();
    GraphMLReader.inputGraph(graph, GraphMLReader.class.getResourceAsStream("graph-example-2.xml"));
    graph.shutdown();

    for (int i = 0; i < TOTAL_RUNS; i++) {
      graph = graphTest.getGraphInstance();
      GraphDatabaseService neo4j = ((Neo4jGraph) graph).getRawGraph();
      int counter = 0;
      this.stopWatch();
      for (final Node node : neo4j.getAllNodes()) {
        counter++;
        for (final Relationship relationship : node.getRelationships(Direction.OUTGOING)) {
          counter++;
          final Node node2 = relationship.getEndNode();
          counter++;
          for (final Relationship relationship2 : node2.getRelationships(Direction.OUTGOING)) {
            counter++;
            final Node node3 = relationship2.getEndNode();
            counter++;
            for (final Relationship relationship3 : node3.getRelationships(Direction.OUTGOING)) {
              counter++;
              relationship3.getEndNode();
              counter++;
            }
          }
        }
      }
      double currentTime = this.stopWatch();
      totalTime = totalTime + currentTime;
      BaseTest.printPerformance(
          neo4j.toString(), counter, "Neo4j raw elements touched", currentTime);
      graph.shutdown();
    }
    BaseTest.printPerformance(
        "Neo4jRaw", 1, "Neo4j Raw experiment average", totalTime / (double) TOTAL_RUNS);
  }
  @Override
  public void write(GraphDatabaseService transNeo) {

    BufferedWriter bufferedWriter = null;

    Transaction tx = transNeo.beginTx();

    try {
      bufferedWriter = new BufferedWriter(new FileWriter(chacoFile));

      long nodeCount = 0;
      long edgeCount = 0;
      for (Node node : transNeo.getAllNodes()) {

        nodeCount++;

        // Only consider OUTGOING so edges counted once
        for (Relationship rel : node.getRelationships(Direction.OUTGOING)) {
          edgeCount++;
        }
      }

      String firstLine = String.format("%d %d 0", nodeCount, edgeCount * 2);
      bufferedWriter.write(firstLine);

      int flushBuffer = 0;

      for (Node node : transNeo.getAllNodes()) {

        flushBuffer++;

        bufferedWriter.newLine();

        // Chaco files assumed to be undirected. Edges are bidirectional
        for (Relationship rel : node.getRelationships(Direction.BOTH)) {

          Long gId = (Long) rel.getOtherNode(node).getProperty(Consts.NODE_GID);
          bufferedWriter.write(" " + gId.toString());
        }

        // Temporary flush to reduce memory consumption
        if (flushBuffer % Consts.STORE_BUF == 0) {
          bufferedWriter.flush();
        }
      }

    } catch (FileNotFoundException ex) {
      ex.printStackTrace();
    } catch (IOException ex) {
      ex.printStackTrace();
    } finally {

      tx.finish();

      // Close the BufferedWriter
      try {
        if (bufferedWriter != null) {
          bufferedWriter.flush();
          bufferedWriter.close();
        }
      } catch (IOException ex) {
        ex.printStackTrace();
      }
    }
  }