toystore 0.5

data/lib/toy/extensions/date.rb

@@ -0,0 +1,24 @@
+module Toy
+  module Extensions
+    module Date
+      def to_store(value, *)
+        if value.nil? || value == ''
+          nil
+        else
+          date = value.is_a?(::Date) || value.is_a?(::Time) ? value : ::Date.parse(value.to_s)
+          ::Time.utc(date.year, date.month, date.day)
+        end
+      rescue
+        nil
+      end
+
+      def from_store(value, *)
+        value.to_date if value.present?
+      end
+    end
+  end
+end
+
+class Date
+  extend Toy::Extensions::Date
+end

data/lib/toy/extensions/float.rb

@@ -0,0 +1,13 @@
+module Toy
+  module Extensions
+    module Float
+      def to_store(value, *)
+        value.nil? ? nil : value.to_f
+      end
+    end
+  end
+end
+
+class Float
+  extend Toy::Extensions::Float
+end

data/lib/toy/extensions/hash.rb

@@ -0,0 +1,17 @@
+module Toy
+  module Extensions
+    module Hash
+      def store_default
+        {}.with_indifferent_access
+      end
+
+      def from_store(value, *)
+        value.nil? ? store_default : value.with_indifferent_access
+      end
+    end
+  end
+end
+
+class Hash
+  extend Toy::Extensions::Hash
+end

data/lib/toy/extensions/integer.rb

@@ -0,0 +1,22 @@
+module Toy
+  module Extensions
+    module Integer
+      def to_store(value, *)
+        value_to_i = value.to_i
+        if value_to_i == 0 && value != value_to_i
+          value.to_s =~ /^(0x|0b)?0+/ ? 0 : nil
+        else
+          value_to_i
+        end
+      end
+
+      def from_store(value, *)
+        to_store(value)
+      end
+    end
+  end
+end
+
+class Integer
+  extend Toy::Extensions::Integer
+end

data/lib/toy/extensions/nil_class.rb

@@ -0,0 +1,17 @@
+module Toy
+  module Extensions
+    module NilClass
+      def to_store(value, *)
+        value
+      end
+
+      def from_store(value, *)
+        value
+      end
+    end
+  end
+end
+
+class NilClass
+  include Toy::Extensions::NilClass
+end

data/lib/toy/extensions/object.rb

@@ -0,0 +1,26 @@
+module Toy
+  module Extensions
+    module Object
+      module ClassMethods
+        def to_store(value, *)
+          value
+        end
+
+        def from_store(value, *)
+          value
+        end
+      end
+
+      module InstanceMethods
+        def to_store
+          self.class.to_store(self)
+        end
+      end
+    end
+  end
+end
+
+class Object
+  extend Toy::Extensions::Object::ClassMethods
+  include Toy::Extensions::Object::InstanceMethods
+end

data/lib/toy/extensions/set.rb

@@ -0,0 +1,23 @@
+require 'set'
+
+module Toy
+  module Extensions
+    module Set
+      def store_default
+        [].to_set
+      end
+
+      def to_store(value, *)
+        value.to_a
+      end
+
+      def from_store(value, *)
+        value.nil? ? store_default : value.to_set
+      end
+    end
+  end
+end
+
+class Set
+  extend Toy::Extensions::Set
+end

data/lib/toy/extensions/string.rb

@@ -0,0 +1,17 @@
+module Toy
+  module Extensions
+    module String
+      def to_store(value, *)
+        value.nil? ? nil : value.to_s
+      end
+
+      def from_store(value, *)
+        value.nil? ? nil : value.to_s
+      end
+    end
+  end
+end
+
+class String
+  extend Toy::Extensions::String
+end

data/lib/toy/extensions/time.rb

@@ -0,0 +1,29 @@
+module Toy
+  module Extensions
+    module Time
+      def to_store(value, *)
+        if value.nil? || value == ''
+          nil
+        else
+          time_class = ::Time.try(:zone).present? ? ::Time.zone : ::Time
+          time = value.is_a?(::Time) ? value : time_class.parse(value.to_s)
+          # strip milliseconds as Ruby does micro and bson does milli and rounding rounded wrong
+          at(time.to_i).utc if time
+        end
+      end
+
+      def from_store(value, *)
+        value = to_store(value)
+        if ::Time.try(:zone).present? && value.present?
+          value.in_time_zone(::Time.zone)
+        else
+          value
+        end
+      end
+    end
+  end
+end
+
+class Time
+  extend Toy::Extensions::Time
+end

data/lib/toy/identity.rb

@@ -0,0 +1,26 @@
+module Toy
+  module Identity
+    extend ActiveSupport::Concern
+
+    included do
+      key Toy.key_factory
+    end
+
+    module ClassMethods
+      def key(name_or_factory = :uuid)
+        @key_factory = case name_or_factory
+        when :uuid
+          UUIDKeyFactory.new
+        else
+          name_or_factory
+        end
+      end
+
+      def next_key(object = nil)
+        @key_factory.next_key(object).tap do |key|
+          raise "Keys may not be nil" if key.nil?
+        end
+      end
+    end
+  end
+end

data/lib/toy/identity/abstract_key_factory.rb

@@ -0,0 +1,10 @@
+module Toy
+  module Identity
+    class AbstractKeyFactory
+      def next_key(object)
+        raise NotImplementedError, "#{self.class.name}#next_key isn't implemented."
+      end
+    end
+  end
+end
+

data/lib/toy/identity/uuid_key_factory.rb

@@ -0,0 +1,9 @@
+module Toy
+  module Identity
+    class UUIDKeyFactory < AbstractKeyFactory
+      def next_key(object)
+        SimpleUUID::UUID.new.to_guid
+      end
+    end
+  end
+end

data/lib/toy/identity_map.rb

@@ -0,0 +1,109 @@
+module Toy
+  def self.identity_map
+    Thread.current[:toystore_identity_map] ||= {}
+  end
+
+  module IdentityMap
+    extend ActiveSupport::Concern
+
+    included do
+      identity_map_on
+    end
+
+    module ClassMethods
+      def identity_map
+        Toy.identity_map
+      end
+
+      def identity_map_on?
+        @identity_map_on == true
+      end
+
+      def identity_map_off?
+        !identity_map_on?
+      end
+
+      def identity_map_on
+        @identity_map_on = true
+      end
+
+      def identity_map_off
+        @identity_map_on = false
+      end
+
+      def without_identity_map(&block)
+        begin
+          original_identity_map_on = @identity_map_on
+          identity_map_off
+          yield
+        ensure
+          @identity_map_on = original_identity_map_on
+        end
+      end
+
+      def get(id)
+        get_from_identity_map(id) || super
+      end
+
+      def get_from_identity_map(id)
+        return nil unless identity_map_on?
+        key = store_key(id)
+        if record = identity_map[key]
+          logger.debug("ToyStore IMG #{key.inspect}")
+          record
+        end
+      end
+
+      def load(attrs)
+        return nil if attrs.nil?
+
+        if instance = identity_map[store_key(attrs['id'])]
+          instance
+        else
+          super.tap { |doc| doc.add_to_identity_map }
+        end
+      end
+    end
+
+    def identity_map
+      Toy.identity_map
+    end
+
+    def save(*)
+      super.tap do |result|
+        add_to_identity_map if result
+      end
+    end
+
+    def delete(*)
+      super.tap { remove_from_identity_map }
+    end
+
+    def add_to_identity_map
+      return unless self.class.identity_map_on?
+      key = store_key
+      identity_map[key] = self
+      logger.debug("ToyStore IMS #{key.inspect}")
+    end
+
+    def remove_from_identity_map
+      return unless self.class.identity_map_on?
+      key = store_key
+      identity_map.delete(key)
+      logger.debug("ToyStore IMD #{key.inspect}")
+    end
+
+    private
+      def has_embedded_objects?
+        self.class.embedded_lists.any?
+      end
+
+      def each_embedded_object(&block)
+        if has_embedded_objects?
+          self.class.embedded_lists.keys.inject([]) do |objects, name|
+            objects.concat(send(name).to_a.compact)
+          end.each { |object| block.call(object) }
+        end
+      end
+  end
+end

data/lib/toy/index.rb

@@ -0,0 +1,74 @@
+module Toy
+  class Index
+    attr_accessor :model, :name
+
+    def initialize(model, name)
+      @model, @name = model, name.to_sym
+      raise(ArgumentError, "No attribute #{name} for index") unless model.attribute?(name)
+
+      model.indices[name] = self
+      model.send(:include, IndexCallbacks)
+      create_finders
+    end
+
+    def eql?(other)
+      self.class.eql?(other.class) &&
+        model == other.model &&
+        name  == other.name
+    end
+    alias :== :eql?
+
+    def key(value)
+      sha_value = Digest::SHA1.hexdigest(Array.wrap(value).sort.join('')) # sorted for predictability
+      [model.name, name, sha_value].join(':')
+    end
+
+    module IndexCallbacks
+      extend ActiveSupport::Concern
+
+      included do
+        after_create  :index_create
+        after_update  :index_update
+        after_destroy :index_destroy
+      end
+
+      def index_create
+        indices.each_key do |name|
+          create_index(name, send(name), id)
+        end
+      end
+
+      def index_update
+        indices.each_key do |name|
+          if send(:"#{name}_changed?")
+            destroy_index(name, send(:"#{name}_was"), id)
+            create_index(name, send(name), id)
+          end
+        end
+      end
+
+      def index_destroy
+        indices.each_key do |name|
+          destroy_index(name, send(name), id)
+        end
+      end
+    end
+
+    private
+      def create_finders
+        model.class_eval """
+          def self.first_by_#{name}(value)
+            get(get_index(:#{name}, value)[0])
+          end
+
+          def self.first_or_new_by_#{name}(value)
+            first_by_#{name}(value) || new(:#{name} => value)
+          end
+
+          def self.first_or_create_by_#{name}(value)
+            first_by_#{name}(value) || create(:#{name} => value)
+          end
+        """
+      end
+  end
+end

data/lib/toy/indices.rb

@@ -0,0 +1,56 @@
+module Toy
+  module Indices
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+      def indices
+        @indices ||= {}
+      end
+
+      def index(name)
+        Index.new(self, name)
+      end
+
+      def index_key(name, value)
+        if index = indices[name.to_sym]
+          index.key(value)
+        else
+          raise(ArgumentError, "Index for #{name} does not exist")
+        end
+      end
+
+      def get_index(name, value)
+        key = index_key(name, value)
+        store.read(key) || []
+      end
+
+      def create_index(name, value, id)
+        key = index_key(name, value)
+        ids = get_index(name, value)
+        ids.push(id) unless ids.include?(id)
+        store.write(key, ids)
+      end
+
+      def destroy_index(name, value, id)
+        key = index_key(name, value)
+        ids = get_index(name, value)
+        ids.delete(id)
+        store.write(key, ids)
+      end
+    end
+
+    module InstanceMethods
+      def indices
+        self.class.indices
+      end
+
+      def create_index(*args)
+        self.class.create_index(*args)
+      end
+
+      def destroy_index(*args)
+        self.class.destroy_index(*args)
+      end
+    end
+  end
+end