sensu-plugins-aws-boutetnico 1.0.6

data/bin/check-emr-cluster.rb

@@ -0,0 +1,144 @@
+#! /usr/bin/env ruby
+#
+# check-emr-cluster
+#
+# DESCRIPTION:
+#   This plugin checks if a cluster exists.
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: aws-sdk
+#   gem: sensu-plugin
+#
+# USAGE:
+#   ./check-emr-cluster.rb --cluster-name MyCluster --aws-region eu-west-1 --use-iam --warning-over 14400 --critical-over 21600
+#
+# NOTES:
+#
+# LICENSE:
+#   Copyright (c) 2015, Olivier Bazoud, olivier.bazoud@gmail.com
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'aws-sdk'
+
+class CheckEMRCluster < Sensu::Plugin::Check::CLI
+  option :aws_access_key,
+         short: '-a AWS_ACCESS_KEY',
+         long: '--aws-access-key AWS_ACCESS_KEY',
+         description: "AWS Access Key. Either set ENV['AWS_ACCESS_KEY'] or provide it as an option",
+         default: ENV['AWS_ACCESS_KEY']
+
+  option :aws_secret_access_key,
+         short: '-k AWS_SECRET_KEY',
+         long: '--aws-secret-access-key AWS_SECRET_KEY',
+         description: "AWS Secret Access Key. Either set ENV['AWS_SECRET_KEY'] or provide it as an option",
+         default: ENV['AWS_SECRET_KEY']
+
+  option :aws_region,
+         short: '-r AWS_REGION',
+         long: '--aws-region REGION',
+         description: 'AWS Region (defaults to us-east-1).',
+         default: 'us-east-1'
+
+  option :use_iam_role,
+         short: '-u',
+         long: '--use-iam',
+         description: 'Use IAM role authenticiation. Instance must have IAM role assigned for this to work'
+
+  option :cluster_name,
+         short: '-b CLUSTER_NAME',
+         long: '--cluster-name',
+         description: 'The name of the EMR cluster',
+         required: true
+
+  option :warning_over,
+         description: 'Warn if cluster\'s age is greater than provided age in seconds',
+         short: '-w SECONDS',
+         long: '--warning-over SECONDS',
+         default: -1,
+         proc: proc(&:to_i)
+
+  option :critical_over,
+         description: 'Critical if cluster\'s age is greater than provided age in seconds',
+         short: '-c SECONDS',
+         long: '--critical-over SECONDS',
+         default: -1,
+         proc: proc(&:to_i)
+
+  option :warning_under,
+         description: 'Warn if cluster\'s age is lower than provided age in seconds',
+         short: '-w SECONDS',
+         long: '--warning-under SECONDS',
+         default: -1,
+         proc: proc(&:to_i)
+
+  option :critical_under,
+         description: 'Critical if cluster\'s age is lower than provided age in seconds',
+         short: '-C SECONDS',
+         long: '--critical-under SECONDS',
+         default: -1,
+         proc: proc(&:to_i)
+
+  def aws_config
+    { access_key_id: config[:aws_access_key],
+      secret_access_key: config[:aws_secret_access_key],
+      region: config[:aws_region] }
+  end
+
+  def humanize(secs)
+    [[60, :seconds], [60, :minutes], [24, :hours], [1000, :days]].map do |count, name|
+      if secs > 0
+        secs, n = secs.divmod(count)
+        "#{n.to_i} #{name}"
+      end
+    end.compact.reverse.join(' ')
+  end
+
+  def run
+    aws_config = {}
+    if config[:use_iam_role].nil?
+      aws_config[:access_key_id] = config[:aws_access_key]
+      aws_config[:secret_access_key] = config[:aws_secret_access_key]
+    end
+
+    emr = Aws::EMR::Client.new(aws_config.merge!(region: config[:aws_region]))
+    begin
+      emr_clusters = emr.list_clusters(created_after: Time.now - 24 * 60 * 60, created_before: Time.now).clusters
+      clusters = emr_clusters.select { |c| c.name == config[:cluster_name] }
+
+      critical "EMR cluster #{config[:cluster_name]} appears #{clusters.size} times" if clusters.size > 1
+      critical "EMR cluster #{config[:cluster_name]} not found" if clusters.empty?
+
+      cluster = clusters.first
+      state = cluster.status.state
+      if state == 'TERMINATED_WITH_ERRORS'
+        critical "EMR cluster #{config[:cluster_name]} state is '#{state}'"
+      else
+        creation_date_time = cluster.status.timeline.creation_date_time
+        end_date_time = cluster.status.timeline.end_date_time || Time.now
+        age = end_date_time.to_i - creation_date_time.to_i
+        if age >= config[:critical_over]
+          critical "EMR cluster #{config[:cluster_name]} - #{humanize(age)} vs. #{humanize(config[:critical_over])}"
+        elsif age >= config[:warning_over]
+          warning "EMR cluster #{config[:cluster_name]} - #{humanize(age)} vs. #{humanize(config[:warning_over])}"
+        elsif age <= config[:critical_under] && state == 'TERMINATED'
+          critical "EMR cluster #{config[:cluster_name]} - #{humanize(age)} vs. #{humanize(config[:critical_under])}"
+        elsif age <= config[:warning_under] && state == 'TERMINATED'
+          warning "EMR cluster #{config[:cluster_name]} - #{humanize(age)} vs. #{humanize(config[:warning_under])}"
+        else
+          ok "EMR cluster #{config[:cluster_name]} - #{humanize(age)}"
+        end
+      end
+    rescue StandardError => e
+      critical "EMR cluster #{config[:cluster_name]} - #{e.message}"
+    end
+  end
+end

data/bin/check-emr-steps.rb

@@ -0,0 +1,90 @@
+#! /usr/bin/env ruby
+#
+# check-emr-steps
+#
+# DESCRIPTION:
+#   Alerts on any failed steps for a cluster in the past 10 minutes.
+#
+# OUTPUT:
+#   plain-text
+#
+# DEPENDENCIES:
+#   gem: aws-sdk
+#   gem: sensu-plugin
+#
+# USAGE:
+#   check-emr-steps.rb -r us-west-2 -b 'My Cluster' -t FAILED -c 0
+#
+#   This will alert on any failed steps in the past 10 minutes on the latest cluster
+#   with the name 'My Cluster'.
+# NOTES:
+#
+# LICENSE:
+#   Bryan Absher <bryan.absher@gmail.com>
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+
+require 'sensu-plugins-aws'
+require 'sensu-plugin/check/cli'
+require 'aws-sdk'
+
+class CheckEMRSteps < Sensu::Plugin::Check::CLI
+  include Common
+
+  option :scheme,
+         description: 'Metric naming scheme, text to prepend to metric',
+         short: '-s SCHEME',
+         long: '--scheme SCHEME',
+         default: 'sensu.aws.emr'
+
+  option :aws_region,
+         short: '-r AWS_REGION',
+         long: '--aws-region REGION',
+         description: 'AWS Region (defaults to us-east-1).',
+         default: 'us-east-1'
+
+  option :cluster_name,
+         short: '-b CLUSTER_NAME',
+         long: '--cluster-name',
+         description: 'The name of the EMR cluster',
+         required: true
+
+  option :status,
+         short: '-t STEP_STATUS',
+         long: '--step-status',
+         description: 'Step status to check, [PENDING RUNNING COMPLETED CANCELLED FAILED INTERRUPTED]',
+         default: 'FAILED'
+
+  option :count,
+         short: '-c COUNT',
+         long: '--count',
+         description: 'Max number of steps with this status.',
+         proc: proc(&:to_i),
+         default: 0
+
+  def run
+    emr = Aws::EMR::Client.new(aws_config)
+    begin
+      emr_clusters = emr.list_clusters.clusters
+      clusters = emr_clusters.select { |c| c.name == config[:cluster_name] }
+      critical "EMR cluster #{config[:cluster_name]} not found" if clusters.empty?
+      cluster = clusters.sort_by { |c| c.status.timeline.creation_date_time }.reverse.first
+
+      steps = emr.list_steps(
+        cluster_id: cluster.id,
+        step_states: [config[:status]]
+      ).steps
+
+      messages = []
+      now = Time.new
+      failed = steps.select { |step| now - step.status.timeline.end_date_time < 10 * 60 }
+      if failed.size > config[:count]
+        failed.each_entry { |step| messages << "Step #{step.id} '#{step.name}' has failed on cluster #{cluster.id} '#{cluster.name}'" }
+        if messages.count > 0
+          critical("#{messages.count} #{messages.count > 1 ? 'steps have' : 'step has'} failed: #{messages.join(',')}")
+        end
+      end
+      ok
+    end
+  end
+end

data/bin/check-eni-status.rb

@@ -0,0 +1,110 @@
+#! /usr/bin/env ruby
+#
+# check-eni-status
+#
+# DESCRIPTION:
+#   This plugin checks the status of an elastic network interface
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: aws-sdk-v2
+#   gem: sensu-plugin
+#
+# USAGE:
+#   check-eni-status -e eni -w in-use -r region
+#   check-eni-status -e eni -c available -r region
+#
+# NOTES:
+#
+# LICENSE:
+#   Copyright (c) 2013, Damien DURANT <damien.durant@edifixio.com>
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'sensu-plugins-aws'
+require 'aws-sdk'
+
+#
+# Check ENI Status
+#
+class ENIStatus < Sensu::Plugin::Check::CLI
+  include Common
+
+  option :aws_region,
+         short: '-r AWS_REGION',
+         long: '--aws-region REGION',
+         description: 'AWS Region (defaults to us-east-1).',
+         default: 'us-east-1'
+
+  option :eni,
+         short: '-e ENI_ID',
+         long: '--eni ENI_ID',
+         description: 'A comma seperated list of the Elastic Network Interfaces to check',
+         default: ''
+
+  option :warn_status,
+         short: '-w STATUS',
+         long: '--warn STATUS',
+         description: 'ENI status considered to be a warning',
+         default: ''
+
+  option :crit_status,
+         short: '-c STATUS',
+         long: '--crit STATUS',
+         description: 'ENI status considered to be critical',
+         default: ''
+
+  def aws_config
+    { access_key_id: config[:aws_access_key],
+      secret_access_key: config[:aws_secret_access_key],
+      region: config[:aws_region] }
+  end
+
+  def run
+    Aws.config.update(aws_config)
+    client = Aws::EC2::Client.new
+
+    if config[:eni].empty?
+      critical 'Error, ENI(s) should be specified.'
+    end
+
+    if config[:crit_status].empty?
+      if config[:warn_status].empty?
+        critical 'Error, either crit_status or warn_status must be specified'
+      end
+    end
+
+    warnings = []
+    crits = []
+
+    eni = config[:eni].split(',')
+    eni.each do |e|
+      status = client.describe_network_interfaces(
+        filters: [{ name: 'network-interface-id', values: [e] }]
+      )[:network_interfaces].first
+
+      if status.nil?
+        warnings << "No Information found for #{e}"
+      elsif config[:crit_status].casecmp(status[:status]).zero?
+        crits << "#{e} is #{status[:status]}"
+      elsif config[:warn_status].casecmp(status[:status]).zero?
+        warnings << "#{e} is #{status[:status]}"
+      end
+    end
+
+    if crits.any?
+      critical crits.join(', ').to_s
+    elsif warnings.any?
+      warning warnings.join(', ').to_s
+    else
+      ok "all interfaces(s): #{eni} are OK"
+    end
+  end
+end

data/bin/check-expiring-reservations.rb

@@ -0,0 +1,117 @@
+#! /usr/bin/env ruby
+#
+# expiring-reservations
+#
+# DESCRIPTION:
+#   Alert on expiring reservations of an AWS account.
+#
+# OUTPUT:
+#   plain-text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: aws-sdk
+#   gem: sensu-plugin
+#   gem: sensu-plugins-aws
+#
+# USAGE:
+#   metrics-reservation-utilization.rb
+#
+# NOTES:
+#
+# LICENSE:
+#   Copyright (c) 2019, Nicolas Boutet amd3002@gmail.com
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+#
+
+require 'sensu-plugin/check/cli'
+require 'aws-sdk'
+require 'sensu-plugins-aws'
+
+class CheckExpiringReservations < Sensu::Plugin::Check::CLI
+  include Common
+
+  option :reservation_id,
+         description: 'Reservation id (defaults to all reservations if omitted)',
+         short:       '-R RESERVATION_ID',
+         long:        '--reservation-id RESERVATION_ID',
+         default:     nil
+
+  option :offering_class,
+         description: 'The offering class of the Reserved Instance (standard or convertible)',
+         short:       '-o OFFERING_CLASS',
+         long:        '--offering-class OFFERING_CLASS',
+         default:     'standard'
+
+  option :aws_region,
+         short:       '-r AWS_REGION',
+         long:        '--aws-region REGION',
+         description: 'AWS Region (defaults to us-east-1).',
+         default:     'us-east-1'
+
+  option :warning,
+         short:       '-w N',
+         long:        '--warning VALUE',
+         description: 'Issue a warning if a reservation will expire in under VALUE days',
+         default:     0,
+         proc:        proc(&:to_i)
+
+  option :critical,
+         short:       '-c N',
+         long:        '--critical VALUE',
+         description: 'Issue a critical if a reservation will expire in under VALUE days',
+         default:     1,
+         proc:        proc(&:to_i)
+
+  def run
+    begin
+      client = Aws::EC2::Client.new(aws_config)
+
+      params = {
+        filters: [
+          {
+            name: 'state',
+            values: ['active']
+          }
+        ],
+        offering_class: config[:offering_class]
+      }
+
+      unless config[:reservation_id].nil?
+        params[:reserved_instances_ids] = [config[:reservation_id]]
+      end
+
+      reservations = client.describe_reserved_instances(params)
+
+      warnflag = false
+      critflag = false
+      reportstring = ''
+
+      reservations.reserved_instances.each do |reservation|
+        time_left = (reservation.end - Time.now).abs.to_i / (24 * 60 * 60)
+
+        if time_left <= config[:critical]
+          critflag = true
+          reportstring += " reservation #{reservation.reserved_instances_id} (#{reservation.instance_type} x #{reservation.instance_count}) expires in #{time_left} days;"
+        elsif time_left <= config[:warning]
+          warnflag = true
+          reportstring += " reservation #{reservation.reserved_instances_id} (#{reservation.instance_type} x #{reservation.instance_count}) expires in #{time_left} days;"
+        end
+      end
+
+      if critflag
+        critical reportstring
+      elsif warnflag
+        warning reportstring
+      else
+        ok 'All checked reservations are ok'
+      end
+    rescue StandardError => e
+      critical "Error: exception: #{e}"
+    end
+    ok
+  end
+end