Contents

require 'riemann/client'
require 'resolv'
require 'socket'

module Riemann
  module Babbler
    class Sender
      class Client

        include Riemann::Babbler::Logging
        include Riemann::Babbler::Options
        include Riemann::Babbler::Errors

        INTERVAL_FLUSH = 0.1

        attr_accessor :host, :port, :events

        def initialize(host)
          @host, @port   = host.split(':')
          @port ||= opts.riemann.port
          @events = Array.new
          start
        end

        def start
          build_client
          @running = true
          @runner = Thread.new do
            while @running
              sleep INTERVAL_FLUSH
              flush
            end
            @runner = nil
          end
        end

        def stop
          @running = false
        end

        def alive?
          @runner.nil? || @runner.alive?
        end

        def <<(event)
          @events.shift if @events.size > opts.riemann.backlog
          @events << event
        end

        private

        # flush events
        def flush
          return nil if @events.empty?
          while @events.size > 0
            event = @events[0]
            log :debug, "Post event via #{@host}:#{@port} : #{event.inspect}"
            Timeout::timeout(opts.riemann.timeout) {
              @riemann << event
            }
            @events.shift
          end
        end

        # riemann client
        def build_client
          @riemann = nil 
          @riemann = Riemann::Client.new(:host => resolv(@host), :port => @port, :timeout => opts.riemann.timeout)
          @riemann = @riemann.tcp if opts.riemann.tcp
          @riemann
        end

        #@return ipaddress of riemann server
        def resolv(host)
          ip = Resolv.new.getaddress(host)
          log :debug, "Resolv host: #{host} => #{ip}"
          ip
        end

      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
riemann-babbler-2.5.2	lib/riemann/babbler/sender_client.rb