riemann-babbler 1.2.7 → 1.2.8

data/lib/riemann/babbler/plugins/mdadm.rb

@@ -14,8 +14,17 @@ class Riemann::Babbler::Mdadm < Riemann::Babbler
     file = File.read('/proc/mdstat').split("\n")
     status = Array.new
     file.each_with_index do |line, index|
-      next unless line.include? '_'
+      next unless line.include?('blocks')
+
       device = file[index-1].split(':')[0].strip
+
+      mdstatus = line.split(" ").last
+      next if mdstatus == '[UU]' # пропускаем все збс
+      if mdstatus == plugin.states.send(device).to_s # пропускаем если стейт зафикисирован в конфиге
+        status << { :service => plugin.service + " #{device}", :metric => 1, :state => 'ok', :description => "mdadm failed device #{device}, but disabled in config" }
+        next
+      end
+
       status << { :service => plugin.service + " #{device}", :metric => 1, :description => "mdadm failed device #{device}: #{get_failed_parts(device)}" }
     end
     status

data/lib/riemann/babbler/plugins/pgsql.rb

@@ -0,0 +1,70 @@
+class Riemann::Babbler::Pgsql < Riemann::Babbler
+
+  def init
+    plugin.set_default(:service, 'pgsql')
+    plugin.set_default(:host, '127.0.0.1')
+    plugin.set_default(:user, 'postgres')
+    plugin.set_default(:chk_file, '/usr/bin/pg_config')
+
+    plugin.set_default(:rep_lag_sql, "select date_part('epoch', pg_last_xact_replay_timestamp() - now())::int")
+    plugin.set_default(:rep_lag_warn, 500)
+    plugin.set_default(:rep_lag_crit, 1000)
+
+    plugin.set_default(:conn_warn, 5)
+    plugin.set_default(:conn_crit, 3)
+
+    plugin.set_default(:interval, 60)
+  end
+
+  def run_plugin
+    File.exists? plugin.chk_file
+  end
+
+  def run_sql(sql)
+    shell("psql -h #{plugin.host} -U #{plugin.user} -tnc \"#{sql}\" postgres")
+  end
+
+  def in_recovery?
+    run_sql('select pg_is_in_recovery()') == 't'
+  end
+
+  # connection to pg
+  def connections
+    max_conn = run_sql("show max_connections").to_i
+    res_conn = run_sql("show superuser_reserved_connections").to_i
+    cur_conn = run_sql("select count(1) from pg_stat_activity;").to_i
+    [cur_conn, (max_conn - res_conn - cur_conn)]
+  end
+
+  def rep_lag_state
+    rep_lag = run_sql(plugin.rep_lag_sql).to_i
+    if rep_lag >= plugin.rep_lag_crit || rep_lag == 0
+      { :service => plugin.service + ' rep_lag', :description => "Postgresql replication lag state", :state => 'critical', :metric => rep_lag }
+    elsif rep_lag >= plugin.rep_lag_warn
+      { :service => plugin.service + ' rep_lag', :description => "Postgresql replication lag state", :state => 'warning', :metric => rep_lag }
+    else
+      { :service => plugin.service + ' rep_lag', :description => "Postgresql replication lag state", :state => 'ok', :metric => rep_lag }
+    end    
+  end
+
+  def collect
+    status = Array.new
+
+    cur_conn, res_conn = connections
+
+    status << rep_lag_state if in_recovery?
+    status << { :service => plugin.service + ' connections', :description => "Postgresql current connections", :state => 'ok', :metric => cur_conn }
+
+    # check reserved pool size
+    if res_conn < plugin.conn_warn
+      if res_conn > plugin.conn_crit
+        status << { :service => plugin.service + ' reserved connections', :description => "Postgresql reserved connections state", :state => 'warning', :metric => res_conn }
+      else
+        status << { :service => plugin.service + ' reserved connections', :description => "Postgresql reserved connections state", :state => 'critical', :metric => res_conn }
+      end
+    end
+
+    status
+  end
+
+end

data/lib/riemann/babbler/version.rb

@@ -1,5 +1,5 @@
 module Riemann
   class Babbler
-    VERSION = '1.2.7'
+    VERSION = '1.2.8'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: riemann-babbler
 version: !ruby/object:Gem::Version
-  version: 1.2.7
+  version: 1.2.8
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-20 00:00:00.000000000 Z
+date: 2013-09-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: riemann-client
@@ -204,6 +204,7 @@ files:
 - lib/riemann/babbler/plugins/net.rb
 - lib/riemann/babbler/plugins/net_stat.rb
 - lib/riemann/babbler/plugins/nginx.rb
+- lib/riemann/babbler/plugins/pgsql.rb
 - lib/riemann/babbler/plugins/runit.rb
 - lib/riemann/babbler/plugins/status_file.rb
 - lib/riemann/babbler/plugins/tw_cli.rb