riemann-sqs 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d8c4357305c6b39d2a32cf36fdeed6afbba88384
-  data.tar.gz: 62c7ae6d651278f0a6f1e0385f075b7649f883f3
+  metadata.gz: 906463679bbf8708c8831279358ef51c7871ef5f
+  data.tar.gz: e70eb225280c8db54e15fc3243032ef210f3ae1d
 SHA512:
-  metadata.gz: 00e0989b0cfb57d952cbbec8a51f887a3fa9e5d484dc4515480052601b992dc39abb6dc124df7d7dd27296e7979e0db7f7ae5b7b3c58468a4d92bbfd26b28de2
-  data.tar.gz: 4bc44631016bde6d8fc215d36f31d5c1c7b5a2f1eddebcc4377863ae22e47702329680d26aba2c6507203d9ebf7578507a45c884e9c35d821028b35efd42bddb
+  metadata.gz: d98b88d9abb59e4884754b94a384bdbe34f2d9150ed516cda6441f74bdf7caf2c716dcd3d4a98d0d34070fde182d3c0f39ac9223fd74a9964afcec1832ac8aa5
+  data.tar.gz: 6422397499130e80091d509deddbfb175b7b44b5c80a850746c81dc7266d3ae20fdb6666a7cfae99f5e999328f8a5774fb5f469c769cada68cb479c5f3f08c5d

data/bin/riemann-sqs

@@ -13,24 +13,27 @@ class Riemann::Tools::AWS
   opt :access_key, "AWS access key", :type => String
   opt :secret_key, "Secret access key", :type => String
   opt :region, "AWS region", :type => String, :default => 'us-east-1'
-  opt :queue, "SQS Queue name", :type => String
+  opt :queue, "SQS Queue name", :type => String, :multi => true
+
   def initialize
     @sqs = Fog::AWS::SQS.new(:aws_access_key_id => opts[:access_key],
                              :aws_secret_access_key => opts[:secret_key],
                              :region => opts[:region])
-    response = @sqs.list_queues({'QueueNamePrefix' => opts[:queue]})
-    @queue_url = response[:body]['QueueUrls'].first
+    response = @sqs.list_queues
+    @queues_info = opts[:queue].inject({}) {|h, name| h[name] = response[:body]['QueueUrls'].select {|url| url.include? name }.first; h }
   end
 
   def tick
-    response = @sqs.get_queue_attributes(@queue_url, 'All')
-    ['ApproximateNumberOfMessages', 'ApproximateNumberOfMessagesNotVisible'].each do |attr|
-      msg = {
-        metric: response[:body]['Attributes'][attr],
-        service: "#{opts[:queue]} #{attr}",
-        state: 'ok'
-      }
-      report msg
+    @queues_info.each_pair do |queue, queue_url|
+      response = @sqs.get_queue_attributes(queue_url, 'All')
+      ['ApproximateNumberOfMessages', 'ApproximateNumberOfMessagesNotVisible'].each do |attr|
+        msg = {
+          metric: response[:body]['Attributes'][attr],
+          service: "#{queue} #{attr}",
+          state: 'ok'
+        }
+        report msg
+      end
     end
   end
 end

data/riemann-sqs.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "riemann-sqs"
-  spec.version       = '0.0.1'
+  spec.version       = '0.0.2'
   spec.authors       = ["Fernando Alonso"]
   spec.email         = ["krakatoa1987@gmail.com"]
   spec.summary       = %q{Riemann agent to collect SQS queue metrics}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: riemann-sqs
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Fernando Alonso
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-25 00:00:00.000000000 Z
+date: 2015-08-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: riemann-tools