riemann-tools 1.10.0 → 1.11.0

data/lib/riemann/tools/hwmon.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+require 'riemann/tools'
+
+# See https://www.kernel.org/doc/html/latest/hwmon/index.html
+module Riemann
+  module Tools
+    class Hwmon
+      include Riemann::Tools
+
+      class Device
+        attr_reader :hwmon, :type, :number, :crit, :lcrit, :service
+
+        def initialize(hwmon, type, number)
+          @hwmon = hwmon
+          @type = type
+          @number = number
+
+          @crit = scale(read_hwmon_i('crit'))
+          @lcrit = scale(read_hwmon_i('lcrit'))
+          @service = ['hwmon', read_hwmon_file('name'), read_hwmon_s('label')].compact.join(' ')
+        end
+
+        def input
+          read_hwmon_i('input')
+        end
+
+        def report
+          value = scale(input)
+
+          state = :ok
+          state = :critical if crit && value >= crit
+          state = :critical if lcrit && value <= lcrit
+          {
+            service: service,
+            state: state,
+            metric: value,
+            description: fromat_input(value),
+          }
+        end
+
+        private
+
+        def scale(value)
+          return nil if value.nil?
+
+          case type
+          when :fan then value.to_i # rpm
+          when :in, :temp, :curr then value.to_f / 1000 # mV, m°C, mA
+          when :humidity then value.to_f / 100 # %H
+          when :power, :energy then value.to_f / 1_000_000 # uW, uJ
+          end
+        end
+
+        def fromat_input(value)
+          case type
+          when :in then format('%<value>.3f V', { value: value })
+          when :fan then "#{value} RPM"
+          when :temp then format('%<value>.3f °C', { value: value })
+          when :curr then format('%<value>.3f A', { value: value })
+          when :power then format('%<value>.3f W', { value: value })
+          when :energy then format('%<value>.3f J', { value: value })
+          when :humidity then format('%<value>d %H', { value: (value * 100).to_i })
+          end
+        end
+
+        def read_hwmon_i(file)
+          s = read_hwmon_s(file)
+          return nil if s.nil?
+
+          s.to_i
+        end
+
+        def read_hwmon_s(file)
+          read_hwmon_file("#{type}#{number}_#{file}")
+        end
+
+        def read_hwmon_file(file)
+          File.read("/sys/class/hwmon/hwmon#{hwmon}/#{file}").chomp
+        rescue Errno::ENOENT
+          nil
+        end
+      end
+
+      attr_reader :devices
+
+      def initialize
+        super
+
+        @devices = poll_devices
+      end
+
+      def poll_devices
+        res = []
+
+        Dir['/sys/class/hwmon/hwmon[0-9]*/{in,fan,temp,curr,power,energy,humidity}[0-9]*_input'].each do |filename|
+          m = filename.match(%r{/sys/class/hwmon/hwmon(\d+)/([[:alpha:]]+)(\d+)_input})
+          res << Device.new(m[1].to_i, m[2].to_sym, m[3].to_i)
+        end
+
+        res
+      end
+
+      def tick
+        devices.each do |device|
+          report(device.report)
+        end
+      end
+    end
+  end
+end

data/lib/riemann/tools/mdstat_parser.tab.rb

@@ -1,7 +1,7 @@
 #
 # DO NOT MODIFY!!!!
-# This file is automatically generated by Racc 1.6.2
-# from Racc grammar file "".
+# This file is automatically generated by Racc 1.7.3
+# from Racc grammar file "mdstat_parser.y".
 #
 
 require 'racc/parser.rb'
@@ -255,6 +255,7 @@ Racc_arg = [
   racc_shift_n,
   racc_reduce_n,
   racc_use_result_var ]
+Ractor.make_shareable(Racc_arg) if defined?(Ractor)
 
 Racc_token_to_s_table = [
   "$end",
@@ -312,6 +313,7 @@ Racc_token_to_s_table = [
   "bitmap",
   "progress_action",
   "identifiers" ]
+Ractor.make_shareable(Racc_token_to_s_table) if defined?(Ractor)
 
 Racc_debug_parser = false
 

data/lib/riemann/tools/net.rb

@@ -12,6 +12,8 @@ module Riemann
       opt :ignore_interfaces, 'Interfaces to ignore', type: :strings, default: ['\Alo\d*\z']
 
       def initialize
+        super
+
         @old_state = nil
         @interfaces = if opts[:interfaces]
                         opts[:interfaces].reject(&:empty?).map(&:dup)
@@ -118,13 +120,7 @@ module Riemann
 
             delta = metric - @old_state[service]
             svc_state = case service
-                        when /drop$/
-                          if delta.positive?
-                            'warning'
-                          else
-                            'ok'
-                          end
-                        when /errs$/
+                        when /drop$/, /errs$/
                           if delta.positive?
                             'warning'
                           else

data/lib/riemann/tools/nginx_status.rb

@@ -26,6 +26,8 @@ module Riemann
       opt :user_agent, 'User-Agent header for HTTP requests', short: :none, default: "#{File.basename($PROGRAM_NAME)}/#{Riemann::Tools::VERSION} (+https://github.com/riemann/riemann-tools)"
 
       def initialize
+        super
+
         @uri = URI.parse(opts[:uri])
 
         # sample response:
@@ -39,13 +41,13 @@ module Riemann
       end
 
       def state(key, value)
-        if opts.key? "#{key}_critical".to_sym
-          critical_threshold = opts["#{key}_critical".to_sym]
+        if opts.key? :"#{key}_critical"
+          critical_threshold = opts[:"#{key}_critical"]
           return 'critical' if critical_threshold.positive? && (value >= critical_threshold)
         end
 
-        if opts.key? "#{key}_warning".to_sym
-          warning_threshold = opts["#{key}_warning".to_sym]
+        if opts.key? :"#{key}_warning"
+          warning_threshold = opts[:"#{key}_warning"]
           return 'warning' if warning_threshold.positive? && (value >= warning_threshold)
         end
 
@@ -75,6 +77,8 @@ module Riemann
         values = @re.match(response).to_a[1, 7].map(&:to_i)
 
         @keys.zip(values).each do |key, value|
+          next unless opts[:checks].include?(key)
+
           report({
                    service: "nginx #{key}",
                    metric: value,

data/lib/riemann/tools/ntp.rb

@@ -9,6 +9,8 @@ module Riemann
       include Riemann::Tools
 
       def initialize
+        super
+
         @hostname = `hostname`.chomp
         @ostype = `uname -s`.chomp.downcase
         abort 'WARNING: macOS not explicitly supported. Exiting.' if @ostype == 'darwin'

data/lib/riemann/tools/portcheck.rb

@@ -14,6 +14,8 @@ module Riemann
       opt :ports, "List of ports to check, e.g. '-r 80 443'", type: :ints
 
       def initialize
+        super
+
         @hostname = opts.fetch(:hostname)
         @ports = opts.fetch(:ports)
       end

data/lib/riemann/tools/proc.rb

@@ -13,6 +13,8 @@ module Riemann
       opt :proc_max_critical, 'running process count maximum', default: 65_536
 
       def initialize
+        super
+
         @limits = { critical: { min: opts[:proc_min_critical], max: opts[:proc_max_critical] } }
 
         abort 'FATAL: specify a process regular expression, see --help for usage' unless opts[:proc_regex]