mongoid_identity_map 0.0.1

data/.gitignore

@@ -0,0 +1,6 @@
+Gemfile.lock
+pkg/*
+*.gem
+.bundle
+.idea
+.rspec

data/.rvmrc

@@ -0,0 +1,2 @@
+rvm_gemset_create_on_use_flag=1
+rvm use 1.9.2@mongoid_identity_map

data/Gemfile

@@ -0,0 +1,4 @@
+source :gemcutter
+
+# Specify your gem's dependencies in mongoid_identity_map.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,5 @@
+require 'bundler'
+require 'rspec/core/rake_task'
+
+Bundler::GemHelper.install_tasks
+RSpec::Core::RakeTask.new(:spec)

data/lib/mongoid_identity_map/clear_middleware.rb

@@ -0,0 +1,13 @@
+module MongoidIdentityMap
+  class ClearMiddleware
+    def initialize(app)
+      @app = app
+    end
+    
+    def call(env)
+      res = @app.call(env)
+      CurrentThreadHash.clear
+      res
+    end
+  end
+end

data/lib/mongoid_identity_map/current_thread_hash.rb

@@ -0,0 +1,23 @@
+module MongoidIdentityMap
+  class CurrentThreadHash
+    class << self
+      def get(key)
+        current_thread_hash[key]
+      end
+      
+      def set(key, value)
+        current_thread_hash[key] = value
+      end
+      
+      def clear
+        Thread.current[:mongoid_identity_map_current_thread_hash] = nil
+      end
+      
+      private
+      
+      def current_thread_hash
+        Thread.current[:mongoid_identity_map_current_thread_hash] ||= Hash.new
+      end
+    end
+  end
+end

data/lib/mongoid_identity_map/identity_map.rb

@@ -0,0 +1,11 @@
+module MongoidIdentityMap
+  class IdentityMap
+
+    class << self
+      def fetch(selector)
+        CurrentThreadHash.get(selector) || CurrentThreadHash.set(selector, yield)
+      end
+    end
+    
+  end
+end

data/lib/mongoid_identity_map/identity_mappable.rb

@@ -0,0 +1,20 @@
+require 'active_support/core_ext'
+require 'mongoid/collection'
+
+module MongoidIdentityMap
+  module IdentityMappable
+    extend ActiveSupport::Concern
+
+    included do
+      alias_method_chain :find_one, :identity_map
+    end
+
+    def find_one_with_identity_map(selector = {}, options = {})
+      IdentityMap.fetch(selector.merge(:_collection_name => self.name)) do
+        find_one_without_identity_map(selector, options)
+      end
+    end
+  end
+end
+
+Mongoid::Collection.send(:include, MongoidIdentityMap::IdentityMappable)

data/lib/mongoid_identity_map/version.rb

@@ -0,0 +1,3 @@
+module MongoidIdentityMap
+  VERSION = "0.0.1"
+end

data/lib/mongoid_identity_map.rb

@@ -0,0 +1,5 @@
+require "mongoid_identity_map/clear_middleware"
+require "mongoid_identity_map/current_thread_hash"
+require "mongoid_identity_map/identity_map"
+require "mongoid_identity_map/identity_mappable"
+

data/mongoid_identity_map.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path("../lib/mongoid_identity_map/version", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "mongoid_identity_map"
+  s.version     = MongoidIdentityMap::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Guilherme Cirne"]
+  s.email       = ["gcirne@gmail.com"]
+  s.homepage    = "http://rubygems.org/gems/mongoid_identity_map"
+  s.summary     = "IdentityMap for Mongoid"
+  s.description = "Simple and transparent IdentityMap implementation for Mongoid"
+
+  s.required_rubygems_version = ">= 1.3.6"
+  s.rubyforge_project         = "mongoid_identity_map"
+
+  s.add_dependency "activesupport", "~> 3.0.0"
+  s.add_dependency "mongoid", "~> 2.0.0"
+
+  s.add_development_dependency "bson_ext", ">= 1.3.0"
+  s.add_development_dependency "bundler", ">= 1.0.0"
+  s.add_development_dependency "rspec", ">= 2.5.0"
+
+  s.files        = `git ls-files`.split("\n")
+  s.executables  = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
+  s.require_path = 'lib'
+end

data/spec/mongoid_identity_map/clear_middleware_spec.rb

@@ -0,0 +1,20 @@
+require "spec_helper"
+
+describe MongoidIdentityMap::ClearMiddleware do
+  before do
+    @env = {}
+    @resp = stub("resp")
+    @app = stub("app")
+    @app.stub!(:call).with(@env).and_return(@resp)
+    @clear_middleware = MongoidIdentityMap::ClearMiddleware.new(@app)
+  end
+  
+  it "should return app response" do
+    @clear_middleware.call(@env).should be(@resp)
+  end
+  
+  it "should clear identity map" do
+    MongoidIdentityMap::CurrentThreadHash.should_receive(:clear)
+    @clear_middleware.call(@env)
+  end
+end

data/spec/mongoid_identity_map/current_thread_hash_spec.rb

@@ -0,0 +1,30 @@
+require "spec_helper"
+
+describe MongoidIdentityMap::CurrentThreadHash do
+  describe ".set and .get" do
+    it "should set and get key based value" do
+      MongoidIdentityMap::CurrentThreadHash.set(:key, "value")
+      
+      Thread.new do
+        MongoidIdentityMap::CurrentThreadHash.set(:key, "value2")
+        MongoidIdentityMap::CurrentThreadHash.get(:key).should == "value2"
+      end.join
+      
+      MongoidIdentityMap::CurrentThreadHash.get(:key).should == "value"
+    end
+  end
+  
+  describe ".clear" do
+    it "should clear" do
+      MongoidIdentityMap::CurrentThreadHash.set(:key, "value")
+      
+      Thread.new do
+        MongoidIdentityMap::CurrentThreadHash.set(:key, "value")
+        MongoidIdentityMap::CurrentThreadHash.clear
+        MongoidIdentityMap::CurrentThreadHash.get(:key).should be_nil
+      end.join
+      
+      MongoidIdentityMap::CurrentThreadHash.get(:key).should == "value"
+    end
+  end
+end

data/spec/mongoid_identity_map/identity_map_spec.rb

@@ -0,0 +1,48 @@
+require "spec_helper"
+
+describe MongoidIdentityMap::IdentityMap do
+  before do
+    @model = Model.new
+    @selector = {:_id => BSON::ObjectId("4da13240312f916cd2000002")}
+  end
+  
+  describe ".fetch" do
+    context "when document doesn't exist in identity map" do
+      before do
+        MongoidIdentityMap::CurrentThreadHash.stub!(:get).with(@selector).and_return(nil)
+        Model.collection.stub!(:find_one_without_identity_map).with(@selector).and_return(@model)
+      end
+
+      it "should return document from database" do
+        Model.collection.should_receive(:find_one_without_identity_map).with(@selector).and_return(@model)
+        fetch.should be(@model)
+      end
+
+      it "should set document in identity map" do
+        MongoidIdentityMap::CurrentThreadHash.stub!(:set).with(@selector, @model)
+        fetch
+      end
+    end
+
+    context "when document exists in identity map" do
+      before do
+        MongoidIdentityMap::CurrentThreadHash.stub!(:get).with(@selector).and_return(@model)
+      end
+
+      it "should return document from identity map" do
+        fetch.should be(@model)
+      end
+
+      it "should not hit database" do
+        Model.collection.should_not_receive(:find_one_without_identity_map)
+        fetch
+      end
+    end
+
+    def fetch
+      MongoidIdentityMap::IdentityMap.fetch(@selector) do
+        Model.collection.find_one_without_identity_map(@selector)
+      end
+    end
+  end
+end

data/spec/mongoid_identity_map/identity_mappable_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+
+describe MongoidIdentityMap::IdentityMappable do
+  describe "#find_one" do
+    before do
+      @model = Model.new
+      Model.collection.stub!(:find_one_without_identity_map).with({:_id => BSON::ObjectId("4da13240312f916cd2000002")}, {}).and_return(@model)
+    end
+
+    it "should return model" do
+      Model.collection.find_one({:_id => BSON::ObjectId("4da13240312f916cd2000002")}, {}).should be(@model)
+    end
+    
+    it "should include collection name in identity map fetch selector" do
+      MongoidIdentityMap::IdentityMap.should_receive(:fetch).with(hash_including(:_collection_name => "models"))
+      Model.collection.find_one({:_id => BSON::ObjectId("4da13240312f916cd2000002")}, {})
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,12 @@
+require "mongoid_identity_map"
+require "mongoid"
+require "support/model"
+
+RSpec.configure do |config|
+  config.filter_run :focus => true
+  config.run_all_when_everything_filtered = true
+  
+  config.before(:each) do
+    MongoidIdentityMap::CurrentThreadHash.clear
+  end
+end

data/spec/support/model.rb

@@ -0,0 +1,3 @@
+class Model
+  include Mongoid::Document
+end

metadata

@@ -0,0 +1,157 @@
+--- !ruby/object:Gem::Specification 
+name: mongoid_identity_map
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Guilherme Cirne
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-29 00:00:00 -03:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 3
+        - 0
+        - 0
+        version: 3.0.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: mongoid
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 0
+        - 0
+        version: 2.0.0
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: bson_ext
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 3
+        - 0
+        version: 1.3.0
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 2
+        - 5
+        - 0
+        version: 2.5.0
+  type: :development
+  version_requirements: *id005
+description: Simple and transparent IdentityMap implementation for Mongoid
+email: 
+- gcirne@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .rvmrc
+- Gemfile
+- Rakefile
+- lib/mongoid_identity_map.rb
+- lib/mongoid_identity_map/clear_middleware.rb
+- lib/mongoid_identity_map/current_thread_hash.rb
+- lib/mongoid_identity_map/identity_map.rb
+- lib/mongoid_identity_map/identity_mappable.rb
+- lib/mongoid_identity_map/version.rb
+- mongoid_identity_map.gemspec
+- spec/mongoid_identity_map/clear_middleware_spec.rb
+- spec/mongoid_identity_map/current_thread_hash_spec.rb
+- spec/mongoid_identity_map/identity_map_spec.rb
+- spec/mongoid_identity_map/identity_mappable_spec.rb
+- spec/spec_helper.rb
+- spec/support/model.rb
+has_rdoc: true
+homepage: http://rubygems.org/gems/mongoid_identity_map
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: mongoid_identity_map
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: IdentityMap for Mongoid
+test_files: []
+