Sensu Flapjack extension compatible with Sensu 0.25.x

flapjack.rb

# Sends events to Flapjack for notification routing. See http://flapjack.io/
#
# This extension requires Flapjack >= 0.8.7 and Sensu >= 0.13.1
#
# In order for Flapjack to keep its entities up to date, it is necssary to set
# metric to "true" for each check that is using the flapjack handler extension.
#
# Here is an example of what the Sensu configuration for flapjack should
# look like, assuming your Flapjack's redis service is running on the
# same machine as the Sensu server:
#
# {
#   "flapjack": {
#      "host": "localhost",
#      "port": 6379,
#      "db": "0"
#   }
# }
#
# Copyright 2014 Jive Software and contributors.
#
# Released under the same terms as Sensu (the MIT license); see LICENSE for details.

require 'sensu/redis'

module Sensu
  module Extension
    class Flapjack < Bridge
      def name
        'flapjack'
      end

      def description
        'sends sensu events to the flapjack redis queue'
      end

      def options
        return @options if @options
        @options = {
          host: '127.0.0.1',
          port: 6379,
          channel: 'events',
          db: 0
        }
        if @settings[:flapjack].is_a?(Hash)
          @options.merge!(@settings[:flapjack])
        end
        @options
      end

      def definition
        {
          type: 'extension',
          name: name,
          mutator: 'ruby_hash'
        }
      end

      def post_init
        Redis.connect(options) do |connection|
          @redis = connection
          @redis.on_error do |_error|
            @logger.warn('Flapjack Redis instance not available on ' + options[:host])
          end
          @redis.before_reconnect do
            @logger.warn("reconnecting to redis")
            pause
          end
          @redis.after_reconnect do
            @logger.info("reconnected to redis")
            resume
          end
          yield(@redis) if block_given?
        end
      end

      def run(event)
        client = event[:client]
        check = event[:check]
        tags = []
        tags.concat(client[:tags]) if client[:tags].is_a?(Array)
        tags.concat(check[:tags]) if check[:tags].is_a?(Array)
        tags << client[:environment] unless client[:environment].nil?
        # #YELLOW
        unless check[:subscribers].nil? || check[:subscribers].empty? # rubocop:disable UnlessElse
          tags.concat(client[:subscriptions] - (client[:subscriptions] - check[:subscribers]))
        else
          tags.concat(client[:subscriptions])
        end
        details = ['Address:' + client[:address]]
        details << 'Tags:' + tags.join(',')
        details << "Raw Output: #{check[:output]}" if check[:notification]
        flapjack_event = {
          entity: client[:name],
          check: check[:name],
          type: 'service',
          state: Sensu::SEVERITIES[check[:status]] || 'unknown',
          summary: check[:notification] || check[:output],
          details: details.join(' '),
          time: check[:executed],
          tags: tags
        }
        flapjack_event.merge!(initial_failure_delay: check[:initial_failure_delay]) if check[:initial_failure_delay]
        flapjack_event.merge!(repeat_failure_delay: check[:repeat_failure_delay]) if check[:repeat_failure_delay]
        @redis.lpush(options[:channel], Sensu::JSON.dump(flapjack_event))
        yield 'sent an event to the flapjack redis queue', 0
      end
    end
  end
end