Examples with ReturnableEvaluator org.neo4j.graphdb.ReturnableEvaluator used on opensource projects

Example 1 with ReturnableEvaluator

use of org.neo4j.graphdb.ReturnableEvaluator in project graphdb by neo4j-attic.

the class TestTraversal method testDirectedBreadthTraversal.

// Traverses the test "ise-tree-like" population breadth first,
// starting in the middle of the tree and traversing only in the
// "forward" direction
@Test
public void testDirectedBreadthTraversal() throws Exception {
    // Build test population
    Node root = this.buildIseTreePopulation();
    Node startNode = null;
    // Get a node in the middle of the tree:
    try {
        // a) Construct a returnable evaluator that returns node 2
        ReturnableEvaluator returnEvaluator = new ReturnableEvaluator() {

            public boolean isReturnableNode(TraversalPosition pos) {
                try {
                    Node node = pos.currentNode();
                    String key = "node.test.id";
                    String nodeId = (String) node.getProperty(key);
                    return nodeId.equals("2");
                } catch (Exception e) {
                    return false;
                }
            }
        };
        // b) create a traverser
        Traverser toTheMiddleTraverser = root.traverse(BREADTH_FIRST, StopEvaluator.END_OF_GRAPH, returnEvaluator, MyRelTypes.TEST, Direction.BOTH);
        // c) get the first node it returns
        startNode = toTheMiddleTraverser.iterator().next();
        assertEquals("2", startNode.getProperty("node.test.id"));
    } catch (Exception e) {
        e.printStackTrace();
        fail("Something went wrong when trying to get a start node " + "in the middle of the tree: " + e);
    }
    // Construct the real traverser
    Traverser traverser = startNode.traverse(BREADTH_FIRST, StopEvaluator.END_OF_GRAPH, ReturnableEvaluator.ALL, MyRelTypes.TEST, Direction.OUTGOING);
    try {
        this.assertNextNodeId(traverser, "2");
        this.assertNextNodeId(traverser, "5");
        this.assertNextNodeId(traverser, "6");
        this.assertNextNodeId(traverser, "10");
        this.assertNextNodeId(traverser, "11");
        this.assertNextNodeId(traverser, "12");
        this.assertNextNodeId(traverser, "13");
        assertTrue("Too many nodes returned from traversal", traverser.iterator().hasNext() == false);
    } catch (java.util.NoSuchElementException nsee) {
        nsee.printStackTrace();
        fail("Too few nodes returned from traversal");
    } finally {
        this.deleteNodeTreeRecursively(root, 0);
    }
}

Example 2 with ReturnableEvaluator

use of org.neo4j.graphdb.ReturnableEvaluator in project graphdb by neo4j-attic.

the class TestTraversal method testStopOnReturnedNodes.

// Verifies that the stop evaluator can stop based on the amount of
// returned nodes
@Test
public void testStopOnReturnedNodes() throws Exception {
    // Build ise tree
    Node root = this.buildIseTreePopulation();
    // Traverse only ISE relationships
    RelationshipType[] traversableRels = new RelationshipType[] { MyRelTypes.TEST };
    // Construct stop- and returnable evaluators that return 5 nodes
    StopEvaluator stopEvaluator = new StopEvaluator() {

        public boolean isStopNode(TraversalPosition position) {
            // Stop traversing when we've returned 5 nodes
            return position.returnedNodesCount() >= 5;
        }
    };
    ReturnableEvaluator returnEvaluator = new ReturnableEvaluator() {

        public boolean isReturnableNode(TraversalPosition position) {
            // Return nodes until we've reached 5 nodes or end of graph
            return position.returnedNodesCount() < 5;
        }
    };
    // Create a traverser
    Traverser traverser = root.traverse(BREADTH_FIRST, stopEvaluator, returnEvaluator, traversableRels[0], Direction.BOTH);
    try {
        this.assertLevelsOfNodes(traverser, new String[][] { new String[] { "1" }, new String[] { "2", "3", "4" }, new String[] { "5" } });
        assertTrue("Too many nodes returned from traversal", traverser.iterator().hasNext() == false);
    } catch (java.util.NoSuchElementException nsee) {
        fail("Too few nodes returned from traversal");
    } finally {
        // Delete ise tree and commmit work
        this.deleteNodeTreeRecursively(root, 0);
    }
}

Example 3 with ReturnableEvaluator

use of org.neo4j.graphdb.ReturnableEvaluator in project graphdb by neo4j-attic.

the class CircularGraphTest method testCircularBug.

@Test
public void testCircularBug() {
    final long timestamp = 3;
    Transaction tx = beginTx();
    getNodeWithName("2").setProperty("timestamp", 1L);
    getNodeWithName("3").setProperty("timestamp", 2L);
    tx.success();
    tx.finish();
    final RelationshipType type = DynamicRelationshipType.withName("TO");
    Traverser t = referenceNode().traverse(Order.DEPTH_FIRST, new StopEvaluator() {

        public boolean isStopNode(TraversalPosition position) {
            Relationship last = position.lastRelationshipTraversed();
            if (last != null && last.isType(type)) {
                Node node = position.currentNode();
                long currentTime = (Long) node.getProperty("timestamp");
                return currentTime >= timestamp;
            }
            return false;
        }
    }, new ReturnableEvaluator() {

        public boolean isReturnableNode(TraversalPosition position) {
            Relationship last = position.lastRelationshipTraversed();
            if (last != null && last.isType(type)) {
                return true;
            }
            return false;
        }
    }, type, Direction.OUTGOING);
    Iterator<Node> nodes = t.iterator();
    assertEquals("2", nodes.next().getProperty("name"));
    assertEquals("3", nodes.next().getProperty("name"));
    assertFalse(nodes.hasNext());
}