require 'socket' module Fozzie module Socket RESERVED_CHARS_REGEX = /[\:\|\@\s]/ private # Send the statistic to the server # # Creates the Statsd key from the given values, and sends to socket (depending on sample rate) # def send(stat, delta, type, sample_rate) stat = [stat].flatten.compact.collect(&:to_s).join('.').downcase stat = stat.gsub('::', '.').gsub(RESERVED_CHARS_REGEX, '_') k = [Fozzie.c.data_prefix, stat].compact.join('.') k << ":" k << [delta, type].join('|') k << '@%s' % sample_rate.to_s if sample_rate < 1 sampled(sample_rate) { send_to_socket(k.strip) } end # If the statistic is sampled, generate a condition to check if it's good to send def sampled(sample_rate) yield unless sample_rate < 1 and rand > sample_rate end # Send data to the server via the socket def send_to_socket(message) begin Fozzie.logger.debug {"Statsd: #{message}"} if Fozzie.logger Timeout.timeout(Fozzie.c.timeout) { res = socket.send(message, 0, Fozzie.c.host, Fozzie.c.port) Fozzie.logger.debug {"Statsd sent: #{res}"} if Fozzie.logger (res.to_i == message.length) } rescue => exc Fozzie.logger.debug {"Statsd Failure: #{exc.message}\n#{exc.backtrace}"} if Fozzie.logger false end end # The Socket we want to use to send data def socket @socket ||= ::UDPSocket.new end end end