Count the number of requests to mongo, as well as the total request r…

…untime

lib/mongo/rails/instrumentation/controller_runtime.rb

@@ -9,9 +9,9 @@ module ControllerRuntime
     attr_internal :mongo_runtime
 
     def cleanup_view_runtime
-      mongo_rt_before_render = LogSubscriber.reset_runtime
+      mongo_rt_before_render = LogSubscriber.reset
       runtime = super
-      mongo_rt_after_render = LogSubscriber.reset_runtime
+      mongo_rt_after_render = LogSubscriber.reset
       self.mongo_runtime = mongo_rt_before_render + mongo_rt_after_render
       runtime - mongo_rt_after_render
     end

lib/mongo/rails/instrumentation/log_subscriber.rb

@@ -1,20 +1,33 @@
 require 'mongo/rails/instrumentation'
 
 class Mongo::Rails::Instrumentation::LogSubscriber < ActiveSupport::LogSubscriber
+  PREFIX = self.class.name.freeze
+  COUNT_KEY = PREFIX + "#count".freeze
+  RUNTIME_KEY = PREFIX + "#runtime".freeze
+
   def self.runtime=(value)
-    Thread.current["mongo_mongo_runtime"] = value
+    Thread.current[RUNTIME_KEY] = value
   end
 
   def self.runtime
-    Thread.current["mongo_mongo_runtime"] ||= 0
+    Thread.current[RUNTIME_KEY] ||= 0
+  end
+
+  def self.count=(value)
+    Thread.current[COUNT_KEY] = value
+  end
+
+  def self.count
+    Thread.current[COUNT_KEY] ||= 0
   end
 
-  def self.reset_runtime
-    rt, self.runtime = runtime, 0
+  def self.reset
+    rt, self.runtime, self.count = runtime, 0, 0
     rt
   end
 
   def mongo(event)
     self.class.runtime += event.duration
+    self.class.count += 1
   end
 end

lib/mongo/rails/instrumentation/version.rb

@@ -1,7 +1,7 @@
 module Mongo
   module Rails
     module Instrumentation
-      VERSION = "0.1.0"
+      VERSION = "0.2.0"
     end
   end
 end

mongo-rails-instrumentation.gemspec

@@ -9,8 +9,8 @@ Gem::Specification.new do |s|
   s.authors     = ["Tom Ward"]
   s.email       = ["tom@popdog.net"]
   s.homepage    = "http://tomafro.net"
-  s.summary     = %q{Records time spent in mongo and adds to request logs}
-  s.description = %q{Records time spent in mongo and adds to request logs}
+  s.summary     = %q{Record time spent in Mongo and add to request logs}
+  s.description = %q{Record time spent in Mongo and add to request logs}
 
   s.files         = `git ls-files`.split("\n")
   s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")