hq-aws-check-sqs-queue 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 2cc5d9ea4e103b699c1f4e8d1e1d35b733aabd33
+  data.tar.gz: 5a1f97687ffb686992e90b8233dacb04a6f4772c
+SHA512:
+  metadata.gz: df7fe666c58138e0e5d43afb46c1a819f107c0a30579a4994ff851f5a21d33ddd480894b754ed88c4051d50a9464eaec7b3371ed537c2baa71288cfb71571ac6
+  data.tar.gz: 49ecb752ce42cd4c8c953d4855b9e42d6fac45af7b1ecb609c1c49af0ab12d91aa396ebded1230460a4eb7291b2631da2f5648346724bfdf305ac53cdd6c2720

data/features/basic.feature

@@ -0,0 +1,58 @@
+Feature: Check SQS queue size
+
+  Background:
+
+    Given a file "accounts.xml":
+      """
+      <aws-accounts>
+        <aws-account
+          name="account"
+          access-key-id="access key id"
+          secret-access-key="secret access key"/>
+      </aws-accounts>
+      """
+
+    Given a file "default.args":
+      """
+      --config accounts.xml
+      --account account
+      --queue-name queue
+      --warning 1
+      --critical 2
+      """
+
+  Scenario: Ok
+
+    Given the queue has 0 items
+
+    When I invoke hq-aws-check-sqs-queue with "default.args"
+
+    Then the command stdout should be:
+      """
+      SQS queue OK: 0 messages
+      """
+    And the command exit status should be 0
+
+  Scenario: Warning
+
+    Given the queue has 1 items
+
+    When I invoke hq-aws-check-sqs-queue with "default.args"
+
+    Then the command stdout should be:
+      """
+      SQS queue WARNING: 1 messages (warning is 1)
+      """
+    And the command exit status should be 1
+
+  Scenario: Critical
+
+    Given the queue has 2 items
+
+    When I invoke hq-aws-check-sqs-queue with "default.args"
+
+    Then the command stdout should be:
+      """
+      SQS queue CRITICAL: 2 messages (critical is 2)
+      """
+    And the command exit status should be 2

data/features/support/env.rb

@@ -0,0 +1,7 @@
+require "hq/cucumber/command"
+require "hq/cucumber/temp-dir"
+
+require "hq/aws/check-sqs-queue/script"
+
+$commands["hq-aws-check-sqs-queue"] =
+	HQ::AWS::CheckSqsQueue::Script

data/features/support/steps.rb

@@ -0,0 +1,55 @@
+require "cucumber/rspec/doubles"
+
+Before do
+
+	sqs_double = double "sqs"
+	queues_double = double "queues"
+	queue_double = double "queue"
+
+	::AWS::SQS.stub(:new) do
+		|opts|
+
+		opts[:access_key_id].should == "access key id"
+		opts[:secret_access_key].should == "secret access key"
+		opts[:http_open_timeout].should == 10
+		opts[:http_read_timeout].should == 10
+
+		sqs_double
+
+	end
+
+	sqs_double.stub(:queues) do
+
+		queues_double
+
+	end
+
+	queues_double.stub(:named) do
+		|queue_name|
+
+		queue_name.should == "queue"
+
+		queue_double
+
+	end
+
+	queue_double
+		.stub(:approximate_number_of_messages)
+		.and_return { @messages_in_queue }
+
+	queue_double
+		.stub(:approximate_number_of_messages_delayed)
+		.and_return { 0 }
+
+	queue_double
+		.stub(:approximate_number_of_messages_not_visible)
+		.and_return { 0 }
+
+end
+
+Given /^the queue has (\d+) items$/ do
+	|count_str|
+
+	@messages_in_queue = count_str.to_i
+
+end

data/lib/hq/aws/check-sqs-queue/script.rb

@@ -0,0 +1,121 @@
+require "aws-sdk"
+require "xml"
+
+require "hq/tools/check-script"
+require "hq/tools/escape"
+require "hq/tools/future"
+require "hq/tools/getopt"
+require "hq/tools/thread-pool"
+
+module HQ
+module AWS
+module CheckSqsQueue
+
+class Script < Tools::CheckScript
+
+	include Tools::Escape
+
+	def initialize
+		super
+		@name = "SQS queue"
+	end
+
+	def process_args
+
+		@opts, @args =
+			Tools::Getopt.process @args, [
+
+				{ :name => :warning,
+					:required => true,
+					:convert => :to_i },
+
+				{ :name => :critical,
+					:required => true,
+					:convert => :to_i },
+
+				{ :name => :config,
+					:required => true },
+
+				{ :name => :account,
+					:required => true },
+
+				{ :name => :queue_name,
+					:required => true },
+
+			]
+
+		@args.empty? \
+			or raise "Extra args on command line"
+
+	end
+
+	def prepare
+
+		config_doc =
+			XML::Document.file \
+				@opts[:config]
+
+		@config =
+			config_doc.root
+
+		@account =
+			@config.find_first "
+				aws-account [
+					@name = #{esc_xp @opts[:account]}
+				]
+			"
+
+		raise "No such account" \
+			unless @account
+
+		@sqs =
+			::AWS::SQS.new \
+				:access_key_id =>
+					@account.attributes["access-key-id"],
+				:secret_access_key =>
+					@account.attributes["secret-access-key"],
+				:http_open_timeout => 10,
+				:http_read_timeout => 10
+
+	end
+
+	def perform_checks
+
+		queue =
+			@sqs.queues.named @opts[:queue_name]
+
+		message_count = [
+			queue.approximate_number_of_messages,
+			queue.approximate_number_of_messages_delayed,
+			queue.approximate_number_of_messages_not_visible,
+		].inject(:+)
+
+		if message_count >= @opts[:critical]
+
+			critical "%s messages (critical is %s)" % [
+				message_count,
+				@opts[:critical],
+			]
+
+		elsif message_count >= @opts[:warning]
+
+			warning "%s messages (warning is %s)" % [
+				message_count,
+				@opts[:warning],
+			]
+
+		else
+
+			message "%s messages" % [
+				message_count,
+			]
+
+		end
+
+	end
+
+end
+
+end
+end
+end

metadata

@@ -0,0 +1,190 @@
+--- !ruby/object:Gem::Specification
+name: hq-aws-check-sqs-queue
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- James Pharaoh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.9.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.9.5
+- !ruby/object:Gem::Dependency
+  name: hq-tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: libxml-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+- !ruby/object:Gem::Dependency
+  name: cucumber
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.1
+- !ruby/object:Gem::Dependency
+  name: hq-dev
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.0.13
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.0.13
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.7
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.7
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 10.0.4
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+- !ruby/object:Gem::Dependency
+  name: rspec_junit_formatter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.6
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.6
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.1
+description: Icinga/nagios plugin to check the status of an Amazon SQS queue
+email:
+- james@phsys.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/hq/aws/check-sqs-queue/script.rb
+- features/basic.feature
+- features/support/steps.rb
+- features/support/env.rb
+homepage: https://github.com/jamespharaoh/hq-aws-check-sqs-queue
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: 1.3.6
+requirements: []
+rubyforge_project: hq-aws-check-sqs-queue
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: HQ AWS check SQS queue
+test_files:
+- features/basic.feature
+- features/support/steps.rb
+- features/support/env.rb