riemann-tools 1.0.0 → 1.1.0

data/bin/riemann-cloudant

@@ -1,59 +1,63 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-# Gathers load balancer statistics from Cloudant.com (shared cluster) and submits them to Riemann.
+Process.setproctitle($PROGRAM_NAME)
 
-require File.expand_path('../../lib/riemann/tools', __FILE__)
+# Gathers load balancer statistics from Cloudant.com (shared cluster) and submits them to Riemann.
 
-class Riemann::Tools::Cloudant
-  include Riemann::Tools
-  require 'net/http'
-  require 'json'
+require File.expand_path('../lib/riemann/tools', __dir__)
 
-  opt :cloudant_username, "Cloudant username", :type => :string, :required => true
-  opt :cloudant_password, "Cloudant pasword", :type => :string, :required => true
+module Riemann
+  module Tools
+    class Cloudant
+      include Riemann::Tools
+      require 'net/http'
+      require 'json'
 
-  def tick
-    json = JSON.parse(get_json().body)
-    json.each do |node|
-      return if node['svname'] == 'BACKEND' # this is just a sum of all nodes.
+      opt :cloudant_username, 'Cloudant username', type: :string, required: true
+      opt :cloudant_password, 'Cloudant pasword', type: :string, required: true
 
-      ns = "cloudant #{node['pxname']}"
-      cluster_name = node['tracked'].split('.')[0] # ie: meritage.cloudant.com
+      def tick
+        json.each do |node|
+          break if node['svname'] == 'BACKEND' # this is just a sum of all nodes.
 
-      # report health of each node.
-      report(
-        :service => ns,
-        :state   => (node['status'] == 'UP' ? 'ok' : 'critical'),
-        :tags    => ['cloudant', cluster_name]
-      )
+          ns = "cloudant #{node['pxname']}"
+          cluster_name = node['tracked'].split('.')[0] # ie: meritage.cloudant.com
 
-      # report property->metric of each node.
-      node.each do |property, metric|
-        unless ['pxname', 'svname', 'status', 'tracked'].include?(property)
+          # report health of each node.
           report(
-            :host    => node['tracked'],
-            :service => "#{ns} #{property}",
-            :metric  => metric.to_f,
-            :state   => (node['status'] == 'UP' ? 'ok' : 'critical'),
-            :tags    => ['cloudant', cluster_name]
+            service: ns,
+            state: (node['status'] == 'UP' ? 'ok' : 'critical'),
+            tags: ['cloudant', cluster_name],
           )
+
+          # report property->metric of each node.
+          node.each do |property, metric|
+            next if %w[pxname svname status tracked].include?(property)
+
+            report(
+              host: node['tracked'],
+              service: "#{ns} #{property}",
+              metric: metric.to_f,
+              state: (node['status'] == 'UP' ? 'ok' : 'critical'),
+              tags: ['cloudant', cluster_name],
+            )
+          end
         end
       end
 
+      def json
+        http = Net::HTTP.new('cloudant.com', 443)
+        http.use_ssl = true
+        http.start do |h|
+          get = Net::HTTP::Get.new('/api/load_balancer')
+          get.basic_auth opts[:cloudant_username], opts[:cloudant_password]
+          h.request get
+        end
+        JSON.parse(http.boby)
+      end
     end
   end
-
-   def get_json
-    http = Net::HTTP.new('cloudant.com', 443)
-    http.use_ssl = true
-      http.start do |h|
-        get = Net::HTTP::Get.new('/api/load_balancer')
-        get.basic_auth opts[:cloudant_username], opts[:cloudant_password]
-        h.request get
-     end
-   end
-
 end
 
 Riemann::Tools::Cloudant.run

data/bin/riemann-consul

@@ -1,106 +1,112 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
+
+Process.setproctitle($PROGRAM_NAME)
 
 # Reports service and node status to riemann
 
-require File.expand_path('../../lib/riemann/tools', __FILE__)
+require File.expand_path('../lib/riemann/tools', __dir__)
 require 'socket'
 require 'net/http'
 require 'uri'
 require 'json'
 
-class Riemann::Tools::ConsulHealth
-  include Riemann::Tools
-
-  opt :consul_host, "Consul API Host (default to localhost)", :default => "localhost"
-  opt :consul_port, "Consul API Host (default to 8500)", :default => "8500"
-  opt :prefix, "prefix to use for all service names when reporting", :default => "consul "
-  opt :minimum_services_per_node, "minimum services per node (default: 0)", :default => 0
-
-  def initialize
-
-    @hostname = opts[:consul_host]
-    @prefix = opts[:prefix]
-    @minimum_services_per_node = opts[:minimum_services_per_node]
-    @underlying_ip = IPSocket.getaddress(@hostname)
-    @consul_leader_url = URI.parse("http://" + opts[:consul_host] + ":" + opts[:consul_port] + "/v1/status/leader")
-    @consul_services_url = URI.parse("http://" + opts[:consul_host] + ":" + opts[:consul_port] + "/v1/catalog/services")
-    @consul_nodes_url = URI.parse("http://" + opts[:consul_host] + ":" + opts[:consul_port] + "/v1/catalog/nodes")
-    @consul_health_url_prefix = "http://" + opts[:consul_host] + ":" + opts[:consul_port] + "/v1/health/service/"
-
-    @last_services_read = Hash.new
-
-  end
+module Riemann
+  module Tools
+    class ConsulHealth
+      include Riemann::Tools
+
+      opt :consul_host, 'Consul API Host (default to localhost)', default: 'localhost'
+      opt :consul_port, 'Consul API Host (default to 8500)', default: '8500'
+      opt :prefix, 'prefix to use for all service names when reporting', default: 'consul '
+      opt :minimum_services_per_node, 'minimum services per node (default: 0)', default: 0
+
+      def initialize
+        @hostname = opts[:consul_host]
+        @prefix = opts[:prefix]
+        @minimum_services_per_node = opts[:minimum_services_per_node]
+        @underlying_ip = IPSocket.getaddress(@hostname)
+        @consul_leader_url = URI.parse("http://#{opts[:consul_host]}:#{opts[:consul_port]}/v1/status/leader")
+        @consul_services_url = URI.parse("http://#{opts[:consul_host]}:#{opts[:consul_port]}/v1/catalog/services")
+        @consul_nodes_url = URI.parse("http://#{opts[:consul_host]}:#{opts[:consul_port]}/v1/catalog/nodes")
+        @consul_health_url_prefix = "http://#{opts[:consul_host]}:#{opts[:consul_port]}/v1/health/service/"
+
+        @last_services_read = {}
+      end
 
-  def alert(hostname, service, state, metric, description)
+      def alert(hostname, service, state, metric, description)
+        opts = {
+          host: hostname,
+          service: service.to_s,
+          state: state.to_s,
+          metric: metric,
+          description: description,
+        }
 
-    opts = {  :host => hostname,
-              :service => service.to_s,
-              :state => state.to_s,
-              :metric => metric,
-              :description => description }
+        report(opts)
+      end
 
-    report(opts)
-  end
+      def get(url)
+        Net::HTTP.get_response(url).body
+      end
 
-  def get(url)
-    Net::HTTP.get_response(url).body
-  end
+      def tick
+        leader = JSON.parse(get(@consul_leader_url))
+        leader_hostname = URI.parse("http://#{leader}").hostname
 
-  def tick
+        return unless leader_hostname == @underlying_ip
 
-    leader = JSON.parse(get(@consul_leader_url))
-    leader_hostname = URI.parse("http://" + leader).hostname
+        nodes = JSON.parse(get(@consul_nodes_url))
+        services = JSON.parse(get(@consul_services_url))
+        services_by_nodes = {}
 
-    if (leader_hostname == @underlying_ip)
-      nodes = JSON.parse(get(@consul_nodes_url))
-      services = JSON.parse(get(@consul_services_url))
-      services_by_nodes = Hash.new
+        nodes.each do |node|
+          node_name = node['Node']
+          services_by_nodes[node_name] = 0
+        end
 
-      for node in nodes
-        node_name = node["Node"]
-        services_by_nodes[node_name] = 0
-      end
+        # For every service
+        services.each do |service|
+          service_name = service[0]
+          health_url = URI.parse(@consul_health_url_prefix + service_name)
+          health_nodes = JSON.parse(get(health_url))
 
+          total_count = 0
+          ok_count = 0
 
-      # For every service
-      for service in services
-        service_name = service[0]
-        health_url = URI.parse(@consul_health_url_prefix + service_name)
-        health_nodes = JSON.parse(get(health_url))
+          health_nodes.each do |node|
+            hostname = node['Node']['Node']
+            ok = node['Checks'].all? { |check| check['Status'] == 'passing' }
+            alert(hostname, "#{@prefix}#{service_name}", ok ? :ok : :critical, ok ? 1 : 0, JSON.generate(node))
+            total_count += 1
+            ok_count += ok ? 1 : 0
 
-        totalCount = 0
-        okCount = 0
+            last_services_by_nodes = services_by_nodes[hostname].to_i
+            services_by_nodes[hostname] = last_services_by_nodes + 1
+          end
 
-        for node in health_nodes
-          hostname = node["Node"]["Node"]
-          ok = node["Checks"].all? {|check| check["Status"] == "passing"}
-          alert(hostname, "#{@prefix}#{service_name}", ok ? :ok : :critical, ok ? 1 : 0, JSON.generate(node))
-          totalCount += 1
-          okCount += ok ? 1 : 0
+          unless @last_services_read[service_name].nil?
+            last_ok = @last_services_read[service_name]
+            if last_ok != ok_count
+              alert(
+                'total', "#{@prefix}#{service_name}-count", ok_count >= last_ok ? :ok : :critical, ok_count,
+                "Number of passing #{service_name} is: #{ok_count}/#{total_count}, Last time it was: #{last_ok}",
+              )
+            end
+          end
 
-          last_services_by_nodes = services_by_nodes[hostname].to_i
-          services_by_nodes[hostname] = last_services_by_nodes + 1
+          @last_services_read[service_name] = ok_count
         end
 
-        if (@last_services_read[service_name] != nil)
-          lastOk = @last_services_read[service_name]
-          if (lastOk != okCount)
-            alert("total", "#{@prefix}#{service_name}-count", okCount >= lastOk ? :ok : :critical, okCount, "Number of passing #{service_name} is: #{okCount}/#{totalCount}, Last time it was: #{lastOk}")
-          end
+        # For every node
+        services_by_nodes.each do |node, count|
+          alert(
+            node, "#{@prefix}total-services", count >= @minimum_services_per_node ? :ok : :critical, count,
+            "#{count} services in the specified node",
+          )
         end
-
-        @last_services_read[service_name] = okCount
-
-      end
-
-      # For every node
-      for node,count in services_by_nodes
-        alert(node, "#{@prefix}total-services", (count >= @minimum_services_per_node) ? :ok : :critical, count, "#{count} services in the specified node")
       end
-
     end
-
   end
 end
 

data/bin/riemann-dir-files-count

@@ -1,54 +1,60 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-# Gets the number of files present on a directory and submits it to riemann
-
-require File.expand_path('../../lib/riemann/tools', __FILE__)
-
-class Riemann::Tools::DirFilesCount
-  include Riemann::Tools
+Process.setproctitle($PROGRAM_NAME)
 
-  opt :directory, "", :default => '/var/log'
-  opt :service_prefix, "The first part of the service name, before the directory path", :default => "dir-files-count"
-  opt :warning, "Dir files number warning threshold", :type => Integer
-  opt :critical, "Dir files number critical threshold", :type => Integer
-  opt :alert_on_missing, "Send a critical metric if the directory is missing?", :default => true
-
-  def initialize
-    @dir = opts.fetch(:directory)
-    @service_prefix = opts.fetch(:service_prefix)
-    @warning = opts.fetch(:warning, nil)
-    @critical = opts.fetch(:critical, nil)
-    @alert_on_missing = opts.fetch(:alert_on_missing)
-  end
-
-  def tick
-    if Dir.exists?(@dir)
-      metric = Dir.entries(@dir).size - 2
-      report(
-        :service => "#{@service_prefix} #{@dir}",
-        :metric => metric,
-        :state => state(metric),
-        :tags => ['dir_files_count']
-      )
-    elsif @alert_on_missing
-      report(
-        :service => "#{@service_prefix} #{@dir} missing",
-        :description => "#{@service_prefix} #{@dir} does not exist",
-        :metric => metric,
-        :state => 'critical',
-        :tags => ['dir_files_count']
-      )
-    end
-  end
+# Gets the number of files present on a directory and submits it to riemann
 
-  def state(metric)
-    if @critical && metric > @critical
-      'critical'
-    elsif @warning && metric > @warning
-      'warning'
-    else
-      'ok'
+require File.expand_path('../lib/riemann/tools', __dir__)
+
+module Riemann
+  module Tools
+    class DirFilesCount
+      include Riemann::Tools
+
+      opt :directory, '', default: '/var/log'
+      opt :service_prefix, 'The first part of the service name, before the directory path', default: 'dir-files-count'
+      opt :warning, 'Dir files number warning threshold', type: Integer
+      opt :critical, 'Dir files number critical threshold', type: Integer
+      opt :alert_on_missing, 'Send a critical metric if the directory is missing?', default: true
+
+      def initialize
+        @dir = opts.fetch(:directory)
+        @service_prefix = opts.fetch(:service_prefix)
+        @warning = opts.fetch(:warning, nil)
+        @critical = opts.fetch(:critical, nil)
+        @alert_on_missing = opts.fetch(:alert_on_missing)
+      end
+
+      def tick
+        if Dir.exist?(@dir)
+          metric = Dir.entries(@dir).size - 2
+          report(
+            service: "#{@service_prefix} #{@dir}",
+            metric: metric,
+            state: state(metric),
+            tags: ['dir_files_count'],
+          )
+        elsif @alert_on_missing
+          report(
+            service: "#{@service_prefix} #{@dir} missing",
+            description: "#{@service_prefix} #{@dir} does not exist",
+            metric: metric,
+            state: 'critical',
+            tags: ['dir_files_count'],
+          )
+        end
+      end
+
+      def state(metric)
+        if @critical && metric > @critical
+          'critical'
+        elsif @warning && metric > @warning
+          'warning'
+        else
+          'ok'
+        end
+      end
     end
   end
 end

data/bin/riemann-dir-space

@@ -1,54 +1,60 @@
 #!/usr/bin/env ruby
-Process.setproctitle($0)
+# frozen_string_literal: true
 
-# Gathers the space used by a directory and submits it to riemann
-
-require File.expand_path('../../lib/riemann/tools', __FILE__)
-
-class Riemann::Tools::DirSpace
-  include Riemann::Tools
+Process.setproctitle($PROGRAM_NAME)
 
-  opt :directory, "", :default => '/var/log'
-  opt :service_prefix, "The first part of the service name, before the directory path", :default => "dir-space"
-  opt :warning, "Dir space warning threshold (in bytes)", :type => Integer
-  opt :critical, "Dir space critical threshold (in bytes)", :type => Integer
-  opt :alert_on_missing, "Send a critical metric if the directory is missing?", :default => true
-
-  def initialize
-    @dir = opts.fetch(:directory)
-    @service_prefix = opts.fetch(:service_prefix)
-    @warning = opts.fetch(:warning, nil)
-    @critical = opts.fetch(:critical, nil)
-    @alert_on_missing = opts.fetch(:alert_on_missing)
-  end
-
-  def tick
-    if Dir.exists?(@dir)
-      metric = `du '#{@dir}'`.lines.to_a.last.split("\t")[0].to_i
-      report(
-        :service => "#{@service_prefix} #{@dir}",
-        :metric => metric,
-        :state => state(metric),
-        :tags => ['dir_space']
-      )
-    elsif @alert_on_missing
-      report(
-        :service => "#{@service_prefix} #{@dir} missing",
-        :description => "#{@service_prefix} #{@dir} does not exist",
-        :metric => metric,
-        :state => 'critical',
-        :tags => ['dir_space']
-      )
-    end
-  end
+# Gathers the space used by a directory and submits it to riemann
 
-  def state(metric)
-    if @critical && metric > @critical
-      'critical'
-    elsif @warning && metric > @warning
-      'warning'
-    else
-      'ok'
+require File.expand_path('../lib/riemann/tools', __dir__)
+
+module Riemann
+  module Tools
+    class DirSpace
+      include Riemann::Tools
+
+      opt :directory, '', default: '/var/log'
+      opt :service_prefix, 'The first part of the service name, before the directory path', default: 'dir-space'
+      opt :warning, 'Dir space warning threshold (in bytes)', type: Integer
+      opt :critical, 'Dir space critical threshold (in bytes)', type: Integer
+      opt :alert_on_missing, 'Send a critical metric if the directory is missing?', default: true
+
+      def initialize
+        @dir = opts.fetch(:directory)
+        @service_prefix = opts.fetch(:service_prefix)
+        @warning = opts.fetch(:warning, nil)
+        @critical = opts.fetch(:critical, nil)
+        @alert_on_missing = opts.fetch(:alert_on_missing)
+      end
+
+      def tick
+        if Dir.exist?(@dir)
+          metric = `du '#{@dir}'`.lines.to_a.last.split("\t")[0].to_i
+          report(
+            service: "#{@service_prefix} #{@dir}",
+            metric: metric,
+            state: state(metric),
+            tags: ['dir_space'],
+          )
+        elsif @alert_on_missing
+          report(
+            service: "#{@service_prefix} #{@dir} missing",
+            description: "#{@service_prefix} #{@dir} does not exist",
+            metric: metric,
+            state: 'critical',
+            tags: ['dir_space'],
+          )
+        end
+      end
+
+      def state(metric)
+        if @critical && metric > @critical
+          'critical'
+        elsif @warning && metric > @warning
+          'warning'
+        else
+          'ok'
+        end
+      end
     end
   end
 end