hari 0.0.1 → 0.0.3

data/lib/hari/node/repository.rb

@@ -0,0 +1,24 @@
+module Hari
+  class Node < Entity
+    module Repository
+      extend ActiveSupport::Concern
+
+      module ClassMethods
+
+        def find_one(id, options = {})
+          id = "#{node_type}##{id}" unless id.to_s.include?('#')
+          super id, options
+        end
+
+        def find_many(ids, options = {})
+          ids = ids.map do |id|
+            id.to_s.include?('#') ? id : "#{node_type}##{id}"
+          end
+
+          super ids, options
+        end
+
+      end
+    end
+  end
+end

data/lib/hari/node/serialization.rb

@@ -0,0 +1,24 @@
+module Hari
+  class Node < Entity
+    module Serialization
+
+      def from_source(source)
+        return if source.blank?
+
+        hash = Yajl::Parser.parse(source)
+        klass = hash['id'].split('#').first.camelize.constantize
+
+        attrs = hash.inject({}) do |buffer, (key, value)|
+          if prop = klass.properties.find { |p| p.name == key }
+            buffer[key] = prop.desserialize(value)
+          end
+
+          buffer
+        end
+
+        klass.new attrs
+      end
+
+    end
+  end
+end

data/lib/hari/relation.rb

@@ -0,0 +1,72 @@
+require 'hari/relation/linked_list'
+require 'hari/relation/sorted_set'
+
+module Hari
+  class Relation < Entity
+
+    DIRECTIONS = %w(in out)
+
+    attr_accessor :label, :start_node_id, :end_node_id
+
+    validates :label, :start_node_id, :end_node_id, presence: true
+
+    def start_node
+      @start_node ||= Hari::Node.find(start_node_id)
+    end
+
+    def end_node
+      @end_node ||= Hari::Node.find(end_node_id)
+    end
+
+    def key(direction = nil)
+      case direction.try :to_s
+      when nil   then "#{start_node_id}:#{label}:#{end_node_id}"
+      when 'out' then "#{start_node_id}:#{label}:out"
+      when 'in'  then "#{end_node_id}:#{label}:in"
+      end
+    end
+
+    alias :generate_id :key
+
+    def self.create(label, start_node, end_node, attrs = {})
+      new(attrs).tap do |r|
+        r.label = label
+        r.start_node_id = Hari.node_key(start_node)
+        r.end_node_id   = Hari.node_key(end_node)
+        r.save
+      end
+    end
+
+    def self.use!(backend)
+      @backend = backend.kind_of?(Module) ? backend
+       : "Hari::Relation::#{backend.to_s.camelize}".constantize
+    end
+
+    def self.backend
+      @backend ||= Hari::Relation::SortedSet
+    end
+
+    def backend
+      self.class.backend
+    end
+
+    def weight(direction)
+      ::Time.now.to_f
+    end
+
+    private
+
+    def create
+      super
+      backend.create self
+      self
+    end
+
+    def delete
+      backend.delete self
+      super
+      self
+    end
+
+  end
+end

data/lib/hari/relation/linked_list.rb

@@ -0,0 +1,16 @@
+module Hari
+  class Relation < Entity
+    module LinkedList
+      extend self
+
+      def create(rel)
+        Relation::DIRECTIONS.each { |d| Hari.redis.lpush rel.key(d), rel.id }
+      end
+
+      def delete(rel)
+        Relation::DIRECTIONS.each { |d| Hari.redis.lrem rel.key(d), 1, rel.id }
+      end
+
+    end
+  end
+end

data/lib/hari/relation/sorted_set.rb

@@ -0,0 +1,16 @@
+module Hari
+  class Relation < Entity
+    module SortedSet
+      extend self
+
+      def create(rel)
+        Relation::DIRECTIONS.each { |d| Hari.redis.zadd rel.key(d), rel.weight(d), rel.id }
+      end
+
+      def delete(rel)
+        Relation::DIRECTIONS.each { |d| Hari.redis.zrem rel.key(d), rel.id }
+      end
+
+    end
+  end
+end

data/lib/hari/version.rb

@@ -1,3 +1,3 @@
 module Hari
-  VERSION = '0.0.1'
+  VERSION = '0.0.3'
 end

data/spec/hari/configuration_spec.rb

@@ -0,0 +1,5 @@
+require 'spec_helper'
+
+describe Hari::Configuration do
+
+end

data/spec/hari/entity/repository_spec.rb

@@ -3,19 +3,19 @@ require 'spec_helper'
 describe Hari::Entity::Repository do
 
   specify '.create, .find' do
-    model = TestModel.create(name: 'Ze', birth: '2012-04-20', points: '300')
+    model = TestEntity.create(name: 'Ze', birth: '2012-04-20', points: '300')
     model.id.should be
 
-    found = TestModel.find(model.id)
+    found = TestEntity.find(model.id)
     found.birth.year.should == 2012
     found.birth.month.should == 4
     found.birth.day.should == 20
     found.points.should == 300
     found.name.should == 'Ze'
 
-    model2 = TestModel.create(name: 'Jo', birth: '2009-03-21', points: '404')
+    model2 = TestEntity.create(name: 'Jo', birth: '2009-03-21', points: '404')
 
-    founds = TestModel.find(model.id, model2.id)
+    founds = TestEntity.find(model.id, model2.id)
     founds.size.should == 2
   end
 

data/spec/hari/entity/serialization_spec.rb

@@ -4,7 +4,7 @@ describe Hari::Entity::Serialization do
 
   describe '#to_json' do
     it 'serializes instance to json' do
-      model = TestModel.new(name: 'Ze',
+      model = TestEntity.new(name: 'Ze',
                             birth: Date.new(1986, 01, 23),
                             points: '200')
 
@@ -14,7 +14,7 @@ describe Hari::Entity::Serialization do
 
   describe '.from_json' do
     it 'desserializes instance from json' do
-      model = TestModel.from_json('{"name":"Ze","birth":"1986-01-23","points":200}')
+      model = TestEntity.from_json('{"name":"Ze","birth":"1986-01-23","points":200}')
       model.name.should == 'Ze'
       model.birth.should == Date.new(1986, 01, 23)
       model.points.should == 200

data/spec/hari/node/lists_spec.rb

@@ -0,0 +1,126 @@
+require 'spec_helper'
+
+describe Hari::Node::Queries::List do
+
+  let(:node) { Hari.node user: 10 }
+  subject    { node.list :friends }
+
+  before { subject.add 10, 20, 30, 40, 50, 60 }
+
+  specify '#name' do
+    subject.name.should eq(:friends)
+  end
+
+  specify '#range + #members + #list! + ' +
+          '#[i] + #[i, j], + #[i..j] + ' +
+          '#at + #index + #from + #to' do
+    node.list!(:friends).should eq %w(10 20 30 40 50 60)
+    subject.members.should eq %w(10 20 30 40 50 60)
+    subject.range.should eq %w(10 20 30 40 50 60)
+
+    subject.range(1, 3).should eq %w(20 30 40)
+    subject[1, 3].should eq %w(20 30 40)
+    subject[1..3].should eq %w(20 30 40)
+    subject[1].should eq('20')
+
+    subject.at(1).should eq('20')
+    subject.index(1).should eq('20')
+    subject.from(2).should eq %w(30 40 50 60)
+    subject.to(3).should eq %w(10 20 30 40)
+  end
+
+  specify '#include? + #member?' do
+    subject.include?('20').should be_true
+    subject.member?('20').should be_true
+    subject.include?('70').should be_false
+  end
+
+  specify '#count + #size + #length' do
+    subject.count.should eq(6)
+    subject.size.should eq(6)
+    subject.length.should eq(6)
+  end
+
+  specify '#one? + #many?' do
+    subject.should_not be_one
+    subject.should be_many
+
+    list = node.list(:enemies)
+    list.add 90
+    list.should be_one
+    list.should_not be_many
+  end
+
+  specify '#empty?' do
+    subject.should_not be_empty
+    node.list(:wtfs).should be_empty
+  end
+
+  specify '#delete' do
+    subject.delete 30
+    subject.members.should eq %w(10 20 40 50 60)
+  end
+
+  specify '#<<' do
+    subject << 90
+    subject.include?('90').should be_true
+    subject.pop.should eq('90')
+  end
+
+  specify '#pop + #rpop + #shift + #lpop' do
+    subject.pop.should eq('60')
+    subject.count.should eq(5)
+    subject.include?('60').should be_false
+
+    subject.rpop.should eq('50')
+    subject.count.should eq(4)
+    subject.include?('50').should be_false
+
+    subject.lpop.should eq('10')
+    subject.count.should eq(3)
+    subject.include?('10').should be_false
+
+    subject.shift.should eq('20')
+    subject.count.should eq(2)
+    subject.include?('20').should be_false
+  end
+
+  specify '#<< + #push + #rpush + #add + #lpush + #first + #last' do
+    subject << 80
+    subject.last.should eq('80')
+
+    subject.push 90
+    subject.last.should eq('90')
+
+    subject.rpush 100
+    subject.last.should eq('100')
+
+    subject.add 200
+    subject.last.should eq('200')
+
+    subject.lpush 300
+    subject.first.should eq('300')
+  end
+
+  specify '#[]=' do
+    subject[1] = 77
+    subject.members.should eq %w(10 77 30 40 50 60)
+  end
+
+  specify '#trim' do
+    subject.trim 2, 4
+    subject.members.should eq %w(30 40 50)
+  end
+
+  specify '#insert + #insert_before + #insert_after' do
+    subject.insert '20', '55'
+    subject.members.should eq %w(10 20 55 30 40 50 60)
+
+    subject.insert_after '40', '66'
+    subject.members.should eq %w(10 20 55 30 40 66 50 60)
+
+    subject.insert_before '66', '99'
+    subject.members.should eq %w(10 20 55 30 40 99 66 50 60)
+  end
+
+end

data/spec/hari/node/sets_spec.rb

@@ -0,0 +1,85 @@
+require 'spec_helper'
+
+describe Hari::Node::Queries::Set do
+
+  let(:node) { Hari.node user: 10 }
+  subject    { node.set :friends }
+
+  before { subject.add 10, 20, 30 }
+
+  specify '#name' do
+    subject.name.should eq(:friends)
+  end
+
+  specify '#members + #set!' do
+    node.set!(:friends).should eq %w(10 20 30)
+    subject.members.should eq %w(10 20 30)
+  end
+
+  specify '#include? + #member?' do
+    subject.include?(20).should be_true
+    subject.member?(20).should be_true
+    subject.include?(40).should be_false
+  end
+
+  specify '#count + #size + #length' do
+    subject.count.should eq(3)
+    subject.size.should eq(3)
+    subject.length.should eq(3)
+  end
+
+  specify '#one? + #many?' do
+    subject.should_not be_one
+    subject.should be_many
+
+    set = node.set(:enemies)
+    set.add 90
+    set.should be_one
+    set.should_not be_many
+  end
+
+  specify '#empty?' do
+    subject.should_not be_empty
+    node.set(:wtfs).should be_empty
+  end
+
+  specify '#delete' do
+    subject.delete 30, 40
+    subject.members.should eq %w(10 20)
+  end
+
+  specify '#<<' do
+    subject << 40
+    subject.include?(40).should be_true
+  end
+
+  specify '#rand' do
+    %w(10 20 30).include?(subject.rand.first).should be_true
+    subject.rand(3).sort.should eq %w(10 20 30)
+    subject.count.should eq(3)
+  end
+
+  specify '#intersect + #&' do
+    other_friends = Hari.node(user: 20).set(:friends)
+    other_friends.add 20, 30, 40, 50
+
+    (subject & other_friends).sort.should eq %w(20 30)
+    subject.intersect(other_friends).sort.should eq %w(20 30)
+  end
+
+  specify '#diff + #-' do
+    other_friends = Hari.node(user: 20).set(:friends)
+    other_friends.add 20, 30, 40, 50
+
+    (subject - other_friends).sort.should eq %w(10)
+    subject.diff(other_friends).sort.should eq %w(10)
+  end
+
+  specify '#pop' do
+    member = subject.pop
+    %w(10 20 30).include?(member).should be_true
+    subject.count.should eq(2)
+    subject.members.include?(member).should be_false
+  end
+
+end

data/spec/hari/node/sorted_sets_spec.rb

@@ -0,0 +1,89 @@
+require 'spec_helper'
+
+describe Hari::Node::Queries::SortedSet do
+
+  let(:node) { Hari.node user: 10 }
+  subject    { node.sorted_set :friends }
+
+  before { subject.add 10, 'john', 30, 'bill', 50, 'jack' }
+
+  specify '#name' do
+    subject.name.should eq(:friends)
+  end
+
+  specify '#range + #range_with_scores + #members + #sorted_set!' do
+    node.sorted_set!(:friends).should eq %w(john bill jack)
+    subject.members.should eq %w(john bill jack)
+    subject.range.should eq %w(john bill jack)
+
+    subject.range(0, 1).should eq %w(john bill)
+
+    subject.range_with_scores.should eq [['john', 10.0], ['bill', 30.0], ['jack', 50.0]]
+    subject.range(0, 1, with_scores: true).should eq [['john', 10.0], ['bill', 30.0]]
+
+    subject.revrange(0, 1).should eq %w(jack bill)
+    subject.revrange.should eq %w(jack bill john)
+    subject.reverse_range.should eq %w(jack bill john)
+    subject.desc_range.should eq %w(jack bill john)
+
+    subject.revrange_with_scores.should eq [['jack', 50.0], ['bill', 30.0], ['john', 10.0]]
+  end
+
+  specify '#include? + #member?' do
+    subject.include?('jack').should be_true
+    subject.member?('bill').should be_true
+    subject.include?('steve').should be_false
+  end
+
+  specify '#count + #size + #length' do
+    subject.count.should eq(3)
+    subject.size.should eq(3)
+    subject.length.should eq(3)
+  end
+
+  specify '#one? + #many?' do
+    subject.should_not be_one
+    subject.should be_many
+
+    zset = node.sorted_set(:enemies)
+    zset.add 70, 'jill'
+    zset.should be_one
+    zset.should_not be_many
+  end
+
+  specify '#empty?' do
+    subject.should_not be_empty
+    node.sorted_set(:wtfs).should be_empty
+  end
+
+  specify '#delete' do
+    subject.delete 'jack', 'bill', 'jill'
+    subject.members.should eq %w(john)
+  end
+
+  specify '#score' do
+    subject.score('john').should eq(10.0)
+    subject.score('steve').should_not be
+  end
+
+  specify '#rank + #revrank' do
+    subject.rank('john').should eq(0)
+    subject.ranking('bill').should eq(1)
+    subject.position('jack').should eq(2)
+
+    subject.revrank('john').should eq(2)
+    subject.reverse_ranking('bill').should eq(1)
+    subject.reverse_position('jack').should eq(0)
+  end
+
+  specify '#trim_by_rank' do
+    subject.trim_by_rank 1, 1
+    subject.members.should eq %w(john jack)
+  end
+
+  specify '#trim_by_score' do
+    subject.trim_by_score 5, 15
+    subject.members.should eq %w(bill jack)
+  end
+
+end