riemann-smith 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6da5ba98f3d4f25cd285477aac6f54cf3f70881d
-  data.tar.gz: 18a937f08c2deed7cc3aa7313f4a4297054a5acf
+  metadata.gz: ddabc131163b71844a1d0252bd9f5eecf18e06c0
+  data.tar.gz: 6f9b585f1491efdfe0d4f766893451506842ae12
 SHA512:
-  metadata.gz: 48c154b75ff5fc4d8ce4f6037ad9a5399187cb584327f99f732859acda9032a188564ff36a8ccbf91e3a7c7b7d2592a70817db80c8865369938c8fe6265780c3
-  data.tar.gz: 71b3260dfc5dd172502dd3b5d4b75702cd35537768e4a2e2282411c35060dcb84e0dade2015616f38b547f3ead4f8daf28c10d4f517a6e1c43d8de2c92849422
+  metadata.gz: ac33d63cb456c2a4a22e9ce616e04e4551f58220d9c7c6f4b79e386e06fa64c1fa443181abc860b81b7b5e99411192ecd061b37a859fd2df2a4adb75545aaa37
+  data.tar.gz: 49a96af674fc7fa8133abd16477d7c8c37bf11e4edaf18df7447382178472ef721d0448550bd6eb6849a26a09804de8de8404c6d74f69e9721eacdc5cef39174

data/bin/queue-lengths

@@ -0,0 +1,65 @@
+#!/usr/bin/env ruby
+
+require "pp"
+require "pathname"
+
+require "rabbitmq/http/client"
+require 'riemann/tools'
+
+class QueueLength
+  include Riemann::Tools
+
+  opt :limits, "JSON file containing queue:limit key value pairs", :type => :string, :required => true
+  opt :queue_prefix, "Queue prefix", :type => :string, :default => "smith"
+  opt :exclusion_regex, "Exclude queue names that conform to the regex", :type => :string
+  opt :exclusions, "Exclude the given queue names.", :type => :string
+
+  def initialize
+    @rabbitmq = RabbitMQ::HTTP::Client.new("http://guest:guest@localhost:15672")
+  end
+
+  def tick
+    @rabbitmq.list_queues.each do |queue|
+      over_limit(queue)
+    end
+  end
+
+  private
+
+  def over_limit(queue)
+    alert(queue, state(queue), queue.messages, "Number of messages")
+  end
+
+  def state(queue)
+    if queue.messages > limit(queue)['critical']
+      state = :critial
+    elsif queue.messages > limit(queue)['warning']
+      state = :warning
+    else
+      state = :ok
+    end
+  end
+
+  def alert(queue, state, metric, description)
+    report(
+      :service => normliased_queue_name(queue),
+      :state => state.to_s,
+      :metric => metric.to_f,
+      :description => description
+    )
+  end
+
+  def limit(queue)
+    (limits[normliased_queue_name(queue)] || limits['default'])
+  end
+
+  def normliased_queue_name(queue)
+    queue.name.sub(Regexp.new("#{opts[:queue_prefix]}."), '')
+  end
+
+  def limits
+    @limits ||= MultiJson.load(Pathname.new(opts[:limits]))
+  end
+end
+
+QueueLength.run

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: riemann-smith
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Richard Heycock
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-11 00:00:00.000000000 Z
+date: 2014-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: git-version-bump
@@ -124,10 +124,13 @@ dependencies:
         version: '0'
 description: 
 email: rgh@digivizer.com
-executables: []
+executables:
+- queue-lengths
+- riemann-smith
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/queue-lengths
 - bin/riemann-smith
 - doc/LICENSE
 - doc/README