vmpooler/lib/vmpooler/statsd.rb

require 'rubygems' unless defined?(Gem)
require 'statsd-ruby'

module Vmpooler
  class Statsd
    attr_reader :server, :port, :prefix

    def initialize(params = {})
      if params[:server].nil? || params[:server].empty?
        raise ArgumentError, "Statsd server is required. Config: #{params.inspect}"
      end

      host    = params[:server]
      @port    = params[:port]   || 8125
      @prefix = params[:prefix] || 'vmpooler'
      @server = Statsd.new(host, @port)
    end

    def increment(label)
      server.increment(prefix + "." + label)
    rescue => err
      $stderr.puts "Failure incrementing #{prefix}.#{label} on statsd server [#{server}:#{port}]: #{err}"
    end

    def gauge(label, value)
      server.gauge(prefix + "." + label, value)
    rescue => err
      $stderr.puts "Failure updating gauge #{prefix}.#{label} on statsd server [#{server}:#{port}]: #{err}"
    end

    def timing(label, duration)
      server.timing(prefix + "." + label, duration)
    rescue => err
      $stderr.puts "Failure updating timing #{prefix}.#{label} on statsd server [#{server}:#{port}]: #{err}"
    end
  end

  class DummyStatsd
    attr_reader :server, :port, :prefix

    def initialize(params = {})
    end

    def increment(label)
      true
    end

    def gauge(label, value)
      true
    end

    def timing(label, duration)
      true
    end
  end
end