riemann-babbler 0.0.2 → 0.0.3

data/README.md

@@ -0,0 +1,70 @@
+### Установка
+```
+gem install riemann-babbler
+````
+
+### Использование
+```
+$ riemann-babbler --help
+Riemann-babbler is plugin manager for riemann-tools.
+
+Usage:
+       riemann-babbler [options]
+where [options] are:
+  --config, -c <s>:   Config file (default: /etc/riemann-babbler/config.yml)
+  --version, -v:   Print version and exit
+  --help, -h:   Show this message
+```
+
+### Описание конфига
+Bubbler имеет собственные конифиги, значения полученные через --config будут смерджены 
+```yaml
+riemann:
+  host: riemann.host # хост riemann куда слать сообщения
+  port: 5555 # порт
+  tags: # таги которые будут сообшатся
+    - prod
+    - web
+  suffix: ".testing" # окончание `hostname`
+
+plugins:
+  dirs:
+    - /etc/riemann/plugins # загружает все плагины из указаной дирректории
+  files:
+    - /var/lib/riemann-plugins/test.rb # подгружает плагин по указаному пути
+```
+##### Настройки конкретного плагина
+```yaml
+plugins:
+  awesome_plugin:
+  	service: some critical service # описание сервиса для поста на riemann
+  	interval: 1 # как часто дергать плагин (в сек)
+  	states:
+  		warning: 80 # какой стейт давать плагину когда метрика перевалит за указанное значение
+  		critical: 90 # соответственно стейт critical
+  	some_parametr: "pgsql://username:password@database" # например необходимая настройка для плагина
+```
+
+### Написание собственного плагина
+```ruby
+class Riemann::Babbler::Awesomeplugins
+  include Riemann::Babbler
+
+  # быстрый доступ к конфигу
+  def plugin
+    options.plugins.awesome_plugin
+  end
+
+  # то что будет вызыватся таймером по указаному interval
+  def tick
+    status = {
+      :service => plugin.service,
+      :state => 'ok'
+    }
+    report status
+  end
+
+end
+# обязательный вызов
+Riemann::Babbler::Awesomeplugins.run
+```

data/bin/{riemann-babbler.rb → riemann-babbler}

@@ -60,6 +60,12 @@ unless configatron.plugins.dirs.nil?
   end
 end
 
+unless configatron.plugins.files.nil?
+  configatron.plugins.files.each do |file|
+    plugins << file
+  end
+end
+
 # start plugins
 Parallel.each( plugins, :in_threads => plugins.count ) do |plugin|
   require "#{plugin}"

data/config.yml

@@ -1,16 +1,46 @@
 riemann:
-  host: riemann-testing-01.undev.cc
+  host: localhost
   port: 5555
-  tags:
-    - prod
-    - nptv
+
+logger:
+  level: DEBUG
 
 plugins:
-  dirs:
-    - /etc/riemann-babbler/plugins
-  dummy:
-    service: dummy service
+
+  babbler:
+    service: keep alive
+    interval: 1
+
+  la:
+    service: load avverage
     interval: 1
     states:
-      warning: 70
-      critical: 90
+      warning: 4
+      critical: 10
+
+  disk:
+    service: disk usage
+    interval: 1
+    states:
+      warning: 0.7
+      critical: 0.85
+
+  cpu:
+    service: cpu usage
+    interval: 1
+    states:
+      warning: 0.7
+      critical: 0.85
+
+  memory:
+    service: memory ussage
+    interval: 1
+    report_free: true
+    report_total: true
+    states:
+      warning: 0.6
+      critical: 0.8
+
+  net:
+    service: net ussage
+    interval: 1

data/lib/riemann/babbler/plugin.rb

@@ -27,10 +27,18 @@ module Riemann
     alias :opts :options
 
     def report(event)
-      event[:tags] unless options.riemann.tags.nil?
+      event[:tags] = options.riemann.tags unless options.riemann.tags.nil?
+      event[:host] =  host
+      log.debug "Report status: #{event.inspect}"
       riemann << event
     end
 
+    def host
+      hostname = `hostname`.chomp.downcase
+      hostname += options.riemann.suffix unless options.riemann.suffix.nil?
+      hostname
+    end
+
     def riemann
       @riemann ||= Riemann::Client.new(
         :host => options.riemann.host,

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

@@ -0,0 +1,18 @@
+class Riemann::Babbler::Dummy
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.babbler
+  end
+
+  def tick
+    status = {
+      :service => plugin.service,
+      :state => 'ok'
+    }
+    report status
+  end
+
+end
+
+Riemann::Babbler::Dummy.run

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

@@ -0,0 +1,38 @@
+class Riemann::Babbler::Cpu
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.cpu
+  end
+
+  def cpu
+    cpu = File.read('/proc/stat')
+    cpu[/cpu\s+(\d+)\s+(\d+)\s+(\d+)\s+(\d+)/]
+    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
+    end
+
+    @old_cpu = [u2, n2, s2, i2]
+    log.error fraction
+    fraction
+  end
+
+  def tick
+    current_state = cpu
+    # при первом проходе у нас будет nil
+    next unless current_state
+    report({
+      :service => plugin.service,
+      :state => state(current_state),
+      :metric => current_state
+    })
+  end
+
+end
+
+Riemann::Babbler::Cpu.run

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

@@ -0,0 +1,32 @@
+class Riemann::Babbler::Disk
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.disk
+  end
+
+  def disk
+    disk = {}
+    `df -P`.split(/\n/).each do |r|
+      f = r.split(/\s+/)
+      next unless f[0] =~ /^\//
+      next if f[0] == 'Filesystem'
+      x = f[4].to_f/100
+      disk.merge!({f[5] => x})
+    end
+    return disk
+  end
+
+  def tick
+    disk.each do |point, free|
+      report({
+        :service => plugin.service + " on #{point}",
+        :state => state(free),
+        :metric => free
+      })
+    end
+  end
+
+end
+
+Riemann::Babbler::Disk.run

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

@@ -0,0 +1,24 @@
+class Riemann::Babbler::La
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.la
+  end
+
+  def la 
+    File.read('/proc/loadavg').split(/\s+/)[2].to_f
+  end
+
+  def tick
+    current_state = la
+    status = {
+      :service => plugin.service,
+      :state => state(current_state),
+      :metric => current_state
+    }
+    report status
+  end
+
+end
+
+Riemann::Babbler::La.run

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

@@ -0,0 +1,43 @@
+class Riemann::Babbler::Memory
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.memory
+  end
+
+  def 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)
+    return {:free => (free.to_f/1024), :fraction => fraction, :total => (total.to_f/1024)}
+  end
+
+  def tick
+    current_state = memory
+    # соотношение свободной/занятой
+    report({
+      :service => plugin.service,
+      :state => state(current_state[:fraction]),
+      :metric => current_state[:fraction]
+    })
+    # постим сообщение о свобоной памяти
+    report({
+      :service => plugin.service,
+      :metric => current_state[:free]
+    }) if plugin.report_free
+    # постим сообение о том сколько у нас вообще памяти
+    report({
+      :service => plugin.service,
+      :metric => current_state[:total]
+    }) if plugin.report_total
+  end
+
+end
+
+Riemann::Babbler::Memory.run

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

@@ -0,0 +1,51 @@
+class Riemann::Babbler::Net
+  include Riemann::Babbler
+
+  def plugin
+    options.plugins.net
+  end
+
+  def net
+    f = File.read('/proc/net/dev')
+    net = 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 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
+      
+      puts s
+    end
+  end
+
+  def tick
+    status = {
+      :service => plugin.service,
+      :state => 'ok'
+    }
+    report status
+  end
+
+end
+
+Riemann::Babbler::Net.run

data/lib/riemann/version.rb

@@ -1,3 +1,3 @@
 module Babbler
-    VERSION = '0.0.2'
+    VERSION = '0.0.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: riemann-babbler
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -95,19 +95,25 @@ description: Some plugins mannager for riemann.
 email:
 - vadv.mkn@gmail.com
 executables:
-- riemann-babbler.rb
+- riemann-babbler
 extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
 - Gemfile
 - Gemfile.lock
+- README.md
 - Rakefile
-- bin/riemann-babbler.rb
+- bin/riemann-babbler
 - config.yml
 - lib/deep_merge.rb
 - lib/riemann/babbler/plugin.rb
-- lib/riemann/babbler/plugins/dummy.rb
+- lib/riemann/babbler/plugins/babbler.rb
+- lib/riemann/babbler/plugins/cpu.rb
+- lib/riemann/babbler/plugins/disk.rb
+- lib/riemann/babbler/plugins/la.rb
+- lib/riemann/babbler/plugins/memory.rb
+- lib/riemann/babbler/plugins/net.rb
 - lib/riemann/version.rb
 - riemann-babbler.gemspec
 homepage: https://github.com/vadv/riemann-babbler
@@ -123,9 +129,6 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -110392361130597825
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:

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

@@ -1,25 +0,0 @@
-class Riemann::Babbler::Dummy
-  include Riemann::Babbler
-
-  def plugin
-    options.plugins.dummy
-  end
-
-  def status
-    Random.rand(100)
-  end
-
-  def tick
-    current_status = status
-    status = {
-      :service => plugin.service,
-      :metric => current_status,
-      :state => state(current_status)
-    }
-    log.debug "Report status: #{status.inspect}"
-    report status
-  end
-
-end
-
-Riemann::Babbler::Dummy.run