Contents

module Cadet
  class Node
    attr_accessor :node
    include_package "org.neo4j.graphdb"

    directions = {
      "in"   => Direction[0],
      "out"  => Direction[1],
      "both" => Direction[2]
    }

    def initialize(node)
      @node = node
    end
    def create_outgoing(to, type)
      @node.createRelationshipTo(to.node, DynamicRelationshipType.withName(type))
    end
    def add_label(label)
      @node.addLabel(DynamicLabel.label(label))
      self
    end

    def method_missing(name, *args)
      if name.to_s.end_with? "="
        set_property name.to_s.gsub(/=$/, ''), args[0]
      elsif name.to_s.end_with? "_to"
        create_outgoing args[0], name.to_s.gsub(/_to$/, '')
      end
    end

    def get_single_relationship(type, dir="both")
      @node.getSingleRelationship DynamicRelationshipType.withName(type), @directions[dir]
    end

    def get_all_relationships
      @node.getRelationships()
    end
    def get_all_relationships_of_type(type)
      @node.getRelationships(DynamicRelationshipType.withName(type))
    end
    def get_all_relationships_of_type_and_dir(type, dir)
      @node.getRelationships DynamicRelationshipType.withName(type), @directions[dir]
    end

    def set_property(property, value)
      @node.setProperty(property, value)
    end
    def set_properties(props)
      props.each do |k,v|
        set_property(k.to_s, v)
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cadet-0.0.1-java	lib/cadet/node.rb