sensu-plugins-graphite-boutetnico 1.0.0

data/bin/handler-graphite-event.rb

@@ -0,0 +1,86 @@
+#!/usr/bin/env ruby
+#
+# Sensu - Graphite Event Handler
+#
+# This handler takes events and POSTs them to a Graphite events URI.
+#
+# For configuration see: graphite_event.json
+#
+# See here for more details:
+#
+# * https://code.launchpad.net/~lucio.torre/graphite/add-events/+merge/69142
+#
+# Author: Rob Wilson <roobert@gmail.com>
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details
+
+require 'sensu-handler'
+require 'net/http'
+require 'net/https'
+require 'uri'
+require 'json'
+
+class GraphiteEvent < Sensu::Handler
+  option :tags_as_array,
+         description: 'send tags as array - for post graphite 1.0 compatibility.',
+         short: '-t',
+         long: '--tags-as-array',
+         boolean: true,
+         required: false,
+         default: false
+
+  def post_event(uri, body)
+    uri          = URI.parse(uri)
+    req          = Net::HTTP::Post.new(uri.path)
+    sock         = Net::HTTP.new(uri.host, uri.port)
+    sock.use_ssl = uri.scheme == 'https'
+    req.body     = body
+
+    req.basic_auth(uri.user, uri.password) if uri.user
+
+    sock.start { |http| http.request(req) }
+  end
+
+  def event_status
+    case @event['check']['status']
+    when 0
+      'ok'
+    when 1
+      'warning'
+    when 2
+      'critical'
+    else
+      'unknown'
+    end
+  end
+
+  def handle
+    tags = [
+      'sensu',
+      'event',
+      event_status,
+      @event['client']['name'],
+      @event['check']['name']
+    ]
+
+    tags += settings['graphite_event']['tags'] if settings['graphite_event']['tags']
+
+    body = {
+      'what' => 'sensu_event',
+      'tags' => config[:tags_as_array] ? [tags.join(',')] : tags.join(','),
+      'data' => event_status,
+      'when' => Time.now.to_i
+    }
+
+    uri = settings['graphite_event']['server_uri']
+
+    begin
+      post_event(uri, body.to_json)
+    rescue => e
+      bail "failed to send event to #{uri}: #{e}"
+    end
+
+    puts "sent event to Graphite: #{body.to_json}"
+  end
+end

data/bin/handler-graphite-notify.rb

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details
+#
+# This will send a 1 to a Graphite metric when an event is created and 0 when it's resolved
+# See http://imansson.wordpress.com/2012/11/26/why-sensu-is-a-monitoring-router-some-cool-handlers/
+#
+# Config by default is graphite_notify, but can be called with a specific json config
+# using the -j option. This allows multiple graphite handlers to be configured.
+
+require 'sensu-handler'
+require 'simple-graphite'
+
+class Resolve < Sensu::Handler
+  option :json_config,
+         description: 'Config Name',
+         short: '-j JsonConfig',
+         long: '--json_config JsonConfig',
+         required: false,
+         default: 'graphite_notify'
+
+  def handle
+    json_config = config[:json_config]
+    port = settings[json_config]['port'] ? settings[json_config]['port'].to_s : '2003'
+    graphite = Graphite.new(host: settings[json_config]['host'], port: port)
+    return unless graphite
+
+    prop = @event['action'] == 'create' ? 1 : 0
+    message = "#{settings[json_config]['prefix']}.#{@event['client']['name'].tr('.', '_')}.#{@event['check']['name']}"
+    message += " #{prop} #{graphite.time_now + rand(100)}"
+    begin
+      graphite.push_to_graphite do |graphite_socket|
+        graphite_socket.puts message
+      end
+    rescue Errno::ETIMEDOUT
+      error_msg = "Can't connect to #{settings[json_config]['host']}:#{port} and send message #{message}'"
+      raise error_msg
+    end
+  end
+end

data/bin/handler-graphite-occurrences.rb

@@ -0,0 +1,39 @@
+#!/usr/bin/env ruby
+#
+# Copyright 2013 vimov, LLC. <a.gameel@vimov.com>
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+
+require 'sensu-handler'
+
+class GraphiteOccurrences < Sensu::Handler
+  # override filters from Sensu::Handler. not appropriate for metric handlers
+  def filter; end
+
+  def handle
+    hostname = @event['client']['name']
+    # #YELLOW
+    check_name = @event['check']['name'].gsub(%r{[ \.]}, '_')
+    value = @event['action'] == 'create' ? @event['occurrences'] : 0
+    now = Time.now.to_i
+
+    # Get Graphite-like format for Sensu events here
+    check_occurrences = "sensu.#{hostname}.#{check_name} #{value} #{now}"
+
+    graphite_server = settings['graphite']['server']
+    graphite_port = settings['graphite']['port']
+
+    begin
+      timeout(3) do
+        sock = TCPSocket.new(graphite_server, graphite_port)
+        sock.puts check_occurrences
+        sock.close
+      end
+    rescue Timeout::Error
+      puts 'Graphite -- timed out while sending check occurrence'
+    rescue => e
+      puts "Graphite -- failed to send check occurrence: #{e}"
+    end
+  end
+end

data/bin/handler-graphite-status.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details
+#
+# This will send the check status (0,1,2,3) to a graphite metric when a check event state changes
+# Based on handler-graphite-notify.rb.
+# Config by default is graphite_status but can be called with a specific json config
+# using the -j option. This allows multiple graphite handlers to be configured.
+
+require 'sensu-handler'
+require 'simple-graphite'
+
+class Resolve < Sensu::Handler
+  option :json_config,
+         description: 'Config Name',
+         short: '-j JsonConfig',
+         long: '--json_config JsonConfig',
+         required: false,
+         default: 'graphite_status'
+  # override filters from Sensu::Handler. not appropriate for metric handlers
+  def filter; end
+
+  def handle
+    json_config = config[:json_config]
+    port = settings[json_config]['port'] ? settings[json_config]['port'].to_s : '2003'
+    graphite = Graphite.new(host: settings[json_config]['host'], port: port)
+    return unless graphite
+
+    prop = @event['check']['status']
+    message = "#{settings[json_config]['prefix']}.#{@event['client']['name'].tr('.', '_')}.#{@event['check']['name']}"
+    message += " #{prop} #{graphite.time_now + rand(100)}"
+    begin
+      graphite.push_to_graphite do |graphite_socket|
+        graphite_socket.puts message
+      end
+    rescue Errno::ETIMEDOUT
+      error_msg = "Can't connect to #{settings[json_config]['host']}:#{port} and send message #{message}'"
+      raise error_msg
+    end
+  end
+end

data/bin/mutator-graphite.rb

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+#
+# Graphite
+# ===
+#
+# DESCRIPTION:
+#   This mutator is an extension of the OnlyCheckOutput mutator, but
+#   modified for Graphite metrics. This mutator only sends event output
+#   (so you don't need to use OnlyCheckOutput) and it also modifies
+#   the format of the hostname in the output if present.
+#
+#   Note however that using this mutator as a mutator command can be very
+#   expensive, as Sensu has to spawn a new Ruby process to launch this script
+#   for each result of a metrics check. Consider instead producing the correct
+#   metric names from your plugin and sending them directly to Graphite via the
+#   socket handler.
+#   See https://groups.google.com/d/msg/sensu-users/1hkRSvL48ck/8Dhl98lR24kJ
+#   for more information.
+#
+# OUTPUT:
+#   Sensu event output with all dots changed to underlines in host name
+#   If -r or --reverse parameter given script put hostname in reverse order
+#   for better Graphite tree view
+#
+# PLATFORM:
+#   all
+#
+# DEPENDENCIES:
+#   none
+#
+# Copyright 2013 Peter Kepes <https://github.com/kepes>
+#
+# Released under the same terms as Sensu (the MIT license); see LICENSE
+# for details.
+require 'json'
+
+# parse event
+event = JSON.parse(STDIN.read, symbolize_names: true)
+
+if ARGV[0] == '-r' || ARGV[0] == '--reverse'
+  puts event[:check][:output].gsub(event[:client][:name], event[:client][:name].split('.').reverse.join('.'))
+else
+  puts event[:check][:output].gsub(event[:client][:name], event[:client][:name].tr('.', '_'))
+end

data/lib/sensu-plugins-graphite.rb

@@ -0,0 +1 @@
+require 'sensu-plugins-graphite/version'

data/lib/sensu-plugins-graphite/graphite_proxy/options.rb

@@ -0,0 +1,119 @@
+module SensuPluginsGraphite
+  module GraphiteProxy
+    module Options
+      def self.included(base)
+        add_default_options(base)
+      end
+
+      def self.add_default_options(base)
+        default_options.each do |name, vals|
+          base.send(:option, name, vals)
+        end
+      end
+
+      def self.default_options
+        {
+          target: {
+            description: 'Graphite data target',
+            short: '-t TARGET',
+            long: '--target TARGET',
+            required: true
+          },
+
+          server: {
+            description: 'Server host and port',
+            short: '-s SERVER:PORT',
+            long: '--server SERVER:PORT',
+            required: true
+          },
+
+          username: {
+            description: 'Username for basic HTTP authentication',
+            short: '-u USERNAME',
+            long: '--user USERNAME',
+            required: false
+          },
+
+          password: {
+            description: 'User password for basic HTTP authentication',
+            short: '-p PASSWORD',
+            long: '--pass PASSWORD',
+            required: false
+          },
+
+          passfile: {
+            description: 'Password file path for basic HTTP authentication',
+            short: '-P PASSWORDFILE',
+            long: '--passfile PASSWORDFILE',
+            required: false
+          },
+
+          no_ssl_verify: {
+            description: 'Do not verify SSL certs',
+            short: '-v',
+            long: '--nosslverify'
+          },
+
+          help: {
+            description: 'Show this message',
+            short: '-h',
+            long: '--help'
+          },
+
+          auth: {
+            description: 'Add an auth token to the HTTP request, in the form of "Name: Value",
+                                             e.g. --auth yourapitokenvaluegoeshere',
+            short: '-A TOKEN',
+            long: '--auth TOKEN'
+          },
+
+          name: {
+            description: 'Name used in responses',
+            short: '-n NAME',
+            long: '--name NAME',
+            default: 'graphite check'
+          },
+
+          hostname_sub: {
+            description: 'Character used to replace periods (.) in hostname (default: _)',
+            long: '--host-sub CHARACTER'
+          },
+
+          from: {
+            description: 'Get samples starting from FROM (default: -10mins)',
+            short: '-f FROM',
+            long: '--from FROM',
+            default: '-10mins'
+          },
+
+          until: {
+            description: 'Get samples ending at UNTIL (default: now)',
+            short: '-l UNTIL',
+            long: '--until UNTIL',
+            default: 'now'
+          },
+
+          warning: {
+            description: 'Generate warning if number of hosts is below received value',
+            short: '-w VALUE',
+            long: '--warn VALUE',
+            proc: proc(&:to_f)
+          },
+
+          critical: {
+            description: 'Generate critical if number of hosts is below received value',
+            short: '-c VALUE',
+            long: '--critical VALUE',
+            proc: proc(&:to_f)
+          },
+
+          below: {
+            description: 'Alert if number of hosts is below specified thresholds',
+            short: '-b',
+            long: '--below'
+          }
+        }
+      end
+    end
+  end
+end

data/lib/sensu-plugins-graphite/graphite_proxy/proxy.rb

@@ -0,0 +1,106 @@
+require 'open-uri'
+
+module SensuPluginsGraphite
+  module GraphiteProxy
+    class ProxyError < StandardError
+      def initialize(msg)
+        super msg
+      end
+    end
+
+    class Proxy
+      attr_accessor :config
+
+      def initialize(config)
+        self.config = config
+      end
+
+      def formatted_target
+        if config[:target].include?('$')
+          require 'socket'
+          formatted = Socket.gethostbyname(Socket.gethostname).first.gsub('.', config[:hostname_sub] || '_')
+          config[:target].gsub('$', formatted)
+        else
+          CGI.escape config[:target]
+        end
+      end
+
+      def request_auth_options(given_opts)
+        url_opts = {}
+
+        url_opts[:ssl_verify_mode] = OpenSSL::SSL::VERIFY_NONE if given_opts[:no_ssl_verify]
+
+        if given_opts[:username]
+          pass = derive_password(given_opts)
+          url_opts[:http_basic_authentication] = [given_opts[:username], pass.chomp]
+        end # we don't have both username and password trying without
+
+        url_opts['Authorization'] = "Bearer #{given_opts[:auth]}" if given_opts[:auth]
+
+        url_opts
+      end
+
+      def derive_password(given_opts)
+        if given_opts[:passfile]
+          File.open(given_opts[:passfile]).readline
+        elsif given_opts[:password]
+          given_opts[:password]
+        end
+      end
+
+      def format_output(data)
+        output = {}
+
+        data.each do |raw|
+          unless raw['datapoints'].empty?
+            line = output_line(raw)
+            output[line['target']] = line
+          end
+        end
+        output
+      end
+
+      def output_line(raw)
+        raw['datapoints'].delete_if { |v| v.first.nil? }
+        target = raw['target']
+        data = raw['datapoints'].map(&:first)
+        start = raw['datapoints'].first.last
+        dend = raw['datapoints'].last.last
+        step = ((dend - start) / raw['datapoints'].size.to_f).ceil
+
+        { 'target' => target, 'data' => data, 'start' => start, 'end' => dend, 'step' => step }
+      end
+
+      # grab data from graphite
+      def retrieve_data!
+        unless @raw_data
+          begin
+            unless config[:server].start_with?('https://', 'http://')
+              config[:server].prepend('http://')
+            end
+
+            url = "#{config[:server]}/render?format=json&target=#{formatted_target}&from=#{config[:from]}"
+
+            handle = open(url, request_auth_options(config))
+
+            @raw_data = handle.gets
+            json_data = JSON.parse(@raw_data)
+            format_output(json_data)
+          rescue OpenURI::HTTPError
+            raise ProxyError, 'Failed to connect to Graphite server'
+          rescue NoMethodError, JSON::ParserError
+            raise ProxyError, 'No data for time period and/or target'
+          rescue Errno::ECONNREFUSED
+            raise ProxyError, 'Connection refused when connecting to Graphite server'
+          rescue Errno::ECONNRESET
+            raise ProxyError, 'Connection reset by peer when connecting to Graphite server'
+          rescue EOFError
+            raise ProxyError, 'End of file error when reading from Graphite server'
+          rescue => e
+            raise ProxyError, "An unknown error occurred: #{e.inspect}"
+          end
+        end
+      end
+    end
+  end
+end