NodeRelationshipSet.java

package org.neo4j.util;

import java.util.Set;

import org.neo4j.graphdb.Direction;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.RelationshipType;

/**
 * Convenience {@link Set} for when dealing with basic nodes and relationships
 * in contrast to {@link NodeWrapperRelationshipSet} where you're dealing with
 * wrappers around nodes.
 */
public class NodeRelationshipSet extends RelationshipSet<Node>
{
	public NodeRelationshipSet( Node node,
		RelationshipType type )
	{
		super( node, type );
	}
	
	public NodeRelationshipSet( Node node,
		RelationshipType type, Direction direction )
	{
		super( node, type, direction );
	}
	
	@Override
	protected Node getNodeFromItem( Object item )
	{
		return ( Node ) item;
	}
	
	@Override
	protected Node newObject( Node node, Relationship relationship )
	{
		return node;
	}
}