Contents

module Neo4j

  class RelationshipTraverser
    include Enumerable
    include ToJava

    def initialize(node, types, direction)
      @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
      @direction = direction
    end

    def to_s
      if @type
        "#{self.class} [type: #{@type} dir:#{@direction}]"
      elsif @types
        "#{self.class} [types: #{@types.join(',')} dir:#{@direction}]"
      else
        "#{self.class} [types: ANY dir:#{@direction}]"
      end
    end

    def each
      iter = iterator
      while (iter.hasNext())
        rel = iter.next
        yield rel.wrapper if match_to_other?(rel)
      end
    end

    def iterator
      if @types
        @node.get_relationships(@types).iterator
      elsif @type
        @node.get_relationships(@type, dir_to_java(@direction))
      else
        @node.get_relationships(dir_to_java(@direction))
      end
    end

    def match_to_other?(rel)
      if @to_other.nil?
        true
      elsif @direction == :outgoing
        rel._end_node == @to_other
      elsif @direction == :incoming
        rel._start_node == @to_other
      else
        rel._start_node == @to_other || rel._end_node == @to_other
      end
    end

    def to_other(to_other)
      @to_other = to_other
      self
    end

    def del
      each { |rel| rel.del }
    end

    def size
      [*self].size
    end

    def both
      @direction = :both
      self
    end

    def incoming
      raise "Not allowed calling incoming when finding several relationships types" if @types
      @direction = :incoming
      self
    end

    def outgoing
      raise "Not allowed calling outgoing when finding several relationships types" if @types
      @direction = :outgoing
      self
    end

  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
neo4j-1.0.0.beta.19	lib/neo4j/relationship_traverser.rb
neo4j-1.0.0.beta.18	lib/neo4j/relationship_traverser.rb
neo4j-1.0.0.beta.17	lib/neo4j/relationship_traverser.rb
neo4j-1.0.0.beta.16	lib/neo4j/relationship_traverser.rb