Add a count of the number of mongo calls recorded

lib/mongo/rails/instrumentation/log_subscriber.rb

@@ -1,20 +1,32 @@
 require 'mongo/rails/instrumentation'
 
 class Mongo::Rails::Instrumentation::LogSubscriber < ActiveSupport::LogSubscriber
+  RUNTIME_KEY = name + "#runtime"
+  COUNT_KEY = name + "#count"
+
   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
+    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.2.2"
+      VERSION = '0.2.3'
     end
   end
 end