sensu-plugins-storm 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6653f256959833a2357ca4a6e3efda14dc66e6cc
-  data.tar.gz: 119cb6c56efd92b64ef702f9f0ba64f63504f179
+  metadata.gz: a3b87c6eddd1a19b6b85786b8c5e51947590ae63
+  data.tar.gz: d3573bd025d139712f9ca6256fe589bb0c827670
 SHA512:
-  metadata.gz: 4a83364a158399bd142d236d889087ef9047681226ee2c1eadc4d45156e3bd0f425911d8e47bb629c79f8f81a61d188aa9749b7e460c97aa6531cf27f4a5ebad
-  data.tar.gz: 7f7135867b45bc80d74dc6d6cba45571ef20476ac9a144aa4a4d07547b65a98be13877dd7c141e3ed16ac0ad65ddfeb3fa3cabc0320d18168fb7bd240b16f6a1
+  metadata.gz: 296978b96f0e1959b6d9ade1be38db5c57d671c9123a85b8e344e7445a6f7a4d4ce4b7b32f5493731747c1ea50d285c917c7a2fce1f342a70b9c55e17a77eeb5
+  data.tar.gz: 495830c6b61fdb3d47b27195882ba3308b9ff49b517f6135097236b107d3346874c1c839cf84afc99d43e8e52ceafd1fa8f3761ff4421c1064c6ecdbb81552a0

data/CHANGELOG.md

@@ -3,9 +3,13 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 
 This CHANGELOG follows the format listed at [Keep A Changelog](http://keepachangelog.com/)
 
-## [Unreleased]
-# 0.0.1 - 2016-09-01
+# 0.2.0 - 2016-09-01
+### Added
+- check-storm-capacity script
+
+# 0.1.0 - 2016-09-01
 ### Added
 - initial release
 
-[Unreleased]: https://github.com/andyroyle/sensu-plugins-storm/compare/1.0.0...HEAD
+[0.2.0]: https://github.com/andyroyle/sensu-plugins-storm/compare/0.0.2...HEAD
+[0.1.0]: https://github.com/andyroyle/sensu-plugins-storm/compare/0.0.1...0.0.2

data/README.md

@@ -9,6 +9,7 @@
 
 ## Files
  * bin/check-storm-topologies.rb
+ * bin/check-storm-capacity.rb
 
 ## Usage
 
@@ -17,6 +18,12 @@
 /opt/sensu/embedded/bin$ /opt/sensu/embedded/bin/ruby check-storm-topologies.rb --host=my-storm-cluster.com -s --user=admin --password=password --expect=1
 ```
 
+
+**check-storm-capacity** example
+```bash
+/opt/sensu/embedded/bin$ /opt/sensu/embedded/bin/ruby check-storm-capacity.rb --host=my-storm-cluster.com -s --user=admin --password=password -w 1 -c 1.5
+```
+
 ## Installation
 
 [Installation and Setup](http://sensu-plugins.io/docs/installation_instructions.html)

data/bin/check-storm-capacity.rb

@@ -0,0 +1,108 @@
+#!/usr/bin/env ruby
+#
+# Storm Capacity Check
+# ===
+#
+# Copyright 2016 Andy Royle <ajroyle@gmail.com>
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+#
+# Check the capacity of all running bolts (in all topologies) and compare to crit/warn thresholds
+
+require 'sensu-plugin/check/cli'
+require 'rest-client'
+require 'openssl'
+require 'uri'
+require 'json'
+require 'base64'
+
+class CheckStormTopologies < Sensu::Plugin::Check::CLI
+  option :host,
+         short: '-h',
+         long: '--host=VALUE',
+         description: 'Cluster host'
+
+  option :user,
+         short: '-u',
+         long: '--username=VALUE',
+         description: 'username'
+
+  option :pass,
+         short: '-p',
+         long: '--password=VALUE',
+         description: 'password'
+
+  option :ssl,
+         description: 'use HTTPS (default false)',
+         long: '--ssl'
+
+  option :crit,
+         short: '-c',
+         long: '--critical=VALUE',
+         description: 'Critical threshold'
+
+  option :warn,
+         short: '-w',
+         long: '--warn=VALUE',
+         description: 'Warn threshold'
+
+  option :timeout,
+         short: '-t',
+         long: '--timeout=VALUE',
+         description: 'Timeout in seconds',
+         default: 5
+
+  def request(path)
+    protocol = config[:ssl] ? 'https' : 'http'
+    auth = Base64.encode64("#{config[:user]}:#{config[:pass]}")
+    RestClient::Request.execute(
+      method: :get,
+      url: "#{protocol}://#{config[:host]}:#{config[:port]}/#{path}",
+      timeout: config[:timeout],
+      headers: { 'Authorization' => "Basic #{auth}" }
+    )
+  end
+
+  def run
+    critical_usage = config[:crit].to_f
+    warn_usage = config[:warn].to_f
+
+    if [config[:host], config[:user], config[:pass]].any?(&:nil?)
+      unknown 'Must specify host, user and password'
+    end
+
+    r = request('/stormui/api/v1/topology/summary')
+
+    if r.code != 200
+      critical "unexpected status code '#{r.code}'"
+    end
+
+    topologies = JSON.parse(r.to_str)['topologies']
+    topologies.each do |topology|
+      t = request("/stormui/api/v1/topology/#{topology['id']}")
+      if t.code != 200
+        critical "unexpected status code '#{r.code}'"
+      end
+
+      bolts = JSON.parse(t.to_str)['bolts']
+      bolts.each do |bolt|
+        capacity = bolt['capacity'].to_f
+        if capacity > critical_usage
+          critical "bolt #{bolt['boltId']} has capacity #{bolt['capacity']}"
+        elsif capacity > warn_usage
+          warn "bolt #{bolt['boltId']} has capacity #{bolt['capacity']}"
+        end
+      end
+
+      ok 'all capacities ok'
+    end
+
+  rescue Errno::ECONNREFUSED => e
+    critical 'Storm is not responding' + e.message
+  rescue RestClient::RequestTimeout
+    critical 'Storm Connection timed out'
+  rescue StandardError => e
+    unknown 'An exception occurred:' + e.message
+  end
+end

data/bin/check-storm-topologies.rb

@@ -15,6 +15,7 @@ require 'rest-client'
 require 'openssl'
 require 'uri'
 require 'json'
+require 'base64'
 
 class CheckStormTopologies < Sensu::Plugin::Check::CLI
   option :host,
@@ -47,23 +48,32 @@ class CheckStormTopologies < Sensu::Plugin::Check::CLI
          long: '--expect=VALUE',
          description: 'Match exactly the nuber of topologies'
 
-  def request(path, server)
+  option :timeout,
+         short: '-t',
+         long: '--timeout=VALUE',
+         description: 'Timeout in seconds',
+         default: 5
+
+  def request(path)
     protocol = config[:ssl] ? 'https' : 'http'
-    RestClient::Resource.new("#{protocol}://#{config[:user]}:#{config[:pass]}@#{server}:#{config[:port]}/#{path}", timeout: 5).get
+    auth = Base64.encode64("#{config[:user]}:#{config[:pass]}")
+    RestClient::Request.execute(
+      method: :get,
+      url: "#{protocol}://#{config[:host]}:#{config[:port]}/#{path}",
+      timeout: config[:timeout],
+      headers: { 'Authorization' => "Basic #{auth}" }
+    )
   end
 
   def run
-    user = config[:user]
-    pass = config[:pass]
-    host = config[:host]
     critical_usage = config[:crit].to_f
     expect = config[:expect].to_f
 
-    if [host, user, pass].any?(&:nil?)
+    if [config[:host], config[:user], config[:pass]].any?(&:nil?)
       unknown 'Must specify host, user and password'
     end
 
-    r = request('/stormui/api/v1/topology/summary', host)
+    r = request('/stormui/api/v1/topology/summary')
 
     if r.code != 200
       critical "unexpected status code '#{r.code}'"

data/lib/sensu-plugins-storm/version.rb

@@ -1,7 +1,7 @@
 module SensuPluginsStorm
   module Version
     MAJOR = 0
-    MINOR = 1
+    MINOR = 2
     PATCH = 0
 
     VER_STRING = [MAJOR, MINOR, PATCH].compact.join('.')

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-storm
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Andy Royle
@@ -167,10 +167,12 @@ dependencies:
 description: This plugin provides metrics and checks for Apache Storm topologies
 email: <ajroyle@gmail.com>
 executables:
+- check-storm-capacity.rb
 - check-storm-topologies.rb
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/check-storm-capacity.rb
 - bin/check-storm-topologies.rb
 - lib/sensu-plugins-storm/version.rb
 - lib/sensu-plugins-storm.rb