From ef6f6625c91ea789a033799f649e4388e4a71045 Mon Sep 17 00:00:00 2001
From: Joshua Peek <josh@joshpeek.com>
Date: Fri, 18 Jul 2008 15:32:28 -0500
Subject: Changed ActiveSupport::Memoizable API to extend since it mainly adds
 the memoize class method

---
 activesupport/lib/active_support/memoizable.rb | 55 +++++++++++++-------------
 1 file changed, 28 insertions(+), 27 deletions(-)

(limited to 'activesupport/lib')

diff --git a/activesupport/lib/active_support/memoizable.rb b/activesupport/lib/active_support/memoizable.rb
index c41feef4c7..59fecbecb1 100644
--- a/activesupport/lib/active_support/memoizable.rb
+++ b/activesupport/lib/active_support/memoizable.rb
@@ -1,37 +1,38 @@
 module ActiveSupport
-  module Memoizable
-    def self.included(base) #:nodoc:
-      base.extend(ClassMethods)
-    end
-
-    module ClassMethods
-      def memoize(*symbols)
-        symbols.each do |symbol|
-          original_method = "unmemoized_#{symbol}"
-          memoized_ivar = "@#{symbol}"
-          raise "Already memoized #{symbol}" if instance_methods.map(&:to_s).include?(original_method)
-
-          alias_method original_method, symbol
-          class_eval <<-EOS, __FILE__, __LINE__
-            def #{symbol}(reload = false)
-              if !reload && defined? #{memoized_ivar}
-                #{memoized_ivar}
-              else
-                #{memoized_ivar} = #{original_method}.freeze
-              end
+  module Memoizable #:nodoc:
+    def self.extended(obj)
+      klass = obj.respond_to?(:class_eval) ? obj : obj.metaclass
+      klass.class_eval do
+        def freeze
+          methods.each do |method|
+            if m = method.to_s.match(/^unmemoized_(.*)/)
+              send(m[1])
             end
-          EOS
+          end
+          super
         end
       end
     end
 
-    def freeze
-      methods.each do |method|
-        if m = method.to_s.match(/\Aunmemoized_(.*)/)
-          send(m[1])
-        end
+    def memoize(*symbols)
+      symbols.each do |symbol|
+        original_method = "unmemoized_#{symbol}"
+        memoized_ivar = "@#{symbol}"
+
+        klass = respond_to?(:class_eval) ? self : self.metaclass
+        raise "Already memoized #{symbol}" if klass.instance_methods.map(&:to_s).include?(original_method)
+
+        klass.class_eval <<-EOS, __FILE__, __LINE__
+          alias_method :#{original_method}, :#{symbol}
+          def #{symbol}(reload = false)
+            if !reload && defined? #{memoized_ivar}
+              #{memoized_ivar}
+            else
+              #{memoized_ivar} = #{original_method}.freeze
+            end
+          end
+        EOS
       end
-      super
     end
   end
 end
-- 
cgit v1.2.3