Merge 6da8df8 into 3ad5568

tycooon · May 27, 2019 · bf8dae0 · bf8dae0
2 parents 3ad5568 + 6da8df8
commit bf8dae0
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/lib/memery.rb b/lib/memery.rb
@@ -44,16 +44,15 @@ def define_memoized_method!(method_name, condition: nil)
             return super(*args, &block)
           end
 
-          @_memery_memoized_values ||= {}
+          key = "#{method_name}_#{mod_id}"
 
-          key = [method_name, mod_id].join("_").to_sym
-          store = @_memery_memoized_values[key] ||= {}
+          store = @_memery_memoized_values&.[](key)
 
-          if store.key?(args)
-            store[args]
-          else
-            store[args] = super(*args)
-          end
+          return store[args] if store&.key?(args)
+
+          @_memery_memoized_values ||= {}
+          @_memery_memoized_values[key] ||= {}
+          @_memery_memoized_values[key][args] = super(*args)
         end
 
         send(visibility, method_name)