riemann-tools 1.0.0 → 1.1.0

data/bin/riemann-health

@@ -1,349 +1,10 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-# Reports current CPU, disk, load average, and memory use to riemann.
-
-require File.expand_path('../../lib/riemann/tools', __FILE__)
-require File.expand_path('../../lib/riemann/tools/utils', __FILE__)
-
-class Riemann::Tools::Health
-  include Riemann::Tools
-  include Riemann::Tools::Utils
-
-  opt :cpu_warning, "CPU warning threshold (fraction of total jiffies)", :default => 0.9
-  opt :cpu_critical, "CPU critical threshold (fraction of total jiffies)", :default => 0.95
-  opt :disk_warning, "Disk warning threshold (fraction of space used)", :default => 0.9
-  opt :disk_critical, "Disk critical threshold (fraction of space used)", :default => 0.95
-  opt :load_warning, "Load warning threshold (load average / core)", :default => 3
-  opt :load_critical, "Load critical threshold (load average / core)", :default => 8
-  opt :memory_warning, "Memory warning threshold (fraction of RAM)", :default => 0.85
-  opt :memory_critical, "Memory critical threshold (fraction of RAM)", :default => 0.95
-  opt :checks, "A list of checks to run.", :type => :strings, :default => ['cpu', 'load', 'memory', 'disk']
-
-  def initialize
-    @limits = {
-      :cpu => {:critical => opts[:cpu_critical], :warning => opts[:cpu_warning]},
-      :disk => {:critical => opts[:disk_critical], :warning => opts[:disk_warning]},
-      :load => {:critical => opts[:load_critical], :warning => opts[:load_warning]},
-      :memory => {:critical => opts[:memory_critical], :warning => opts[:memory_warning]}
-    }
-    case (@ostype = `uname -s`.chomp.downcase)
-    when 'darwin'
-      @cores = `sysctl -n hw.ncpu`.to_i
-      @cpu = method :darwin_cpu
-      @disk = method :disk
-      @load = method :darwin_load
-      @memory = method :darwin_memory
-      darwin_top
-    when 'freebsd'
-      @cores = `sysctl -n hw.ncpu`.to_i
-      @cpu = method :freebsd_cpu
-      @disk = method :disk
-      @load = method :bsd_load
-      @memory = method :freebsd_memory
-    when 'openbsd'
-      @cores = `sysctl -n hw.ncpu`.to_i
-      @cpu = method :openbsd_cpu
-      @disk = method :disk
-      @load = method :bsd_load
-      @memory = method :openbsd_memory
-    when 'sunos'
-      @cores = `mpstat -a 2>/dev/null`.split[33].to_i
-      @cpu = method :sunos_cpu
-      @disk = method :disk
-      @load = method :bsd_load
-      @memory = method :sunos_memory
-    else
-      @cores = `nproc`.to_i
-      puts "WARNING: OS '#{@ostype}' not explicitly supported. Falling back to Linux" unless @ostype == "linux"
-      @cpu = method :linux_cpu
-      @disk = method :disk
-      @load = method :linux_load
-      @memory = method :linux_memory
-      @supports_exclude_type = `df --help 2>&1 | grep -e "--exclude-type"` != ""
-    end
-
-    opts[:checks].each do |check|
-      case check
-      when "disk"
-        @disk_enabled = true
-      when "load"
-        @load_enabled = true
-      when "cpu"
-        @cpu_enabled = true
-      when "memory"
-        @memory_enabled = true
-      end
-    end
-  end
-
-  def alert(service, state, metric, description)
-    report(
-      :service => service.to_s,
-      :state => state.to_s,
-      :metric => metric.to_f,
-      :description => description
-    )
-  end
-
-  def report_pct(service, fraction, report)
-    if fraction
-      if fraction > @limits[service][:critical]
-        alert service, :critical, fraction, "#{sprintf("%.2f", fraction * 100)}% #{report}"
-      elsif fraction > @limits[service][:warning]
-        alert service, :warning, fraction, "#{sprintf("%.2f", fraction * 100)}% #{report}"
-      else
-        alert service, :ok, fraction, "#{sprintf("%.2f", fraction * 100)}% #{report}"
-      end
-    end
-  end
-
-  def linux_cpu
-    new = File.read('/proc/stat')
-    unless new[/cpu\s+(\d+)\s+(\d+)\s+(\d+)\s+(\d+)/]
-      alert 'cpu', :unknown, nil, "/proc/stat doesn't include a CPU line"
-      return false
-    end
-    u2, n2, s2, i2 = [$1, $2, $3, $4].map { |e| e.to_i }
-
-    if @old_cpu
-      u1, n1, s1, i1 = @old_cpu
-
-      used = (u2+n2+s2) - (u1+n1+s1)
-      total = used + i2-i1
-      fraction = used.to_f / total
-
-      report_pct :cpu, fraction, "user+nice+system\n\n#{reverse_numeric_sort_with_header(`ps -eo pcpu,pid,comm`)}"
-    end
-
-    @old_cpu = [u2, n2, s2, i2]
-  end
-
-  def linux_load
-    load = File.read('/proc/loadavg').split(/\s+/)[0].to_f / @cores
-    if load > @limits[:load][:critical]
-      alert "load", :critical, load, "1-minute load average/core is #{load}"
-    elsif load > @limits[:load][:warning]
-      alert "load", :warning, load, "1-minute load average/core is #{load}"
-    else
-      alert "load", :ok, load, "1-minute load average/core is #{load}"
-    end
-  end
-
-  def linux_memory
-    m = File.read('/proc/meminfo').split(/\n/).inject({}) { |info, line|
-      x = line.split(/:?\s+/)
-      # Assume kB...
-      info[x[0]] = x[1].to_i
-      info
-    }
-
-    free = m['MemFree'].to_i + m['Buffers'].to_i + m['Cached'].to_i
-    total = m['MemTotal'].to_i
-    fraction = 1 - (free.to_f / total)
-
-    report_pct :memory, fraction, "used\n\n#{reverse_numeric_sort_with_header(`ps -eo pmem,pid,comm`)}"
-  end
-
-  def freebsd_cpu
-    u2, n2, s2, t2, i2 = `sysctl -n kern.cp_time 2>/dev/null`.split.map{ |e| e.to_i } #FreeBSD has 5 cpu stats
-
-    if @old_cpu
-      u1, n1, s1, t1, i1 = @old_cpu
-
-      used = (u2+n2+s2+t2) - (u1+n1+s1+t1)
-      total = used + i2-i1
-      fraction = used.to_f / total
-
-      report_pct :cpu, fraction, "user+nice+sytem+interrupt\n\n#{reverse_numeric_sort_with_header(`ps -axo pcpu,pid,comm`)}"
-    end
-
-    @old_cpu = [u2, n2, s2, t2, i2]
-  end
-
-  def openbsd_cpu
-    u2, n2, s2, t2, i2 = `sysctl -n kern.cp_time 2>/dev/null`.split(',').map{ |e| e.to_i } #OpenBSD separates with ,
-
-    if @old_cpu
-      u1, n1, s1, t1, i1 = @old_cpu
+Process.setproctitle($PROGRAM_NAME)
 
-      used = (u2+n2+s2+t2) - (u1+n1+s1+t1)
-      total = used + i2-i1
-      fraction = used.to_f / total
-
-      report_pct :cpu, fraction, "user+nice+sytem+interrupt\n\n#{reverse_numeric_sort_with_header(`ps -axo pcpu,pid,comm`)}"
-    end
-
-    @old_cpu = [u2, n2, s2, t2, i2]
-  end
-
-  def sunos_cpu
-    mpstats = `mpstat -a 2>/dev/null`.split
-    u2 = mpstats[29].to_i
-    s2 = mpstats[30].to_i
-    t2 = mpstats[31].to_i
-    i2 = mpstats[32].to_i
-
-    if @old_cpu
-      u1, s1, t1, i1 = @old_cpu
-
-      used = (u2+s2+t2) - (u1+s1+t1)
-      total = used + i2-i1
-      if i2 == i1 && used == 0 #If the system is <1% used in both samples then total will be 0 + (99 - 99), avoid a div by 0
-        fraction = 0
-      else
-        fraction = used.to_f / total
-      end
-
-      report_pct :cpu, fraction, "user+sytem+interrupt\n\n#{reverse_numeric_sort_with_header(`ps -ao pcpu,pid,comm`)}"
-    end
-
-    @old_cpu = [u2, s2, t2, i2]
-  end
-
-  def bsd_load
-    m = `uptime`.split(':')[-1].chomp.gsub(/\s+/,'').split(',')
-    load = m[0].to_f / @cores
-    if load > @limits[:load][:critical]
-      alert "load", :critical, load, "1-minute load average/core is #{load}"
-    elsif load > @limits[:load][:warning]
-      alert "load", :warning, load, "1-minute load average/core is #{load}"
-    else
-      alert "load", :ok, load, "1-minute load average/core is #{load}"
-    end
-  end
-
-  def freebsd_memory
-    meminfo = `sysctl -n vm.stats.vm.v_page_count vm.stats.vm.v_wire_count vm.stats.vm.v_active_count 2>/dev/null`.chomp.split
-    fraction = (meminfo[1].to_f + meminfo[2].to_f) / meminfo[0].to_f
-
-    report_pct :memory, fraction, "used\n\n#{reverse_numeric_sort_with_header(`ps -axo pmem,pid,comm`)}"
-  end
-
-  def openbsd_memory
-    meminfo = `vmstat 2>/dev/null`.chomp.split
-    fraction = meminfo[28].to_f / meminfo[29].to_f #The ratio of active to free memory unlike the others :(
-
-    report_pct :memory, fraction, "used\n\n#{reverse_numeric_sort_with_header(`ps -axo pmem,pid,comm`)}"
-  end
-
-  def sunos_memory
-    meminfo = `vmstat 2>/dev/null`.chomp.split
-    total_mem = `prtconf | grep Memory`.split[2].to_f * 1024 # reports in GB but vmstat is in MB
-    fraction = ( total_mem - meminfo[32].to_f ) / total_mem
-
-    report_pct :memory, fraction, "used\n\n#{reverse_numeric_sort_with_header(`ps -ao pmem,pid,comm`)}"
-  end
-
-  def darwin_top
-    raw = `top -l 1 | grep -i "^\\(cpu\\|physmem\\|load\\)"`.chomp
-    @topdata = {:stamp => Time.now.to_i }
-    raw.each_line do |ln|
-      if ln.match(/Load Avg: [0-9.]+, [0-9.]+, ([0-9.])+/i)
-        @topdata[:load] = $1.to_f
-      elsif ln.match(/CPU usage: [0-9.]+% user, [0-9.]+% sys, ([0-9.]+)% idle/i)
-        @topdata[:cpu] = 1  - ($1.to_f / 100)
-      elsif mdat = ln.match(/PhysMem: ([0-9]+)([BKMGT]) wired, ([0-9]+)([BKMGT]) active, ([0-9]+)([BKMGT]) inactive, ([0-9]+)([BKMGT]) used, ([0-9]+)([BKMGT]) free/i)
-        wired = mdat[1].to_i * (1024 ** "BKMGT".index(mdat[2]))
-        active = mdat[3].to_i * (1024 ** "BKMGT".index(mdat[4]))
-        inactive = mdat[5].to_i * (1024 ** "BKMGT".index(mdat[6]))
-        used = mdat[7].to_i * (1024 ** "BKMGT".index(mdat[8]))
-        free = mdat[9].to_i * (1024 ** "BKMGT".index(mdat[10]))
-        @topdata[:memory] = (wired + active + used).to_f / (wired + active + used + inactive + free)
-      # This is for OSX Mavericks which
-      # uses a different format for top
-      # Example: PhysMem: 4662M used (1328M wired), 2782M unused.
-      elsif mdat = ln.match(/PhysMem: ([0-9]+)([BKMGT]) used \(([0-9]+)([BKMGT]) wired\), ([0-9]+)([BKMGT]) unused/i)
-        used  = mdat[1].to_i * (1024 ** "BKMGT".index(mdat[2]))
-        wired = mdat[3].to_i * (1024 ** "BKMGT".index(mdat[4]))
-        unused = mdat[5].to_i * (1024 ** "BKMGT".index(mdat[6]))
-        @topdata[:memory] = (used).to_f / (used + unused)
-      end
-    end
-  end
-
-  def darwin_cpu
-    darwin_top unless (Time.now.to_i - @topdata[:stamp]) < opts[:interval]
-    unless @topdata[:cpu]
-      alert 'cpu', :unknown, nil, "unable to get CPU stats from top"
-      return false
-    end
-    report_pct :cpu,  @topdata[:cpu], "usage\n\n#{reverse_numeric_sort_with_header(`ps -eo pcpu,pid,comm`)}"
-  end
-
-  def darwin_load
-    darwin_top unless (Time.now.to_i - @topdata[:stamp]) < opts[:interval]
-    unless @topdata[:load]
-      alert 'load', :unknown, nil, "unable to get load ave from top"
-      return false
-    end
-    metric = @topdata[:load] / @cores
-    if metric > @limits[:load][:critical]
-      alert "load", :critical, metric, "1-minute load average per core is #{metric}"
-    elsif metric > @limits[:load][:warning]
-      alert "load", :warning, metric, "1-minute load average per core is #{metric}"
-    else
-      alert "load", :ok, metric, "1-minute load average per core is #{metric}"
-    end
-  end
-
-  def darwin_memory
-    darwin_top unless (Time.now.to_i - @topdata[:stamp]) < opts[:interval]
-    unless @topdata[:memory]
-      alert 'memory', :unknown, nil, "unable to get memory data from top"
-      return false
-    end
-    report_pct :memory,  @topdata[:memory], "usage\n\n#{reverse_numeric_sort_with_header(`ps -eo pmem,pid,comm`)}"
-  end
-
-  def df
-    case @ostype
-    when 'darwin', 'freebsd', 'openbsd'
-      `df -P -t noiso9660`
-    when 'sunos'
-      `df -P` # Is there a good way to exlude iso9660 here?
-    else
-      if @supports_exclude_type
-        `df -P --exclude-type=iso9660 --exclude-type=nfs`
-      else
-        `df -P`
-      end
-    end
-  end
-
-  def disk
-    df.split(/\n/).each do |r|
-      f = r.split(/\s+/)
-      next if f[0] == 'Filesystem'
-      next unless f[0] =~ /\// # Needs at least one slash in the mount path
-
-      # Calculate capacity
-      x = f[4].to_f/100
-
-      if x > @limits[:disk][:critical]
-        alert "disk #{f[5]}", :critical, x, "#{f[4]} used"
-      elsif x > @limits[:disk][:warning]
-        alert "disk #{f[5]}", :warning, x, "#{f[4]} used"
-      else
-        alert "disk #{f[5]}", :ok, x, "#{f[4]} used"
-      end
-    end
-  end
+# Reports current CPU, disk, load average, and memory use to riemann.
 
-  def tick
-    if @cpu_enabled
-      @cpu.call
-    end
-    if @memory_enabled
-      @memory.call
-    end
-    if @disk_enabled
-      @disk.call
-    end
-    if @load_enabled
-      @load.call
-    end
-  end
-end
+require 'riemann/tools/health'
 
 Riemann::Tools::Health.run

data/bin/riemann-kvminstance

@@ -1,22 +1,27 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-require File.expand_path('../../lib/riemann/tools', __FILE__)
+Process.setproctitle($PROGRAM_NAME)
 
-class Riemann::Tools::KVM
-  include Riemann::Tools
+require File.expand_path('../lib/riemann/tools', __dir__)
 
-  def tick
+module Riemann
+  module Tools
+    class KVM
+      include Riemann::Tools
 
-  #determine how many instances I have according to libvirt
-  kvm_instances = %x[LANG=C virsh list | grep -c running]
+      def tick
+        # determine how many instances I have according to libvirt
+        kvm_instances = `LANG=C virsh list | grep -c running`
 
-  #submit them to riemann
-  report(
-     :service => "KVM Running VMs",
-     :metric => kvm_instances.to_i,
-     :state => "info"
-       )
+        # submit them to riemann
+        report(
+          service: 'KVM Running VMs',
+          metric: kvm_instances.to_i,
+          state: 'info',
+        )
+      end
+    end
   end
 end
 

data/bin/riemann-memcached

@@ -1,37 +1,43 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
+
+Process.setproctitle($PROGRAM_NAME)
 
 # Gathers memcached STATS and submits them to Riemann.
 
-require File.expand_path('../../lib/riemann/tools', __FILE__)
-
-class Riemann::Tools::Memcached
-  include Riemann::Tools
-  require 'socket'
-
-  opt :memcached_host, "Memcached hostname", :default => 'localhost'
-  opt :memcached_port, "Memcached port", :default => 11211
-
-  def tick
-    sock = TCPSocket.new(opts[:memcached_host], opts[:memcached_port])
-    sock.print("stats\r\n")
-    sock.flush
-    stats = sock.gets
-
-    data = {}
-    while true
-      stats = sock.gets
-      break if stats.strip == 'END'
-      m = stats.match /STAT (\w+) (\S+)/
-      report(
-             :host => opts[:memcached_host].dup,
-             :service => "memcached #{m[1]}",
-             :metric => m[2].to_f,
-             :state => 'ok',
-             :tags => ['memcached']
-             )
+require File.expand_path('../lib/riemann/tools', __dir__)
+
+module Riemann
+  module Tools
+    class Memcached
+      include Riemann::Tools
+      require 'socket'
+
+      opt :memcached_host, 'Memcached hostname', default: 'localhost'
+      opt :memcached_port, 'Memcached port', default: 11_211
+
+      def tick
+        sock = TCPSocket.new(opts[:memcached_host], opts[:memcached_port])
+        sock.print("stats\r\n")
+        sock.flush
+        stats = sock.gets
+
+        loop do
+          stats = sock.gets
+          break if stats.strip == 'END'
+
+          m = stats.match(/STAT (\w+) (\S+)/)
+          report(
+            host: opts[:memcached_host].dup,
+            service: "memcached #{m[1]}",
+            metric: m[2].to_f,
+            state: 'ok',
+            tags: ['memcached'],
+          )
+        end
+        sock.close
+      end
     end
-    sock.close
   end
 end
 

data/bin/riemann-net

@@ -1,110 +1,10 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-# Gathers network interface statistics and submits them to Riemann.
-
-require File.expand_path('../../lib/riemann/tools', __FILE__)
-
-class Riemann::Tools::Net
-  include Riemann::Tools
-
-  opt :interfaces, "Interfaces to monitor", :type => :strings, :default => nil
-  opt :ignore_interfaces, "Interfaces to ignore", :type => :strings, :default =>['lo']
-
-  def initialize
-    @old_state = nil
-    @interfaces = opts[:interfaces].map(&:dup) if opts[:interfaces]
-    @ignore_interfaces = opts[:ignore_interfaces].map(&:dup)
-  end
-
-  def state
-    f = File.read('/proc/net/dev')
-    state = f.split("\n").inject({}) do |s, line|
-      if line =~ /\s*(\w+?):\s*([\s\d]+)\s*/
-        iface = $1
-
-        ['rx bytes',
-        'rx packets',
-        'rx errs',
-        'rx drop',
-        'rx fifo',
-        'rx frame',
-        'rx compressed',
-        'rx multicast',
-        'tx bytes',
-        'tx packets',
-        'tx errs',
-        'tx drops',
-        'tx fifo',
-        'tx colls',
-        'tx carrier',
-        'tx compressed'].map do |service|
-          "#{iface} #{service}"
-        end.zip(
-          $2.split(/\s+/).map { |str| str.to_i }
-        ).each do |service, value|
-          s[service] = value
-        end
-      end
-      
-      s
-    end
+Process.setproctitle($PROGRAM_NAME)
 
-    # Filter interfaces
-    if is = @interfaces
-      state = state.select do |service, value|
-        is.include? service.split(' ').first
-      end
-    end
-
-    state = state.reject do |service, value|
-      @ignore_interfaces.include? service.split(' ').first
-    end
-
-    state
-  end
-  
-  def tick
-    state = self.state
-
-    if @old_state
-      # Report services from `@old_state` that don't exist in `state` as expired
-      @old_state.reject { |k| state.has_key?(k) }.each do |service, metric|
-        report(:service => service.dup, :state => 'expired')
-      end
-
-      # Report delta for services that have values in both `@old_state` and `state`
-      state.each do |service, metric|
-        next unless @old_state.has_key?(service)
-
-        delta = metric - @old_state[service]
-        svc_state = case service
-          when /drop$/
-            if delta > 0
-              'warning'
-            else
-              'ok'
-            end
-          when /errs$/
-            if delta > 0
-              'warning'
-            else
-              'ok'
-            end
-          else
-            'ok'
-          end
+# Gathers network interface statistics and submits them to Riemann.
 
-        report(
-          :service => service.dup,
-          :metric => (delta.to_f / opts[:interval]),
-          :state => svc_state
-        )
-      end
-    end
+require 'riemann/tools/net'
 
-    @old_state = state
-  end
-end
-  
 Riemann::Tools::Net.run