Updated mcollective.init according to OSCI-658

[packages/precise/mcollective.git] / lib / mcollective / runnerstats.rb

diff --git a/lib/mcollective/runnerstats.rb b/lib/mcollective/runnerstats.rb
new file mode 100644 (file)
index 0000000..337c969
--- /dev/null
+++ b/lib/mcollective/runnerstats.rb
@@ -0,0 +1,90 @@
+module MCollective
+  # Class to store stats about the mcollectived, it should live in the PluginManager
+  # so that agents etc can get hold of it and return the stats to callers
+  class RunnerStats
+    def initialize
+      @starttime = Time.now.to_i
+      @validated = 0
+      @unvalidated = 0
+      @filtered = 0
+      @passed = 0
+      @total = 0
+      @replies = 0
+      @ttlexpired = 0
+
+      @mutex = Mutex.new
+    end
+
+    # Records a message that failed TTL checks
+    def ttlexpired
+      Log.debug("Incrementing ttl expired stat")
+      @ttlexpired += 1
+    end
+
+    # Records a message that passed the filters
+    def passed
+      Log.debug("Incrementing passed stat")
+      @passed += 1
+    end
+
+    # Records a message that didnt pass the filters
+    def filtered
+      Log.debug("Incrementing filtered stat")
+      @filtered += 1
+    end
+
+    # Records a message that validated ok
+    def validated
+      Log.debug("Incrementing validated stat")
+      @validated += 1
+    end
+
+    def unvalidated
+      Log.debug("Incrementing unvalidated stat")
+      @unvalidated += 1
+    end
+
+    # Records receipt of a message
+    def received
+      Log.debug("Incrementing total stat")
+      @total += 1
+    end
+
+    # Records sending a message
+    def sent
+      @mutex.synchronize do
+        Log.debug("Incrementing replies stat")
+        @replies += 1
+      end
+    end
+
+    # Returns a hash with all stats
+    def to_hash
+      stats = {:validated => @validated,
+        :unvalidated => @unvalidated,
+        :passed => @passed,
+        :filtered => @filtered,
+        :starttime => @starttime,
+        :total => @total,
+        :ttlexpired => @ttlexpired,
+        :replies => @replies}
+
+      reply = {:stats => stats,
+        :threads => [],
+        :pid => Process.pid,
+        :times => {} }
+
+      ::Process.times.each_pair{|k,v|
+        k = k.to_sym
+        reply[:times][k] = v
+      }
+
+      Thread.list.each do |t|
+        reply[:threads] << "#{t.inspect}"
+      end
+
+      reply[:agents] = Agents.agentlist
+      reply
+    end
+  end
+end