Sha256: 4d4097069b4844fbd8570ce65725a6036819fad51ca0e0dc6b6f7ba63935b171
Contents?: true
Size: 1.17 KB
Versions: 8
Compression:
Stored size: 1.17 KB
Contents
module Neo4j
class RelationshipTraverser
include Enumerable
include ToJava
def initialize(node, types, dir)
@node = node
if types.size > 1
@types = types.inject([]) { |result, type| result << type_to_java(type) }.to_java(:'org.neo4j.graphdb.RelationshipType')
elsif types.size == 1
@type = type_to_java(types[0])
end
@dir = dir_to_java(dir)
end
def each
iter = iterator
while (iter.hasNext) do
yield iter.next
end
end
def iterator
if @types
@node.get_relationships(@types).iterator
elsif @type
@node.get_relationships(@type, @dir).iterator
else
@node.get_relationships(@dir).iterator
end
end
def size
[*self].size
end
def both
@dir = dir_to_java(:both)
self
end
def incoming
raise "Not allowed calling incoming when finding several relationships types" if @types
@dir = dir_to_java(:incoming)
self
end
def outgoing
raise "Not allowed calling outgoing when finding several relationships types" if @types
@dir = dir_to_java(:outgoing)
self
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems