yieldbot/sensu-yieldbot-plugins

#! /usr/bin/env ruby
#
# metrics-bongo.rb
#
# DESCRIPTION:
#
#
# OUTPUT:
#   JSON
#
# PLATFORMS:
#   Linux
#
# DEPENDENCIES:
#   gem: sensu-plugin
#   gem: rest-client
#
# USAGE:
#   #YELLOW
#
# NOTES:
#
# LICENSE:
#   Copyright (c) 2015,Yieldbot <devops@yieldbot.com>
#   Released under the same terms as Sensu (the MIT license); see LICENSE
#   for details.
#

require 'sensu-plugin/metric/cli'
require 'json'
require 'rest-client'

#
# Get a set of metrics from an app running in Mesos
#
class MesosAppMetrics < Sensu::Plugin::Metric::CLI::Statsd
  option :server,
         description: 'The consul dns name of the mesos server',
         short: '-s SERVER',
         long: '--server SERVER'

  option :port,
         short: '-p PORT',
         long: '--port PORT',
         description: 'The port used to query mesos'

  option :app,
         short: '-a APP',
         long: '--app APP',
         description: 'The name of the app to get metrics from'

 option :scheme,
        description: 'Metric naming scheme, text to prepend to .$parent.$child',
        long: '--scheme SCHEME'

  # Acquire the slave that a particular app is running on
  #
  def acquire_app_slave
    # consul dns should always be used
    server = config[:server] || 'us-east-1-perpetuum.mesos-marathon.service.consul'
    # the default port is the one supplied by consul
    port   = config[:port] || '443'
    app    = config[:app]
    @failures = []

    # break out if the client fails to connect to the mesos master
    begin
      r = RestClient::Resource.new("https://#{server}:#{port}/v2/apps/#{app}", timeout: 10, verify_ssl: false).get
    rescue Errno::ECONNREFUSED, RestClient::ResourceNotFound, SocketError
      critical "#{server} connection was refused"
    rescue RestClient::RequestTimeout
      critical "#{server} connection timed out"
    end
    JSON.parse(r)['app']['tasks'][0]['host']
  end

  # reterive the metrics from the app
  #
  def acquire_metrics(current_slave)
    JSON.parse(`curl -s -k http://#{current_slave}:31550/v1/kafka/metrics`)
  end

  def run
    timestamp = Time.now.to_i
    current_slave = acquire_app_slave
    acquire_metrics(current_slave).each do |key, value|
      output [config[:scheme], key].join('.'), value, timestamp
    end
    ok
  end
end