sensu-plugins-mesos 0.1.1 → 1.0.0

data/bin/check-mesos-mem-balance.rb

@@ -0,0 +1,154 @@
+#! /usr/bin/env ruby
+#
+#   check-mesos-mem-balance
+#
+# DESCRIPTION:
+#   This plugin checks that there is less Memory imbalance than specified on a certain mesos cluster
+#
+# OUTPUT:
+#   plain text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#   gem: rest-client
+#   gem: json
+#
+# USAGE:
+#   #YELLOW
+#
+# NOTES:
+#
+# LICENSE:
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'rest-client'
+require 'json'
+
+class MesosMemBalanceCheck < Sensu::Plugin::Check::CLI
+  check_name 'MesosMemBalanceCheck'
+  @metrics_name = 'slaves'.freeze
+  CHECK_TYPE = 'mem'.freeze
+
+  class << self
+    attr_reader :metrics_name
+  end
+
+  option :server,
+         description: 'Mesos server',
+         short: '-s SERVER',
+         long: '--server SERVER',
+         default: 'localhost'
+
+  option :port,
+         description: 'port (default 5050)',
+         short: '-p PORT',
+         long: '--port PORT',
+         default: 5050,
+         required: false
+
+  option :uri,
+         description: 'Endpoint URI',
+         short: '-u URI',
+         long: '--uri URI',
+         default: '/master/slaves'
+
+  option :timeout,
+         description: 'timeout in seconds',
+         short: '-t TIMEOUT',
+         long: '--timeout TIMEOUT',
+         proc: proc(&:to_i),
+         default: 5
+
+  option :crit,
+         description: 'Critical value to check against',
+         short: '-c VALUE',
+         long: '--critical VALUE',
+         proc: proc(&:to_i),
+         default: 0,
+         required: false
+
+  option :warn,
+         description: 'Warning value to check against',
+         short: '-w VALUE',
+         long: '--warning VALUE',
+         proc: proc(&:to_i),
+         default: 0,
+         required: false
+
+  def run
+    if config[:crit] < 0 || config[:warn] < 0
+      unknown "Thresholds cannot be negative, crit: #{config[:crit]}, warn: #{config[:warn]}"
+    end
+
+    server = config[:server]
+    port = config[:port]
+    uri = config[:uri]
+    timeout = config[:timeout]
+    crit = config[:crit]
+    warn = config[:warn]
+
+    begin
+      server = get_leader_url server, port
+      r = RestClient::Resource.new("#{server}#{uri}", timeout).get
+      compare = get_check_diff(get_slaves(r))
+      if compare['diff'] >= crit
+        critical "There is a Memory usage diff of #{compare['diff']} bigger than #{crit} " + compare['msg']
+      end
+      if compare['diff'] >= warn
+        warning "There is a Memory usage diff of #{compare['diff']} bigger than #{warn} " + compare['msg']
+      end
+    rescue Errno::ECONNREFUSED, RestClient::ResourceNotFound, SocketError
+      unknown  "Mesos #{server} is not responding"
+    rescue RestClient::RequestTimeout
+      unknown  "Mesos #{server} connection timed out"
+    end
+    ok
+  end
+
+  # Redirects server call to discover the Leader
+  # @param server [String] Server address
+  # @param port [Number] api port
+  # @return [Url] Url representing the Leader
+
+  def get_leader_url(server, port)
+    RestClient::Resource.new("http://#{server}:#{port}/redirect").get.request.url
+  end
+
+  # Parses JSON data as returned from Mesos's metrics API
+  # @param data [String] Server response
+  # @return [Integer] Number of failed tasks in Mesos
+  def get_slaves(data)
+    begin
+      slaves = JSON.parse(data)[MesosMemBalanceCheck.metrics_name]
+    rescue JSON::ParserError
+      raise "Could not parse JSON response: #{data}"
+    end
+
+    if slaves.nil?
+      raise "No metrics for [#{MesosMemBalanceCheck.metrics_name}] in server response: #{data}"
+    end
+
+    slaves
+  end
+
+  def get_check_diff(slavelist)
+    begin
+      usages = {}
+      check_diff = {}
+      slavelist.each do |slaveinfo|
+        usages.store(slaveinfo['hostname'], slaveinfo['used_resources'][CHECK_TYPE] * 100 / slaveinfo['resources'][CHECK_TYPE])
+      end
+      sorted = usages.sort_by { |_hostname, total| total }
+      max = usages.length - 1
+      check_diff['diff'] = sorted[max][1] - sorted[0][1]
+      check_diff['msg'] = "Hostname #{sorted[0][0]} uses #{sorted[0][1]}% and Hostname #{sorted[max][0]} uses #{sorted[max][1]}%"
+    end
+    check_diff
+  end
+end

data/bin/check-mesos-running-tasks.rb

@@ -0,0 +1,182 @@
+#! /usr/bin/env ruby
+#
+#   check-mesos-running-tasks
+#
+# DESCRIPTION:
+#   This plugin checks that there are running tasks on a mesos cluster
+#
+# OUTPUT:
+#   plain text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#   gem: rest-client
+#   gem: json
+#
+# USAGE:
+#   #YELLOW
+#
+# NOTES:
+#
+# LICENSE:
+#   Copyright 2016, Oskar Flores (oskar.flores@gmail.com)
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'rest-client'
+require 'json'
+require 'daybreak'
+
+class MesosRunningTaskCheck < Sensu::Plugin::Check::CLI
+  check_name 'CheckMesosRunningTask'
+  @metrics_name = 'master/tasks_running'.freeze
+
+  class << self
+    attr_reader :metrics_name
+  end
+
+  option :server,
+         description: 'Mesos server',
+         short: '-s SERVER',
+         long: '--server SERVER',
+         default: 'localhost'
+
+  option :port,
+         description: 'port (default 5050)',
+         short: '-p PORT',
+         long: '--port PORT',
+         default: 5050,
+         required: false
+
+  option :uri,
+         description: 'Endpoint URI',
+         short: '-u URI',
+         long: '--uri URI',
+         default: '/metrics/snapshot'
+
+  option :timeout,
+         description: 'timeout in seconds',
+         short: '-t TIMEOUT',
+         long: '--timeout TIMEOUT',
+         proc: proc(&:to_i),
+         default: 5
+
+  option :mode,
+         description: 'eq ne lt gt or rg',
+         short: '-m MODE',
+         long: '--mode MODE',
+         required: true
+
+  option :min,
+         description: 'min value on range',
+         short: '-l VALUE',
+         long: '--low VALUE',
+         required: false,
+         proc: proc(&:to_i),
+         derfault: 0
+
+  option :max,
+         description: 'max value on range',
+         short: '-h VALUE',
+         long: '--high VALUE',
+         required: false,
+         proc: proc(&:to_i),
+         default: 1
+
+  option :value,
+         description: 'value to check against',
+         short: '-v VALUE',
+         long: '--value VALUE',
+         proc: proc(&:to_i),
+         default: 0,
+         required: false
+
+  option :delta,
+         short: '-d',
+         long: '--delta',
+         description: 'Use this flag to compare the metric with the previously retrieved value',
+         boolean: true
+
+  def run
+    port = config[:port]
+    uri = config[:uri]
+    timeout = config[:timeout]
+    mode = config[:mode]
+    value = config[:value]
+    server = config[:server]
+    min = config[:min]
+    max = config[:max]
+
+    begin
+      server = get_leader_url server, port
+      r = RestClient::Resource.new("#{server}#{uri}", timeout).get
+      metric_value = check_tasks(r)
+      check_mesos_tasks(metric_value, mode, value, min, max)
+    rescue Errno::ECONNREFUSED, RestClient::ResourceNotFound, SocketError
+      unknown  "Mesos #{server} is not responding"
+    rescue RestClient::RequestTimeout
+      unknown  "Mesos #{server} connection timed out"
+    end
+    ok "Found #{metric_value} tasks running"
+  end
+
+  # Redirects server call to discover the Leader
+  # @param server [String] Server address
+  # @param port [Number] api port
+  # @return [Url] Url representing the Leader
+
+  def get_leader_url(server, port)
+    RestClient::Resource.new("http://#{server}:#{port}/redirect").get.request.url
+  end
+
+  # Parses JSON data as returned from Mesos  API
+  # @param data [String] Server response
+  # @return [Numeric] Number of running tasks
+
+  def check_tasks(data)
+    begin
+      running_tasks = JSON.parse(data)[MesosRunningTaskCheck.metrics_name]
+    rescue JSON::ParserError
+      raise "Could not parse JSON response: #{data}"
+    end
+
+    if running_tasks.nil?
+      raise "No tasks in server response: #{data}"
+    end
+
+    running_tasks.round
+  end
+
+  def check_mesos_tasks(metric_value, mode, value, min, max)
+    if config[:delta]
+      db = Daybreak::DB.new '/tmp/mesos-metrics.db', default: 0
+      prev_value = db['task_running']
+      db.lock do
+        db['task_running'] = metric_value
+      end
+      metric_value -= prev_value
+      db.flush
+      db.compact
+      db.close
+    end
+    case mode
+    when 'eq'
+      critical "The number of running tasks cluster is equal to #{value}!" if metric_value.equal? value
+    when 'ne'
+      critical "The number of running tasks cluster is not equal to #{value}!" if metric_value != value
+    when 'lt'
+      critical "The number of running tasks cluster is lower than #{value}!" if metric_value < value
+    when 'gt'
+      critical "The number of running tasks cluster is greater than #{value}!" if metric_value > value
+    when 'rg'
+      unless (min.to_i..max.to_i).cover? metric_value
+        critical "The number of running tasks in cluster is not in #{min} - #{max} value range!"
+      end
+    end
+  end
+end

data/bin/check-mesos.rb

@@ -29,10 +29,6 @@
 require 'sensu-plugin/check/cli'
 require 'rest-client'
 
-# Mesos default ports are defined here: http://mesos.apache.org/documentation/latest/configuration
-MASTER_DEFAULT_PORT = '5050'.freeze
-SLAVE_DEFAULT_PORT = '5051'.freeze
-
 class MesosNodeStatus < Sensu::Plugin::Check::CLI
   option :server,
          description: 'Mesos servers, comma separated',
@@ -40,18 +36,19 @@ class MesosNodeStatus < Sensu::Plugin::Check::CLI
          long: '--server SERVER1,SERVER2,...',
          default: 'localhost'
 
-  option :mode,
-         description: 'master or slave',
-         short: '-m MODE',
-         long: '--mode MODE',
-         required: true
-
   option :port,
-         description: "port (default #{MASTER_DEFAULT_PORT} for master, #{SLAVE_DEFAULT_PORT} for slave)",
+         description: 'port (default 5050, use 5051 for slaves)',
          short: '-p PORT',
          long: '--port PORT',
+         default: 5050,
          required: false
 
+  option :uri,
+         description: 'Endpoint URI',
+         short: '-u URI',
+         long: '--uri URI',
+         default: '/health'
+
   option :timeout,
          description: 'timeout in seconds',
          short: '-t TIMEOUT',
@@ -60,31 +57,24 @@ class MesosNodeStatus < Sensu::Plugin::Check::CLI
          default: 5
 
   def run
-    mode = config[:mode]
     servers = config[:server]
-    case mode
-    when 'master'
-      port = config[:port] || MASTER_DEFAULT_PORT
-      uri = '/master/health'
-    when 'slave'
-      port = config[:port] || SLAVE_DEFAULT_PORT
-      uri = '/slave(1)/health'
-    end
+    uri = config[:uri]
+    port = config[:port]
     failures = []
     servers.split(',').each do |server|
       begin
         r = RestClient::Resource.new("http://#{server}:#{port}#{uri}", timeout: config[:timeout]).get
         if r.code != 200
-          failures << "#{config[:mode]} on #{server} is not responding"
+          failures << "Mesos on #{server} is not responding"
         end
       rescue Errno::ECONNREFUSED, RestClient::ResourceNotFound, SocketError
-        failures << "Mesos #{mode} on #{server} is not responding"
+        failures << "Mesos on #{server} is not responding"
       rescue RestClient::RequestTimeout
-        failures << "Mesos #{mode} on #{server} connection timed out"
+        failures << "Mesos on #{server} connection timed out"
       end
     end
     if failures.empty?
-      ok "Mesos #{mode} is running on #{servers}"
+      ok "Mesos is running on #{servers}"
     else
       critical failures.join("\n")
     end

data/bin/check-metronome.rb

@@ -0,0 +1,80 @@
+#! /usr/bin/env ruby
+#
+#   check-metronome
+#
+# DESCRIPTION:
+#   This plugin checks that Metronome can query the existing job graph.
+#
+# OUTPUT:
+#   plain text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#   gem: rest-client
+#
+# USAGE:
+#
+#
+# NOTES:
+#
+# LICENSE:
+#   Copyright 2017, PTC (www.ptc.com)
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'rest-client'
+
+class MetronomeNodeStatus < Sensu::Plugin::Check::CLI
+  option :server,
+         description: 'Metronome hosts, comma separated',
+         short: '-s SERVER',
+         long: '--server SERVER',
+         default: 'localhost'
+
+  option :port,
+         description: 'Metronome port',
+         short: '-p PORT',
+         long: '--port PORT',
+         default: '9942'
+
+  option :uri,
+         description: 'Endpoint URI',
+         short: '-u URI',
+         long: '--uri URI',
+         default: '/v1/jobs'
+
+  option :timeout,
+         description: 'timeout in seconds',
+         short: '-t TIMEOUT',
+         long: '--timeout TIMEOUT',
+         proc: proc(&:to_i),
+         default: 5
+
+  def run
+    servers = config[:server]
+    uri = config[:uri]
+    failures = []
+    servers.split(',').each do |server|
+      begin
+        r = RestClient::Resource.new("http://#{server}:#{config[:port]}#{uri}", timeout: config[:timeout]).get
+        if r.code != 200
+          failures << "Metronome on #{server} is not responding"
+        end
+      rescue Errno::ECONNREFUSED, RestClient::ResourceNotFound, SocketError
+        failures << "Metronome on #{server} is not responding"
+      rescue RestClient::RequestTimeout
+        failures << "Metronome on #{server} connection timed out"
+      end
+    end
+    if failures.empty?
+      ok "Metronome is running on #{servers}"
+    else
+      critical failures.join("\n")
+    end
+  end
+end