vmpooler/lib/vmpooler/graphite.rb

# frozen_string_literal: true

require 'rubygems' unless defined?(Gem)

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

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

      @server = params['server']
      @port   = params['port'] || 2003
      @prefix = params['prefix'] || 'vmpooler'
    end

    def increment(label)
      log label, 1
    end

    def gauge(label, value)
      log label, value
    end

    def timing(label, duration)
      log label, duration
    end

    def log(path, value)
      Thread.new do
        socket = TCPSocket.new(server, port)
        begin
          socket.puts "#{prefix}.#{path} #{value} #{Time.now.to_i}"
        ensure
          socket.close
        end
      end
    rescue => err
      $stderr.puts "Failure logging #{path} to graphite server [#{server}:#{port}]: #{err}"
    end
  end
end