majormoses · October 12, 2017 23:00 · Oct 12, 2017
diff --git a/check-http-healthcheck.rb b/check-http-healthcheck.rb
@@ -0,0 +1,84 @@
+#! /usr/bin/env ruby
+#
+# check-cc-heap-limit
+#
+# DESCRIPTION:
+#   A script that checks an http endpoint, parses json, and alerts when healthchecks fail
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#   gem: httparty
+#
+# USAGE:
+#
+# NOTES:
+#   
+#
+# LICENSE:
+#   Copyright (c) 2017, Huy Nguyen <[email protected]>
+
+require 'sensu-plugin/check/cli'
+require 'httparty'
+require 'json'
+
+class HealthCheck < Sensu::Plugin::Check::CLI
+  option :host,
+         short: '-h HOSTNAME',
+         long: '--hostname HOSTNAME',
+         description: 'Host to connect to query for healthcheck',
+         default: 'localhost'
+
+  option :port,
+         short: '-p PORT',
+         long: '--port PORT',
+         description: 'Port to connect to query for healthcheck',
+         default: '8081'
+
+  option :timeout,
+         short: '-t HTTP_TIMEOUT',
+         long: '--timeout HTTP_TIMEOUT',
+         description: 'HTTP timeout in case the HTTP request takes too long',
+         default: 10,
+         proc: proc(&:to_i)
+
+  option :blacklist,
+         short: '-b BLACKLIST',
+         long: '--blacklist BLACKLIST',
+         description: 'comma seperated list of checks that does not need to be alerted out if fails',
+         default: nil,
+         proc: proc(&:to_s)
+
+  def run
+    begin
+      response = HTTParty.get("http://#{config[:host]}:#{config[:port]}/healthcheck", timeout: config[:timeout])
+    rescue Errno::ECONNREFUSED => e
+      critical "failed to connect to service healthcheck on #{config[:host]}:#{config[:port]}. #{e}"
+    rescue Net::ReadTimeout => e
+      warning "Timed out after #{config[:timeout]} seconds connecing to http://#{config[:port]}/healthcheck. #{e}"
+    rescue StandardError => e
+      unknown "Something went wrong accessing http://#{config[:host]}:#{config[:port]}/healthcheck. #{e}"
+    end
+
+    checks = JSON.parse(response.body)
+    failed_checks = {}
+
+    checks.keys.each do |key|
+      unless !config[:blacklist].nil? && (config[:blacklist].include? key)
+        next unless checks[key]['healthy'] == false
+        failed_checks.store(key, checks[key]['message'])
+      end
+    end
+
+    if failed_checks.any?
+      critical "#{failed_checks} healthcheck(s) is/are failing on #{config[:host]}:#{config[:port]}"
+    else
+      ok "All healthchecks are passing for #{config[:host]}:#{config[:port]}"
+    end
+  end
+end