1 files changed, 41 insertions, 0 deletions
diff --git a/activesupport/lib/active_support/weak_hash.rb b/activesupport/lib/active_support/weak_hash.rb
new file mode 100644
index 0000000000..d6096e606e
--- /dev/null
+++ b/activesupport/lib/active_support/weak_hash.rb
@@ -0,0 +1,41 @@
+module ActiveSupport
+  if defined?(RUBY_ENGINE) && RUBY_ENGINE == 'jruby'
+    WeakHash = ::Weakling::WeakHash
+  else
+    class WeakHash
+      def initialize(cache = Hash.new)
+        @cache = cache
+        @key_map = {}
+        @rev_cache = Hash.new{|h,k| h[k] = {}}
+        @reclaim_value = lambda do |value_id|
+          if value = @rev_cache.delete(value_id)
+            value.each_key{|key| @cache.delete key}
+          end
+        end
+      end
+
+      def [](key)
+        value_id = @cache[key]
+        value_id && ObjectSpace._id2ref(value_id)
+      rescue RangeError
+        nil
+      end
+
+      def []=(key, value)
+        @rev_cache[value.object_id][key] = true
+        @cache[key] = value.object_id
+        @key_map[key.object_id] = key
+
+        ObjectSpace.define_finalizer(value, @reclaim_value)
+      end
+
+      def clear
+        @cache.clear
+      end
+
+      def delete(key)
+        @cache.delete(key)
+      end
+    end
+  end
+end