From 72e8442797b90d865811e39418ef9d49238b193a Mon Sep 17 00:00:00 2001
From: Sean Griffin <sean@thoughtbot.com>
Date: Mon, 24 Nov 2014 11:24:31 -0700
Subject: Ensure the type map's cache is thread safe

Thanks to @thedarkone for pointing out that an instance of this object
is used in a shared context.
---
 activerecord/lib/active_record/type/type_map.rb | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'activerecord/lib/active_record/type/type_map.rb')

diff --git a/activerecord/lib/active_record/type/type_map.rb b/activerecord/lib/active_record/type/type_map.rb
index 7c194c0cdf..09f5ba6b74 100644
--- a/activerecord/lib/active_record/type/type_map.rb
+++ b/activerecord/lib/active_record/type/type_map.rb
@@ -1,10 +1,12 @@
+require 'thread_safe'
+
 module ActiveRecord
   module Type
     class TypeMap # :nodoc:
       def initialize
         @mapping = {}
-        @cache = Hash.new do |h, key|
-          h[key] = {}
+        @cache = ThreadSafe::Cache.new do |h, key|
+          h.fetch_or_store(key, ThreadSafe::Cache.new)
         end
       end
 
@@ -13,7 +15,9 @@ module ActiveRecord
       end
 
       def fetch(lookup_key, *args, &block)
-        @cache[lookup_key][args] ||= perform_fetch(lookup_key, *args, &block)
+        @cache[lookup_key].fetch_or_store(args) do
+          perform_fetch(lookup_key, *args, &block)
+        end
       end
 
       def register_type(key, value = nil, &block)
-- 
cgit v1.2.3