Sha256: 60b4a29ad643867e758d47b7a2fb5d6ec2db3aa4857709578f4e619a3ec82e9f

Contents?: true

Size: 1.18 KB

Versions: 3

Compression:

Stored size: 1.18 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.wrapper
      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

3 entries across 3 versions & 1 rubygems

Version Path
neo4j-1.0.0.beta.11 lib/neo4j/relationship_traverser.rb
neo4j-1.0.0.beta.10 lib/neo4j/relationship_traverser.rb
neo4j-1.0.0.beta.9 lib/neo4j/relationship_traverser.rb