class Graphite

# File lib/graphite.rb, line 4
def initialize(token, host, port = 2003)
  @token, @host, @port = token, host, port
end

# File lib/graphite.rb, line 8
def measure(stat, value = nil)
  if block_given?
    start = Time.now
    yield
    value = ((Time.now - start) * 1000).round
  end

  send_to_socket("#{stat} #{value}") unless value.nil?
end

# File lib/graphite.rb, line 20
def send_to_socket(message)
  return if defined?(Rails) && !Rails.env.production?

  socket.send("#{@token}.#{message}", 0, @host, @port) rescue nil
end

# File lib/graphite.rb, line 26
def socket
  @socket ||= UDPSocket.new
end