monitoring-client 0.2.8 → 0.2.9

data/Gemfile.lock

@@ -1,13 +1,14 @@
 PATH
   remote: .
   specs:
-    monitoring-client (0.2.6)
+    monitoring-client (0.2.9)
       yajl-ruby
 
 GEM
   remote: http://rubygems.org/
   specs:
     diff-lcs (1.1.3)
+    rake (0.9.2)
     rspec (2.6.0)
       rspec-core (~> 2.6.0)
       rspec-expectations (~> 2.6.0)
@@ -23,4 +24,5 @@ PLATFORMS
 
 DEPENDENCIES
   monitoring-client!
+  rake
   rspec

data/bin/monitor-rabbitmq-queue-lengths

@@ -0,0 +1,171 @@
+#!/usr/bin/env ruby
+
+module Monitoring
+  class RabbitMqQueueLengths
+
+    require "optparse"
+    require "yaml"
+    require "monitoring/client/version"
+    require "monitoring/client/profiler"
+    
+    RC_OK = 0
+    RC_USAGE = 1
+    RC_KILLED = 2
+    
+    
+    def initialize()
+      @config_fn = File.join(ENV["HOME"], ".monitoring.rabbitmq-queue-lengths.yml")
+      @daemon_mode = true
+      @pid_file_fn = "/var/run/monitor-rabbitmq-queue-lengths.pid"
+      @sleep_period = 60
+      @stdout = "/dev/null"
+      @stderr = "/dev/null"
+      @alive = true
+    end
+    
+
+    def parse_opts(argv)
+      optparser = OptionParser.new() do |o|
+        o.banner = "monitor-rabbitmq-queue-lengths [options]"
+
+        o.separator("")
+        o.separator("options")
+
+        o.on("-c", "--config=CONFIG",
+             "Config file to use",
+             "Default: #{@config_fn.inspect()}") do |v|
+          @config_fn = v
+        end
+
+        o.on("-f", "--forgeround",
+             "Run in forground",
+             "Default: #{!@daemon_mode}") do |v|
+          @daemon_mode = !v
+        end
+
+        o.on("--pid-file=FILE",
+             "Place to store our pid file",
+             "Default: " + @pid_file_fn.inspect()) do |v|
+          @pid_file_fn = v
+        end
+
+        o.on("--sleep-period=s",
+             "Time to sleep between posts",
+             "Default: #{@sleep_period}") do |v|
+          @sleep_period = Float(v)
+        end
+
+        o.on("--stdout=FILE",
+             "Place to write stdout to",
+             "Default: " + @stdout.inspect()) do |v|
+          @stdout = v
+        end
+
+        o.on("-v", "--version",
+             "Print version and exit") do |v|
+          STDOUT.puts(Monitoring::Client::VERSION)
+          Kernel.exit(RC_OK)
+        end
+
+        o.on("--stderr=FILE",
+             "Place to write stdout to",
+             "Default: " + @stderr.inspect()) do |v|
+          @stderr = v
+        end
+        
+        o.separator("")
+        o.on_tail("-h", "--help", "You're reading it :-)") do
+          puts(o)
+          Kernel.exit(RC_USAGE)
+        end
+      end
+
+      optparser.parse(argv)
+      self
+    end
+
+    def load_config()
+      @config = YAML.load_file(@config_fn)
+      [
+       :profiler_name,
+      ].each do |expected_key|
+        unless @config.has_key?(expected_key) &&
+            !@config[expected_key].nil? &&
+            @config[expected_key].size > 0
+          die("Bad config at #{@config_fn}: "\
+              "Expected #{expected_key} to exist and not be blank")
+        end
+      end
+    end
+
+    def run()
+      load_config()
+      daemonize() if @daemon_mode
+
+      trap(:INT) do
+        if @alive
+          STDOUT.puts("SIGINT: going away gracefully")
+          @alive = false
+        else
+          STDOUT.puts("SIGINT (again): going away not so gracefully")
+          Kernel.exit!(RC_KILLED)
+        end
+      end
+      
+      while @alive
+        post_queue_lengths
+        Kernel.sleep(@sleep_period)
+      end
+    end
+
+    def daemonize()
+      pid_file_fd = File.open(@pid_file_fn, "w")
+      pid_file_fd.flock(File::LOCK_EX | File::LOCK_NB) ||
+        die("Daemon already exists")
+
+      die("First fork failed") if (pid = Kernel.fork()) == -1
+      exit unless pid.nil?
+
+      Process.setsid
+      pid_file_fd.puts(Process.pid)
+      pid_file_fd.flush()
+
+      Dir.chdir("/")
+      File.umask(0000)
+      STDIN.reopen("/dev/null")
+      STDOUT.reopen(@stdout)
+      STDERR.reopen(@stderr)
+    end
+
+    def die(why)
+      STDERR.puts(why)
+      Kernel.exit!(1)
+    end
+
+    def get_profiler
+      @profiler ||= Monitoring::Client::Profiler.new(@config[:profiler_name])
+    end
+    
+    def routine(routine_name)
+      get_profiler.routine(routine_name) do |routine|
+        yield(routine)
+      end
+    end
+    
+    def post_queue_lengths
+      list_queues = %x{sudo rabbitmqctl list_queues}.split("\n")
+      list_queues.slice!(0)
+      list_queues.slice!(list_queues.size - 1)
+      routine("list_queues") do |m|
+        m.count("n-queues", list_queues.size, "n")
+        list_queues.each do |line|
+          queue_name, size = line.split("\t")
+          size = Integer(size)
+          m.count(queue_name, size, "n")
+        end
+      end
+    end
+  end
+end
+
+Monitoring::RabbitMqQueueLengths.new().parse_opts(ARGV).run()

data/bin/monitor-system-info

@@ -14,9 +14,9 @@ module Monitoring
     
     
     def initialize()
-      @config_fn = File.join(ENV["HOME"], ".monitoring.yml")
+      @config_fn = File.join(ENV["HOME"], ".monitoring.system-info.yml")
       @daemon_mode = true
-      @pid_file_fn = "/var/run/monitoringd.pid"
+      @pid_file_fn = "/var/run/monitor-system-info.pid"
       @sleep_period = 60
       @stdout = "/dev/null"
       @stderr = "/dev/null"

data/lib/monitoring/client/version.rb

@@ -1,5 +1,5 @@
 module Monitoring
   module Client
-    VERSION = "0.2.8"
+    VERSION = "0.2.9"
   end
 end

data/monitoring-client.gemspec

@@ -21,4 +21,5 @@ Gem::Specification.new do |s|
   
   s.add_runtime_dependency "yajl-ruby"
   s.add_development_dependency "rspec"
+  s.add_development_dependency "rake"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: monitoring-client
 version: !ruby/object:Gem::Version 
-  hash: 7
+  hash: 5
   prerelease: 
   segments: 
   - 0
   - 2
-  - 8
-  version: 0.2.8
+  - 9
+  version: 0.2.9
 platform: ruby
 authors: 
 - Marc Bowes
@@ -15,12 +15,11 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-09-24 00:00:00 +02:00
+date: 2011-10-10 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
-  name: yajl-ruby
-  prerelease: false
+  type: :runtime
   requirement: &id001 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
@@ -30,11 +29,11 @@ dependencies:
         segments: 
         - 0
         version: "0"
-  type: :runtime
+  name: yajl-ruby
   version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: rspec
   prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
   requirement: &id002 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
@@ -44,12 +43,28 @@ dependencies:
         segments: 
         - 0
         version: "0"
-  type: :development
+  name: rspec
   version_requirements: *id002
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  type: :development
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  name: rake
+  version_requirements: *id003
+  prerelease: false
 description: Ruby client which uses UDP to communicate with a daemon which uses HTTP POST to communicate with a Web server
 email: 
 - marcbowes+monitoring+client@gmail.com
 executables: 
+- monitor-rabbitmq-queue-lengths
 - monitor-system-info
 - monitoringd
 extensions: []
@@ -61,6 +76,7 @@ files:
 - Gemfile
 - Gemfile.lock
 - Rakefile
+- bin/monitor-rabbitmq-queue-lengths
 - bin/monitor-system-info
 - bin/monitoringd
 - lib/monitoring.rb
@@ -109,5 +125,8 @@ rubygems_version: 1.6.2
 signing_key: 
 specification_version: 3
 summary: Simple client & daemon to post metrics
-test_files: []
-
+test_files: 
+- spec/compiler_spec.rb
+- spec/config_spec.rb
+- spec/profiler_spec.rb
+- spec/spec_helper.rb