allow class/singleton methods to be profiled

Author: kbrock

lib/mini_profiler/profiling_methods.rb

@@ -109,6 +109,14 @@ def profile_method(klass, method, type = :profile, &blk)
         klass.send :alias_method, method, with_profiling
       end
 
+      def profile_singleton_method(klass, method, type = :profile, &blk)
+        profile_method(singleton_class(klass), method, type, &blk)
+      end
+
+      def unprofile_singleton_method(klass, method)
+        unprofile_method(singleton_class(klass), method)
+      end
+
       # Add a custom timing. These are displayed similar to SQL/query time in
       # columns expanding to the right.
       #
@@ -134,6 +142,10 @@ def counter(type, duration_ms=nil)
 
       private
 
+      def singleton_class(klass)
+        class << klass; self; end
+      end
+
       def clean_method_name(method)
         method.to_s.gsub(/[\?\!]/, "")
       end

spec/components/profiler_spec.rb

@@ -56,6 +56,10 @@ class TestClass
         def foo(bar,baz)
           return [bar, baz, yield]
         end
+
+        def self.bar(baz,boo)
+          return [baz, boo, yield]
+        end
       end
     end
 
@@ -65,6 +69,12 @@ def foo(bar,baz)
       Rack::MiniProfiler.unprofile_method TestClass, :foo
     end
 
+    it 'should not destroy a singleton method' do
+      Rack::MiniProfiler.profile_singleton_method TestClass, :bar
+      TestClass.bar("a", "b"){"c"}.should == ["a","b","c"]
+      Rack::MiniProfiler.unprofile_singleton_method TestClass, :bar
+    end
+
   end
 
   describe 'step' do