riemann-babbler 0.0.1

data/Gemfile

@@ -0,0 +1,13 @@
+source :rubygems
+
+gem "riemann-client"
+gem "configatron"
+gem "logger"
+gem "trollop"
+gem "parallel"
+
+gem "rspec"
+gem "rake"
+
+gem "pry"
+gem "awesome_print"

data/Gemfile.lock

@@ -0,0 +1,47 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    awesome_print (1.1.0)
+    beefcake (0.3.7)
+    coderay (1.0.9)
+    configatron (2.10.0)
+      yamler (>= 0.1.0)
+    diff-lcs (1.2.2)
+    logger (1.2.8)
+    method_source (0.8.1)
+    mtrc (0.0.4)
+    parallel (0.6.4)
+    pry (0.9.12)
+      coderay (~> 1.0.5)
+      method_source (~> 0.8)
+      slop (~> 3.4)
+    rake (10.0.4)
+    riemann-client (0.2.1)
+      beefcake (>= 0.3.5)
+      mtrc (>= 0.0.4)
+      trollop (>= 1.16.2)
+    rspec (2.13.0)
+      rspec-core (~> 2.13.0)
+      rspec-expectations (~> 2.13.0)
+      rspec-mocks (~> 2.13.0)
+    rspec-core (2.13.1)
+    rspec-expectations (2.13.0)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.13.0)
+    slop (3.4.3)
+    trollop (2.0)
+    yamler (0.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  awesome_print
+  configatron
+  logger
+  parallel
+  pry
+  rake
+  riemann-client
+  rspec
+  trollop

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/riemann-babbler.rb

@@ -0,0 +1,66 @@
+#!/usr/bin/env ruby
+#encoding: utf-8
+
+require 'trollop'
+require 'yaml'
+require 'configatron'
+require 'parallel'
+require 'logger'
+
+require File.expand_path('../../lib/riemann/version', __FILE__)
+require File.expand_path('../../lib/deep_merge', __FILE__)
+require File.expand_path('../../lib/riemann/babbler/plugin', __FILE__)
+
+opts = Trollop::options do
+  version "Riemann babbler #{Babbler::VERSION}"
+  banner <<-EOS
+Riemann-babbler is plugin manager for riemann-tools.
+
+Usage:
+       riemann-babbler [options] 
+where [options] are:
+EOS
+
+  opt :config, "Config file", :default => "/etc/riemann-babbler/config.yml"
+end
+
+# logger
+log = Logger.new(STDOUT)
+
+# merge configs
+config_file =  if File.exist?( opts[:config] )
+  YAML.load_file( opts[:config] ).to_hash
+else
+  log.error "Can't load config file #{opts[:config]}"
+  Hash.new
+end
+
+config_default = YAML.load_file( File.expand_path('../../config.yml', __FILE__) )
+
+config = config_default.deep_merge( config_file )
+configatron.configure_from_hash config
+
+# отправляем к плагинам
+$configatron = configatron
+$logger = log
+
+# get plugins
+plugins = []
+default_plugins_dir = File.expand_path('../../lib/riemann/babbler/plugins/', __FILE__)
+Dir.glob( default_plugins_dir + "/*.rb" ) do |file|
+  plugins <<  file
+end
+
+unless configatron.plugins.dirs.nil?
+  configatron.plugins.dirs.each do |dir|
+    next unless Dir.exist? dir
+    Dir.glob( dir + "/*.rb" ) do |file|
+      plugins << file
+    end
+  end
+end
+
+# start plugins
+Parallel.each( plugins, :in_threads => plugins.count ) do |plugin|
+  require "#{plugin}"
+end

data/config.yml

@@ -0,0 +1,16 @@
+riemann:
+  host: riemann-testing-01.undev.cc
+  port: 5555
+  tags:
+    - prod
+    - nptv
+
+plugins:
+  dirs:
+    - /etc/riemann-babbler/plugins
+  dummy:
+    service: dummy service
+    interval: 1
+    states:
+      warning: 70
+      critical: 90

data/lib/deep_merge.rb

@@ -0,0 +1,48 @@
+# https://raw.github.com/Offirmo/hash-deep-merge/master/lib/hash_deep_merge.rb
+class Hash
+
+  def deep_merge!(specialized_hash)
+    return internal_deep_merge!(self, specialized_hash)
+  end
+
+
+  def deep_merge(specialized_hash)
+    return internal_deep_merge!(Hash.new.replace(self), specialized_hash)
+  end
+
+
+  protected
+
+  # better, recursive, preserving method
+  # OK OK this is not the most efficient algorithm,
+  # but at last it's *perfectly clear and understandable*
+  # so fork and improve if you need 5% more speed, ok ?
+  def internal_deep_merge!(source_hash, specialized_hash)
+
+    #puts "starting deep merge..."
+
+    specialized_hash.each_pair do |rkey, rval|
+      #puts "   potential replacing entry : " + rkey.inspect
+
+      if source_hash.has_key?(rkey) then
+        #puts "   found potentially conflicting entry for #{rkey.inspect} : #{rval.inspect}, will merge :"
+        if rval.is_a?(Hash) and source_hash[rkey].is_a?(Hash) then
+          #puts "      recursing..."
+          internal_deep_merge!(source_hash[rkey], rval)
+        elsif rval == source_hash[rkey] then
+          #puts "      same value, skipping."
+        else
+          #puts "      replacing."
+          source_hash[rkey] = rval
+        end
+      else
+        #puts "   found new entry #{rkey.inspect}, adding it..."
+        source_hash[rkey] = rval
+      end
+    end
+
+    #puts "deep merge done."
+
+    return source_hash
+  end
+end

data/lib/riemann/babbler/plugin.rb

@@ -0,0 +1,77 @@
+module Riemann
+  module Babbler
+
+    require 'riemann/client'
+
+    def self.included(base)
+      base.instance_eval do
+        def run
+          new.run
+        end
+
+      end
+    end
+
+    def initialize
+      @configatron = $configatron
+      super
+    end
+
+    def log
+      @logger ||= $logger
+    end
+
+    def options
+      @configatron
+    end
+    alias :opts :options
+
+    def report(event)
+      event[:tags] unless options.riemann.tags.nil?
+      riemann << event
+    end
+
+    def riemann
+      @riemann ||= Riemann::Client.new(
+        :host => options.riemann.host,
+        :port => options.riemann.port
+      )
+    end
+    alias :r :riemann
+
+    def run
+      t0 = Time.now
+      loop do
+        begin
+          tick
+        rescue => e
+          $stderr.puts "#{e.class} #{e}\n#{e.backtrace.join "\n"}"
+        end
+
+        # Sleep.
+        sleep(plugin.interval - ((Time.now - t0) % plugin.interval))
+      end
+    end
+
+    # Переодически вызываемое действие
+    def tick
+    end
+
+    # Доступ к конфигу определенного плагина
+    def plugin
+    end
+
+    # описание статуса
+    def state(my_state)
+      case
+      when my_state.between?(plugin.states.warning, plugin.states.critical)
+        'warning'
+      when my_state > plugin.states.warning
+        'critical'
+      else
+        'ok'
+      end
+    end
+
+  end
+end

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

@@ -0,0 +1,25 @@
+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

data/lib/riemann/version.rb

@@ -0,0 +1,3 @@
+module Babbler
+    VERSION = '0.0.1'
+end

data/riemann-babbler.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/riemann/version', __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "riemann-babbler"
+  s.version     = Babbler::VERSION
+  s.authors     = ["Vasiliev D.V."]
+  s.email       = %w(vadv.mkn@gmail.com)
+  s.homepage    = "https://github.com/vadv/riemann-babbler"
+  s.summary     = %q{Riemann health cheker.}
+  s.description = %q{Some plugins mannager for riemann.}
+  s.licenses    = %w(MIT)
+  
+  s.add_dependency('riemann-client')
+  s.add_dependency('configatron')
+  s.add_dependency('logger')
+  s.add_dependency('trollop')
+  s.add_dependency('yaml')
+  s.add_dependency('parallel')
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = %w(lib)
+end

metadata

@@ -0,0 +1,156 @@
+--- !ruby/object:Gem::Specification
+name: riemann-babbler
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Vasiliev D.V.
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-04-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: riemann-client
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: configatron
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yaml
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: parallel
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Some plugins mannager for riemann.
+email:
+- vadv.mkn@gmail.com
+executables:
+- riemann-babbler.rb
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- Rakefile
+- bin/riemann-babbler.rb
+- config.yml
+- lib/deep_merge.rb
+- lib/riemann/babbler/plugin.rb
+- lib/riemann/babbler/plugins/dummy.rb
+- lib/riemann/version.rb
+- riemann-babbler.gemspec
+homepage: https://github.com/vadv/riemann-babbler
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 4400440854509407224
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Riemann health cheker.
+test_files: []