From 3df4460a742c18af8cc71e41009c9028467d16b5 Mon Sep 17 00:00:00 2001
From: Marcin Raczkowski <marcin.raczkowski@gmail.com>
Date: Sat, 28 Aug 2010 18:29:28 +0200
Subject: [PATCH] IdentityMap - Adding Weakling and IM Base as concern

---
 .gitignore                                    |  1 +
 Gemfile                                       |  2 +
 activerecord/lib/active_record.rb             |  2 +
 .../lib/active_record/identity_map.rb         | 64 +++++++++++++++++++
 4 files changed, 69 insertions(+)
 create mode 100644 activerecord/lib/active_record/identity_map.rb

diff --git a/.gitignore b/.gitignore
index a18fba3780d0b..81403942c9e62 100644
--- a/.gitignore
+++ b/.gitignore
@@ -20,3 +20,4 @@ railties/test/initializer/root/log
 railties/doc
 railties/guides/output
 railties/tmp
+nbproject
diff --git a/Gemfile b/Gemfile
index 4e13331626a47..4e1bd1325e970 100644
--- a/Gemfile
+++ b/Gemfile
@@ -25,6 +25,8 @@ gem "memcache-client", ">= 1.8.5"
 # AM
 gem "text-format", "~> 1.0.0"
 
+gem "weakling", :git => "git://github.com/swistak/weakling.git"
+
 platforms :mri_18 do
   gem "system_timer"
   gem "ruby-debug", ">= 0.10.3"
diff --git a/activerecord/lib/active_record.rb b/activerecord/lib/active_record.rb
index c80bce28492f3..ff6cf44a73b51 100644
--- a/activerecord/lib/active_record.rb
+++ b/activerecord/lib/active_record.rb
@@ -32,6 +32,7 @@
 require 'active_support/i18n'
 require 'active_model'
 require 'arel'
+require 'weakling'
 
 require 'active_record/version'
 
@@ -79,6 +80,7 @@ module ActiveRecord
     autoload :Timestamp
     autoload :Transactions
     autoload :Validations
+    autoload :IdentityMap
   end
 
   module AttributeMethods
diff --git a/activerecord/lib/active_record/identity_map.rb b/activerecord/lib/active_record/identity_map.rb
new file mode 100644
index 0000000000000..79eb4dd83a5d7
--- /dev/null
+++ b/activerecord/lib/active_record/identity_map.rb
@@ -0,0 +1,64 @@
+module ActiveRecord
+  module IdentityMap
+    extend ActiveSupport::Concern
+
+    class << self
+      attr_accessor :repositories
+      attr_accessor :current_repository_name
+      attr_accessor :enabled
+
+      def current
+        repositories[current_repository_name] ||= Weakling::WeakHash.new
+      end
+
+      def with_repository(name = :default, &block)
+        old_repository = self.current_repository_name
+        self.current_repository_name = name
+
+        block.call(current)
+      ensure
+        self.current_repository_name = old_repository
+      end
+
+      def without(&block)
+        old, self.enabled = self.enabled, false
+
+        block.call
+      ensure
+        self.enabled = old
+      end
+
+      def get(class_name, primary_key)
+        current[[class_name, primary_key]]
+      end
+
+      def add(record)
+        current[[record.class.name, record.id]] = record
+      end
+
+      def remove(record)
+        current.delete([record.class.name, record.id])
+      end
+
+      def clear
+        current.clear
+      end
+
+      alias enabled? enabled
+    end
+
+    self.repositories ||= Hash.new
+    self.current_repository_name ||= :default
+    self.enabled = true
+
+    module InstanceMethods
+
+    end
+
+    module ClassMethods
+      def identity_map
+        ActiveRecord::IdentityMap
+      end
+    end
+  end
+end