sensu-plugins-rabbitmq-temp 1.4.0

data/bin/check-rabbitmq-consumers.rb

@@ -0,0 +1,131 @@
+#!/usr/bin/env ruby
+#  encoding: UTF-8
+#
+# Check RabbitMQ consumers
+# ===
+#
+# DESCRIPTION:
+# This plugin checks the number of consumers on the RabbitMQ server
+#
+# PLATFORMS:
+#   Linux, BSD, Solaris
+#
+# DEPENDENCIES:
+#   RabbitMQ rabbitmq_management plugin
+#   gem: sensu-plugin
+#   gem: carrot-top
+#
+# LICENSE:
+# Copyright 2014 Daniel Kerwin <d.kerwin@gini.net>
+# Copyright 2014 Tim Smith <tim@cozy.co>
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+
+require 'sensu-plugin/check/cli'
+require 'carrot-top'
+
+# main plugin class
+class CheckRabbitMQConsumers < Sensu::Plugin::Check::CLI
+  option :host,
+         description: 'RabbitMQ management API host',
+         long: '--host HOST',
+         default: 'localhost'
+
+  option :port,
+         description: 'RabbitMQ management API port',
+         long: '--port PORT',
+         proc: proc(&:to_i),
+         default: 15_672
+
+  option :ssl,
+         description: 'Enable SSL for connection to the API',
+         long: '--ssl',
+         boolean: true,
+         default: false
+
+  option :user,
+         description: 'RabbitMQ management API user',
+         long: '--user USER',
+         default: 'guest'
+
+  option :password,
+         description: 'RabbitMQ management API password',
+         long: '--password PASSWORD',
+         default: 'guest'
+
+  option :queue,
+         description: 'Comma separated list of RabbitMQ queues to monitor.',
+         long: '--queue queue_name',
+         proc: proc { |q| q.split(',') }
+
+  option :exclude,
+         description: 'Comma separated list of RabbitMQ queues to NOT monitor.  All others will be monitored.',
+         long: '--exclude queue_name',
+         proc: proc { |q| q.split(',') }
+
+  option :warn,
+         short: '-w NUM_CONSUMERS',
+         long: '--warn NUM_CONSUMERS',
+         proc: proc(&:to_i),
+         description: 'WARNING consumer count threshold',
+         default: 5
+
+  option :critical,
+         short: '-c NUM_CONSUMERS',
+         long: '--critical NUM_CONSUMERS',
+         description: 'CRITICAL consumer count threshold',
+         proc: proc(&:to_i),
+         default: 2
+
+  def rabbit
+    begin
+      connection = CarrotTop.new(
+        host: config[:host],
+        port: config[:port],
+        user: config[:user],
+        password: config[:password],
+        ssl: config[:ssl]
+      )
+    rescue
+      warning 'could not connect to rabbitmq'
+    end
+    connection
+  end
+
+  def return_condition(missing, critical, warning)
+    if critical.count > 0 || missing.count > 0
+      message = ''
+      message << "Queues in critical state: #{critical.join(', ')}. " if critical.count > 0
+      message << "Queues missing: #{missing.join(', ')}" if missing.count > 0
+      critical(message)
+    elsif warning.count > 0
+      warning("Queues in warning state: #{warning.join(', ')}")
+    else
+      ok
+    end
+  end
+
+  def run
+    # create arrays to hold failures
+    missing = config[:queue] || []
+    critical = []
+    warn = []
+
+    rabbit.queues.each do |queue|
+      # if specific queues were passed only monitor those.
+      # if specific queues to exclude were passed then skip those
+      if config[:queue]
+        next unless config[:queue].include?(queue['name'])
+      elsif config[:exclude]
+        next if config[:exclude].include?(queue['name'])
+      end
+      missing.delete(queue['name'])
+      consumers = queue['consumers']
+      critical.push(queue['name']) if consumers <= config[:critical]
+      warn.push(queue['name']) if consumers <= config[:warn]
+    end
+
+    return_condition(missing, critical, warn)
+  end
+end

data/bin/check-rabbitmq-messages.rb

@@ -0,0 +1,135 @@
+#!/usr/bin/env ruby
+#  encoding: UTF-8
+#
+# Check RabbitMQ Messages
+# ===
+#
+# DESCRIPTION:
+# This plugin checks the total number of messages queued on the RabbitMQ server
+#
+# PLATFORMS:
+#   Linux, BSD, Solaris
+#
+# DEPENDENCIES:
+#   RabbitMQ rabbitmq_management plugin
+#   gem: sensu-plugin
+#   gem: carrot-top
+#
+# LICENSE:
+# Copyright 2012 Evan Hazlett <ejhazlett@gmail.com>
+# Copyright 2015 Tim Smith <tim@cozy.co> and Cozy Services Ltd.
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+
+require 'sensu-plugin/check/cli'
+require 'socket'
+require 'carrot-top'
+
+# main plugin class
+class CheckRabbitMQMessages < Sensu::Plugin::Check::CLI
+  option :host,
+         description: 'RabbitMQ management API host',
+         long: '--host HOST',
+         default: 'localhost'
+
+  option :port,
+         description: 'RabbitMQ management API port',
+         long: '--port PORT',
+         proc: proc(&:to_i),
+         default: 15_672
+
+  option :user,
+         description: 'RabbitMQ management API user',
+         long: '--user USER',
+         default: 'guest'
+
+  option :password,
+         description: 'RabbitMQ management API password',
+         long: '--password PASSWORD',
+         default: 'guest'
+
+  option :ssl,
+         description: 'Enable SSL for connection to the API',
+         long: '--ssl',
+         boolean: true,
+         default: false
+
+  option :warn,
+         short: '-w NUM_MESSAGES',
+         long: '--warn NUM_MESSAGES',
+         description: 'WARNING message count threshold',
+         default: 250
+
+  option :critical,
+         short: '-c NUM_MESSAGES',
+         long: '--critical NUM_MESSAGES',
+         description: 'CRITICAL message count threshold',
+         default: 500
+
+  option :queuelevel,
+         short: '-q',
+         long: '--queuelevel',
+         description: 'Monitors that no individual queue is above the thresholds specified'
+
+  option :excluded,
+         short: '-e queue_name',
+         long: '--excludedqueues queue_name',
+         description: 'Comma separated list of queues to exclude when using queue level monitoring',
+         proc: proc { |q| q.split(',') },
+         default: []
+
+  def generate_message(status_hash)
+    message = []
+    status_hash.each_pair do |k, v|
+      message << "#{k}: #{v}"
+    end
+    message.join(', ')
+  end
+
+  def acquire_rabbitmq_info
+    begin
+      rabbitmq_info = CarrotTop.new(
+        host: config[:host],
+        port: config[:port],
+        user: config[:user],
+        password: config[:password],
+        ssl: config[:ssl]
+      )
+    rescue
+      warning 'Could not connect to rabbitmq'
+    end
+    rabbitmq_info
+  end
+
+  def run
+    rabbitmq = acquire_rabbitmq_info
+
+    # monitor counts in each queue or monitor the total number of messages in the system
+    if config[:queuelevel]
+      warn_queues = {}
+      crit_queues = {}
+      rabbitmq.queues.each do |queue|
+        next if config[:excluded].include?(queue['name'])
+        queue['messages'] ||= 0
+        if queue['messages'] >= config[:critical].to_i
+          crit_queues[(queue['name']).to_s] = queue['messages']
+          next
+        end
+        if queue['messages'] >= config[:warn].to_i
+          warn_queues[(queue['name']).to_s] = queue['messages']
+          next
+        end
+      end
+      message crit_queues.empty? ? generate_message(warn_queues) : generate_message(crit_queues)
+      critical unless crit_queues.empty?
+      warning unless warn_queues.empty?
+    else
+      total = rabbitmq.overview['queue_totals']['messages']
+      message total.to_s
+      critical if total > config[:critical].to_i
+      warning if total > config[:warn].to_i
+    end
+    ok
+  end
+end

data/bin/check-rabbitmq-network-partitions.rb

@@ -0,0 +1,82 @@
+#!/usr/bin/env ruby
+#  encoding: UTF-8
+#
+# RabbitMQ Network Partitions Check
+# ===
+#
+# DESCRIPTION:
+# This plugin checks if a RabbitMQ network partition has occured.
+# https://www.rabbitmq.com/partitions.html
+#
+# PLATFORMS:
+#   Linux, BSD, Solaris
+#
+# DEPENDENCIES:
+#   RabbitMQ rabbitmq_management plugin
+#   gem: sensu-plugin
+#   gem: carrot-top
+#
+# LICENSE:
+# Copyright 2015 Ed Robinson <ed@reevoo.com> and Reevoo LTD.
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+
+require 'sensu-plugin/check/cli'
+require 'carrot-top'
+
+# main plugin class
+class CheckRabbitMQPartitions < Sensu::Plugin::Check::CLI
+  option :host,
+         description: 'RabbitMQ management API host',
+         short: '-w',
+         long: '--host HOST',
+         default: 'localhost'
+
+  option :port,
+         description: 'RabbitMQ management API port',
+         long: '--port PORT',
+         proc: proc(&:to_i),
+         default: 15_672
+
+  option :username,
+         description: 'RabbitMQ management API user',
+         short: '-u',
+         long: '--username USERNAME',
+         default: 'guest'
+
+  option :password,
+         description: 'RabbitMQ management API password',
+         short: '-p',
+         long: '--password PASSWORD',
+         default: 'guest'
+
+  option :ssl,
+         description: 'Enable SSL for connection to the API',
+         long: '--ssl',
+         boolean: true,
+         default: false
+
+  def run
+    critical 'network partition detected' if partition?
+    ok 'no network partition detected'
+  rescue Errno::ECONNREFUSED => e
+    critical e.message
+  rescue => e
+    unknown e.message
+  end
+
+  def partition?
+    rabbitmq_management.nodes.map { |node| node['partitions'] }.any?(&:any?)
+  end
+
+  def rabbitmq_management
+    CarrotTop.new(
+      host: config[:host],
+      port: config[:port],
+      user: config[:username],
+      password: config[:password],
+      ssl: config[:ssl]
+    )
+  end
+end

data/bin/check-rabbitmq-node-health.rb

@@ -0,0 +1,204 @@
+#!/usr/bin/env ruby
+#  encoding: UTF-8
+#
+# RabbitMQ check node health plugin
+# ===
+#
+# DESCRIPTION:
+# This plugin checks if RabbitMQ server node is in a running state.
+#
+# The plugin is based on the RabbitMQ cluster node health plugin by Tim Smith
+#
+# PLATFORMS:
+#   Linux, Windows, BSD, Solaris
+#
+# DEPENDENCIES:
+#   RabbitMQ rabbitmq_management plugin
+#   gem: sensu-plugin
+#   gem: rest-client
+#
+# LICENSE:
+# Copyright 2012 Abhijith G <abhi@runa.com> and Runa Inc.
+# Copyright 2014 Tim Smith <tim@cozy.co> and Cozy Services Ltd.
+# Copyright 2015 Edward McLain <ed@edmclain.com> and Daxko, LLC.
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+
+require 'sensu-plugin/check/cli'
+require 'json'
+require 'rest_client'
+
+# main plugin class
+class CheckRabbitMQNodeHealth < Sensu::Plugin::Check::CLI
+  option :host,
+         description: 'RabbitMQ host',
+         short: '-w',
+         long: '--host HOST',
+         default: 'localhost'
+
+  option :username,
+         description: 'RabbitMQ username',
+         short: '-u',
+         long: '--username USERNAME',
+         default: 'guest'
+
+  option :password,
+         description: 'RabbitMQ password',
+         short: '-p',
+         long: '--password PASSWORD',
+         default: 'guest'
+
+  option :port,
+         description: 'RabbitMQ API port',
+         short: '-P',
+         long: '--port PORT',
+         default: '15672'
+
+  option :ssl,
+         description: 'Enable SSL for connection to the API',
+         long: '--ssl',
+         boolean: true,
+         default: false
+
+  option :verify_ssl_off,
+         description: 'Do not check validity of SSL cert. Use for self-signed certs, etc (insecure)',
+         long: '--verify_ssl_off',
+         boolean: true,
+         default: false
+
+  option :memwarn,
+         description: 'Warning % of mem usage vs high watermark',
+         short: '-m',
+         long: '--mwarn PERCENT',
+         proc: proc(&:to_f),
+         default: 80
+
+  option :memcrit,
+         description: 'Critical % of mem usage vs high watermark',
+         short: '-c',
+         long: '--mcrit PERCENT',
+         proc: proc(&:to_f),
+         default: 90
+
+  option :fdwarn,
+         description: 'Warning % of file descriptor usage vs high watermark',
+         short: '-f',
+         long: '--fwarn PERCENT',
+         proc: proc(&:to_f),
+         default: 80
+
+  option :fdcrit,
+         description: 'Critical % of file descriptor usage vs high watermark',
+         short: '-F',
+         long: '--fcrit PERCENT',
+         proc: proc(&:to_f),
+         default: 90
+
+  option :socketwarn,
+         description: 'Warning % of socket usage vs high watermark',
+         short: '-s',
+         long: '--swarn PERCENT',
+         proc: proc(&:to_f),
+         default: 80
+
+  option :socketcrit,
+         description: 'Critical % of socket usage vs high watermark',
+         short: '-S',
+         long: '--scrit PERCENT',
+         proc: proc(&:to_f),
+         default: 90
+
+  option :watchalarms,
+         description: 'Sound critical if one or more alarms are triggered',
+         short: '-a BOOLEAN',
+         long: '--alarms BOOLEAN',
+         default: 'true'
+
+  def run
+    res = node_healthy?
+
+    if res['status'] == 'ok'
+      ok res['message']
+    elsif res['status'] == 'warning'
+      warning res['message']
+    elsif res['status'] == 'critical'
+      critical res['message']
+    else
+      unknown res['message']
+    end
+  end
+
+  def node_healthy?
+    host       = config[:host]
+    port       = config[:port]
+    username   = config[:username]
+    password   = config[:password]
+    ssl        = config[:ssl]
+    verify_ssl = config[:verify_ssl_off]
+
+    begin
+      url_prefix = ssl ? 'https' : 'http'
+      resource = RestClient::Resource.new(
+        "#{url_prefix}://#{host}:#{port}/api/nodes",
+        user: username,
+        password: password,
+        verify_ssl: !verify_ssl
+      )
+      # Parse our json data
+      nodeinfo = JSON.parse(resource.get)[0]
+
+      # Determine % memory consumed
+      pmem = format('%.2f', nodeinfo['mem_used'].fdiv(nodeinfo['mem_limit']) * 100)
+      # Determine % sockets consumed
+      psocket = format('%.2f', nodeinfo['sockets_used'].fdiv(nodeinfo['sockets_total']) * 100)
+      # Determine % file descriptors consumed
+      pfd = format('%.2f', nodeinfo['fd_used'].fdiv(nodeinfo['fd_total']) * 100)
+
+      # build status and message
+      status = 'ok'
+      message = 'Server is healthy'
+
+      # criticals
+      if pmem.to_f >= config[:memcrit]
+        message = "Memory usage is critical: #{pmem}%"
+        status = 'critical'
+      elsif psocket.to_f >= config[:socketcrit]
+        message = "Socket usage is critical: #{psocket}%"
+        status = 'critical'
+      elsif pfd.to_f >= config[:fdcrit]
+        message = "File Descriptor usage is critical: #{pfd}%"
+        status = 'critical'
+      # warnings
+      elsif pmem.to_f >= config[:memwarn]
+        message = "Memory usage is at warning: #{pmem}%"
+        status = 'warning'
+      elsif psocket.to_f >= config[:socketwarn]
+        message = "Socket usage is at warning: #{psocket}%"
+        status = 'warning'
+      elsif pfd.to_f >= config[:fdwarn]
+        message = "File Descriptor usage is at warning: #{pfd}%"
+        status = 'warning'
+      end
+
+      # If we are set to watch alarms then watch those and set status and messages accordingly
+      if config[:watchalarms] == 'true'
+        if nodeinfo['mem_alarm'] == true
+          status = 'critical'
+          message += ' Memory Alarm ON'
+        end
+
+        if nodeinfo['disk_free_alarm'] == true
+          status = 'critical'
+          message += ' Disk Alarm ON'
+        end
+      end
+
+      { 'status' => status, 'message' => message }
+    rescue Errno::ECONNREFUSED => e
+      { 'status' => 'critical', 'message' => e.message }
+    rescue => e
+      { 'status' => 'unknown', 'message' => e.message }
+    end
+  end
+end