check-disk-io 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 511a2d6021d61c83dff4ea328072133b28189630
+  data.tar.gz: 7bbbb4f7969e7067c798bddb7cc864922ca5b05c
+SHA512:
+  metadata.gz: df704c6954867e6a764944ac0ab6126c4caf590a2eab8e0a5a2d088f9e060f6a89983213c3222f49812839bf623e105a83514840e6d1989e849d59d8b5f4bf6d
+  data.tar.gz: 3d02a9c7285b8a5c0f016fe03d65f0eedfc0602bac08cbf39cc071186f33c92cf3fff1ad0ceb18e23cff236962e80e6b11571b789029e825f0ed6848771fc0cc

data/lib/check-disk-io_gem.rb

@@ -0,0 +1,150 @@
+#!/opt/sensu/embedded/bin/ruby
+#
+#  encoding: UTF-8
+#   check-disk-io.rb
+#
+# DESCRIPTION:
+#   Check disks io and alert if over threshold
+#
+# OUTPUT:
+#   plain text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#
+# USAGE:
+#
+# ./check-disk-io.rb -w 1000 -c 2000
+#
+#
+# LICENSE:
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+require 'sensu-plugin/check/cli'
+require 'ostruct'
+
+class Diskstats
+  include Enumerable
+
+  COLUMNS = [
+    :major,
+    :minor,
+    :device_name,
+    :reads_success,
+    :reads_merged,
+    :sectors_read,
+    :read_time_ms,
+    :writes_success,
+    :writes_merged,
+    :sectors_written,
+    :write_time_ms,
+    :current_ios,
+    :io_time_ms,
+    :weightd_io_time_ms
+  ]
+
+  attr_reader :raw_stats, :now, :later
+
+  def initialize(stats)
+    @raw_stats = stats
+  end
+
+  def rows
+    @rows ||= raw_stats.lines.map do |line|
+      OpenStruct.new(Hash[COLUMNS.zip(line.strip.split)])
+    end
+  end
+
+  def each(&block)
+    rows.each(&block)
+  end
+
+  def self.from_proc
+    new(File.read('/proc/diskstats'))
+  end
+end
+
+class CheckDiskIO < Sensu::Plugin::Check::CLI
+
+  check_name 'check_disk_io'
+
+  option :warning,
+    short: '-w WARNING',
+    long: '--warning WARNING',
+    description: 'Warning threshold',
+    default: 3000
+
+  option :critical,
+    short: '-c CRITICAL',
+    long: '--critical CRITICAL',
+    description: 'Critical threshold',
+    default: 4000
+
+  def run
+    disks = Diskstats.from_proc.reject { |s| s.device_name =~ /^sr0|fd0|ram|loop|dm-/ }.map(&:device_name)
+    counters = Hash[disks.each_with_object({}).to_a]
+    for i in 0..1 do
+      disks.each do |disk|
+        new_hash = {
+          disk => {
+            i => {
+              timestamp: Time.new.to_i,
+              writes_success: Diskstats.from_proc.select { |s| s.device_name == disk }.map(&:writes_success)
+            }
+          }
+        }
+        counters[disk] = counters[disk].merge(new_hash[disk])
+        sleep 3
+      end
+    end
+
+    alerts = []
+
+    counters.each do |counter|
+
+      counter[1].keys.sort.each_with_index do |key, index|
+        if index == counter[1].length - 1
+          last_writes_success = counter[1][key][:writes_success]
+          first_writes_success = counter[1][0][:writes_success]
+          result = last_writes_success.join.to_i - first_writes_success.join.to_i
+
+          device = counter[0]
+
+          if result >= config[:critical].to_i && result > config[:warning].to_i
+            alerts << [device, 'critical', result]
+          end
+          if result >= config[:warning].to_i && result < config[:critical].to_i
+            alerts << [device, 'warning', result]
+          end
+        end
+      end
+    end
+
+    if alerts.empty?
+      message 'Everything is ok'
+      ok
+    else
+      status = ''
+      messages = ''
+
+      alerts.each do |alert|
+        messages << "\r\n/dev/#{alert[0]} has reached #{alert[2]} IOPS"
+        if alert[1] == 'critical' && status.empty?
+          status = 'critical'
+        end
+      end
+
+      message messages
+
+      critical if status == 'critical'
+      warning if status != 'critical'
+    end
+
+  rescue => err
+    puts "Exception: #{err}"
+    err
+  end
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: check-disk-io
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Maxime Devalland
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sensu-plugin
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+description: 
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/check-disk-io_gem.rb
+homepage: 
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.11
+signing_key: 
+specification_version: 4
+summary: check disk io
+test_files: []