dogstatsd-ruby 3.3.0 → 4.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 046a0af9f230451c91fdeb9ae128c550b1c20623fa017d326e63a2f053f7cc9a
-  data.tar.gz: e2ba838497de521e89d386645a0228f819fdd03c28acc05a4789aa7f8d979de8
+  metadata.gz: fde2be6b614574c2555c8cbbd2ca16ef33bcd09837502f482f5f6baf582fd67b
+  data.tar.gz: cc1523c5aa78643d93fc891116c46bb5322dda297f1676cd73d201c9beee2a26
 SHA512:
-  metadata.gz: f2048376477bd1af31a63b75d17638df999c9f6341f967b2c826177add2a13831a54a72729e2a5c62aa09fa4aca6dccbaba308c79d9adcd470808d3bb80cde59
-  data.tar.gz: 2e0ac5fc90f8a2359f35a754034bb6474c1169ac8c8f3301257555553e1aca4f708cecd4d9245ff835cbe4396561a89d3bd3700ef6cea6c92c343a1e41ca2055
+  metadata.gz: 9c96177f1ea0ab2288712cf9829cb8408687e12ce0f22f06ddbb1433447a3a4eaf05fe13f0fda8746b9f0e8fad71ea3b0b50ba7d8a028e464db334349951c64b
+  data.tar.gz: 11e23d434a7e88925e049484dfbb252d5c78ce12a4a07fd3c09edca57835db30224cba202f86f57352357acdf2073b8c8e93f20bd403975242b201321bef8ecc

data/README.md

@@ -4,7 +4,7 @@ dogstatsd-ruby
 
 A client for DogStatsD, an extension of the StatsD metric server for Datadog.
 
-[![Build Status](https://secure.travis-ci.org/DataDog/dogstatsd-ruby.png)](http://travis-ci.org/DataDog/dogstatsd-ruby)
+[![Build Status](https://secure.travis-ci.org/DataDog/dogstatsd-ruby.svg)](http://travis-ci.org/DataDog/dogstatsd-ruby)
 
 Quick Start Guide
 -----------------
@@ -28,9 +28,10 @@ statsd = Datadog::Statsd.new('localhost', 8125)
 
 # Increment a counter.
 statsd.increment('page.views')
+statsd.increment('messages.count', by: 2, tags: ['kind:incoming'])
 
 # Record a gauge 50% of the time.
-statsd.gauge('users.online', 123, :sample_rate=>0.5)
+statsd.gauge('users.online', 123, sample_rate: 0.5)
 
 # Sample a histogram
 statsd.histogram('file.upload.size', 1234)
@@ -48,7 +49,15 @@ statsd.batch do |s|
 end
 
 # Tag a metric.
-statsd.histogram('query.time', 10, :tags => ["version:1"])
+statsd.histogram('query.time', 10, tags: ['version:1'])
+
+# Tag a metric by passing in a Hash
+statsd.histogram('query.time', 10, :tags => {version: 1})
+
+# Auto-close socket after end of block
+Datadog::Statsd.open('localhost', 8125) do |s|
+  s.increment('page.views')
+end
 ```
 
 You can also post events to your stream. You can tag them, set priority and even aggregate them with other events.
@@ -57,14 +66,34 @@ Aggregation in the stream is made on hostname/event_type/source_type/aggregation
 
 ``` ruby
 # Post a simple message
-statsd.event("There might be a storm tomorrow", "A friend warned me earlier.")
+statsd.event('There might be a storm tomorrow', 'A friend warned me earlier.')
 
 # Cry for help
-statsd.event("SO MUCH SNOW", "Started yesterday and it won't stop !!", :alert_type => "error", :tags => ["urgent", "endoftheworld"])
+statsd.event(
+  'SO MUCH SNOW',
+  "Started yesterday and it won't stop !!",
+  alert_type: 'error',
+  tags: ['urgent', 'endoftheworld']
+)
 ```
 
 
 
+Origin detection over UDP
+-------------
+Origin detection is a method to detect which pod DogStatsD packets are coming from in order to add the pod's tags to the tag list.
+
+To enable origin detection over UDP, add the following lines to your application manifest
+```yaml
+env:
+  - name: DD_ENTITY_ID
+    valueFrom:
+      fieldRef:
+        fieldPath: metadata.uid
+```
+The DogStatsD client attaches an internal tag, `entity_id`. The value of this tag is the content of the `DD_ENTITY_ID` environment variable, which is the pod’s UID.
+
+
 Documentation
 -------------
 

data/lib/datadog/statsd.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 require 'socket'
 
 # = Datadog::Statsd: A DogStatsd client (https://www.datadoghq.com)
@@ -15,12 +16,151 @@ require 'socket'
 #   statsd = Datadog::Statsd.new 'localhost', 8125, :namespace => 'account'
 #   statsd.increment 'activate'
 # @example Create a statsd client with global tags
-#   statsd = Datadog::Statsd.new 'localhost', 8125, :tags => 'tag1:true'
+#   statsd = Datadog::Statsd.new 'localhost', 8125, tags: 'tag1:true'
 module Datadog
   class Statsd
 
-    DEFAULT_HOST = '127.0.0.1'
-    DEFAULT_PORT = 8125
+    class Connection
+      DEFAULT_HOST = '127.0.0.1'
+      DEFAULT_PORT = 8125
+
+      # StatsD host. Defaults to 127.0.0.1.
+      attr_reader :host
+
+      # StatsD port. Defaults to 8125.
+      attr_reader :port
+
+      # DogStatsd unix socket path. Not used by default.
+      attr_reader :socket_path
+
+      # Close the underlying socket
+      def close
+        @socket && @socket.close
+      end
+
+      def write(message)
+        @logger.debug { "Statsd: #{message}" } if @logger
+        send_message(message)
+      rescue StandardError => boom
+        # Try once to reconnect if the socket has been closed
+        retries ||= 1
+        if retries <= 1 &&
+          (boom.is_a?(Errno::ENOTCONN) or
+           boom.is_a?(Errno::ECONNREFUSED) or
+           boom.is_a?(IOError) && boom.message =~ /closed stream/i)
+          retries += 1
+          begin
+            @socket = connect
+            retry
+          rescue StandardError => e
+            boom = e
+          end
+        end
+
+        @logger.error { "Statsd: #{boom.class} #{boom}" } if @logger
+        nil
+      end
+
+      private
+
+      def socket
+        @socket ||= connect
+      end
+    end
+
+    class UDPConnection < Connection
+      def initialize(host, port, logger)
+        @host = host || ENV.fetch('DD_AGENT_HOST', nil) || DEFAULT_HOST
+        @port = port || ENV.fetch('DD_DOGSTATSD_PORT', nil) || DEFAULT_PORT
+        @logger = logger
+      end
+
+      private
+
+      def connect
+        socket = UDPSocket.new
+        socket.connect(@host, @port)
+        socket
+      end
+
+      def send_message(message)
+        socket.send(message, 0)
+      end
+    end
+
+    class UDSConnection < Connection
+      class BadSocketError < StandardError; end
+
+      def initialize(socket_path, logger)
+        @socket_path = socket_path
+        @logger = logger
+      end
+
+      private
+
+      def connect
+        socket = Socket.new(Socket::AF_UNIX, Socket::SOCK_DGRAM)
+        socket.connect(Socket.pack_sockaddr_un(@socket_path))
+        socket
+      end
+
+      def send_message(message)
+        socket.sendmsg_nonblock(message)
+      rescue Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ENOENT => e
+        @socket = nil
+        raise BadSocketError, "#{e.class}: #{e}"
+      end
+    end
+
+    class Batch
+      def initialize(connection, max_buffer_bytes)
+        @connection = connection
+        @max_buffer_bytes = max_buffer_bytes
+        @depth = 0
+        reset
+      end
+
+      def open
+        @depth += 1
+        yield
+      ensure
+        @depth -= 1
+        flush if !open?
+      end
+
+      def open?
+        @depth > 0
+      end
+
+      def add(message)
+        message_bytes = message.bytesize
+
+        unless @buffer_bytes == 0
+          if @buffer_bytes + 1 + message_bytes >= @max_buffer_bytes
+            flush
+          else
+            @buffer << NEW_LINE
+            @buffer_bytes += 1
+          end
+        end
+
+        @buffer << message
+        @buffer_bytes += message_bytes
+      end
+
+      def flush
+        return if @buffer_bytes == 0
+        @connection.write @buffer
+        reset
+      end
+
+      private
+
+      def reset
+        @buffer = String.new
+        @buffer_bytes = 0
+      end
+    end
 
     # Create a dictionary to assign a key to every parameter's name, except for tags (treated differently)
     # Goal: Simple and fast to add some other parameters
@@ -46,78 +186,84 @@ module Datadog
     CRITICAL  = 2
     UNKNOWN   = 3
 
+    MAX_EVENT_SIZE = 8 * 1024
+
     COUNTER_TYPE = 'c'.freeze
     GAUGE_TYPE = 'g'.freeze
     HISTOGRAM_TYPE = 'h'.freeze
     DISTRIBUTION_TYPE = 'd'.freeze
     TIMING_TYPE = 'ms'.freeze
     SET_TYPE = 's'.freeze
-    VERSION = "3.3.0".freeze
+    VERSION = "4.5.0".freeze
 
     # A namespace to prepend to all statsd calls. Defaults to no namespace.
     attr_reader :namespace
 
-    # StatsD host. Defaults to 127.0.0.1.
-    attr_reader :host
-
-    # StatsD port. Defaults to 8125.
-    attr_reader :port
-
-    # DogStatsd unix socket path. Not used by default.
-    attr_reader :socket_path
-
     # Global tags to be added to every statsd call. Defaults to no tags.
     attr_reader :tags
 
     # Buffer containing the statsd message before they are sent in batch
     attr_reader :buffer
 
-    # Maximum number of metrics in the buffer before it is flushed
-    attr_accessor :max_buffer_size
+    # Maximum buffer size in bytes before it is flushed
+    attr_reader :max_buffer_bytes
 
-    class << self
-      # Set to a standard logger instance to enable debug logging.
-      attr_accessor :logger
-    end
+    # Default sample rate
+    attr_reader :sample_rate
 
-    # Return the current version of the library.
-    # deprecated, but cannot be removed since uses might use it to check the version against older releases
-    def self.VERSION
-      VERSION
-    end
+    # Connection
+    attr_reader :connection
 
     # @param [String] host your statsd host
     # @param [Integer] port your statsd port
-    # @option opts [String] :namespace set a namespace to be prepended to every metric name
-    # @option opts [Array<String>] :tags tags to be added to every metric
-    def initialize(host = DEFAULT_HOST, port = DEFAULT_PORT, opts = {}, max_buffer_size=50)
-      self.host, self.port = host, port
-      @socket_path = opts[:socket_path]
-      @prefix = nil
-      @socket = connect_to_socket if @socket_path.nil?
-      self.namespace = opts[:namespace]
-      self.tags = opts[:tags]
-      @buffer = Array.new
-      self.max_buffer_size = max_buffer_size
-      @batch_nesting_depth = 0
-    end
+    # @option [String] namespace set a namespace to be prepended to every metric name
+    # @option [Array<String>|Hash] tags tags to be added to every metric
+    # @option [Logger] logger for debugging
+    # @option [Integer] max_buffer_bytes max bytes to buffer when using #batch
+    # @option [String] socket_path unix socket path
+    # @option [Float] default sample rate if not overridden
+    def initialize(
+      host = nil,
+      port = nil,
+      namespace: nil,
+      tags: nil,
+      max_buffer_bytes: 8192,
+      socket_path: nil,
+      logger: nil,
+      sample_rate: nil
+    )
+      if socket_path.nil?
+        @connection = UDPConnection.new(host, port, logger)
+      else
+        @connection = UDSConnection.new(socket_path, logger)
+      end
+      @logger = logger
 
-    def namespace=(namespace) #:nodoc:
       @namespace = namespace
-      @prefix = namespace.nil? ? nil : "#{namespace}.".freeze
-    end
+      @prefix = @namespace ? "#{@namespace}.".freeze : nil
 
-    def host=(host) #:nodoc:
-      @host = host || DEFAULT_HOST
-    end
+      @sample_rate = sample_rate
 
-    def port=(port) #:nodoc:
-      @port = port || DEFAULT_PORT
-    end
+      unless tags.nil? or tags.is_a? Array or tags.is_a? Hash
+        raise ArgumentError, 'tags must be a Array<String> or a Hash'
+      end
 
-    def tags=(tags) #:nodoc:
-      raise ArgumentError, 'tags must be a Array<String>' unless tags.nil? or tags.is_a? Array
+      tags = tag_hash_to_array(tags) if tags.is_a? Hash
       @tags = (tags || []).compact.map! {|tag| escape_tag_content(tag)}
+
+      # append the entity id to tags if DD_ENTITY_ID env var is not nil
+      @tags << 'dd.internal.entity_id:' + escape_tag_content(ENV.fetch('DD_ENTITY_ID', nil)) unless ENV.fetch('DD_ENTITY_ID', nil).nil?
+
+      @batch = Batch.new @connection, max_buffer_bytes
+    end
+
+    # yield a new instance to a block and close it when done
+    # for short-term use-cases that don't want to close the socket manually
+    def self.open(*args)
+      instance = new(*args)
+      yield instance
+    ensure
+      instance.close
     end
 
     # Sends an increment (count = 1) for the given stat to the statsd server.
@@ -128,7 +274,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @option opts [Numeric] :by increment value, default 1
     # @see #count
-    def increment(stat, opts={})
+    def increment(stat, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       incr_value = opts.fetch(:by, 1)
       count stat, incr_value, opts
@@ -142,7 +288,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @option opts [Numeric] :by decrement value, default 1
     # @see #count
-    def decrement(stat, opts={})
+    def decrement(stat, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       decr_value = - opts.fetch(:by, 1)
       count stat, decr_value, opts
@@ -155,7 +301,7 @@ module Datadog
     # @param [Hash] opts the options to create the metric with
     # @option opts [Numeric] :sample_rate sample rate, 1 for always
     # @option opts [Array<String>] :tags An array of tags
-    def count(stat, count, opts={})
+    def count(stat, count, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       send_stats stat, count, COUNTER_TYPE, opts
     end
@@ -173,7 +319,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @example Report the current user count:
     #   $statsd.gauge('user.count', User.count)
-    def gauge(stat, value, opts={})
+    def gauge(stat, value, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       send_stats stat, value, GAUGE_TYPE, opts
     end
@@ -187,7 +333,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @example Report the current user count:
     #   $statsd.histogram('user.count', User.count)
-    def histogram(stat, value, opts={})
+    def histogram(stat, value, opts=EMPTY_OPTIONS)
       send_stats stat, value, HISTOGRAM_TYPE, opts
     end
 
@@ -203,7 +349,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @example Report the current user count:
     #   $statsd.distribution('user.count', User.count)
-    def distribution(stat, value, opts={})
+    def distribution(stat, value, opts=EMPTY_OPTIONS)
       send_stats stat, value, DISTRIBUTION_TYPE, opts
     end
 
@@ -217,7 +363,7 @@ module Datadog
     # @param [Hash] opts the options to create the metric with
     # @option opts [Numeric] :sample_rate sample rate, 1 for always
     # @option opts [Array<String>] :tags An array of tags
-    def timing(stat, ms, opts={})
+    def timing(stat, ms, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       send_stats stat, ms, TIMING_TYPE, opts
     end
@@ -235,7 +381,7 @@ module Datadog
     # @see #timing
     # @example Report the time (in ms) taken to activate an account
     #   $statsd.time('account.activate') { @account.activate! }
-    def time(stat, opts={})
+    def time(stat, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       start = (PROCESS_TIME_SUPPORTED ? Process.clock_gettime(Process::CLOCK_MONOTONIC) : Time.now.to_f)
       return yield
@@ -243,6 +389,7 @@ module Datadog
       finished = (PROCESS_TIME_SUPPORTED ? Process.clock_gettime(Process::CLOCK_MONOTONIC) : Time.now.to_f)
       timing(stat, ((finished - start) * 1000).round, opts)
     end
+
     # Sends a value to be tracked as a set to the statsd server.
     #
     # @param [String] stat stat name.
@@ -252,7 +399,7 @@ module Datadog
     # @option opts [Array<String>] :tags An array of tags
     # @example Record a unique visitory by id:
     #   $statsd.set('visitors.uniques', User.id)
-    def set(stat, value, opts={})
+    def set(stat, value, opts=EMPTY_OPTIONS)
       opts = {:sample_rate => opts} if opts.is_a? Numeric
       send_stats stat, value, SET_TYPE, opts
     end
@@ -262,37 +409,14 @@ module Datadog
     # @param [String] name Service check name
     # @param [String] status Service check status.
     # @param [Hash] opts the additional data about the service check
-      # @option opts [Integer, nil] :timestamp (nil) Assign a timestamp to the event. Default is now when none
-      # @option opts [String, nil] :hostname (nil) Assign a hostname to the event.
+      # @option opts [Integer, String, nil] :timestamp (nil) Assign a timestamp to the service check. Default is now when none
+      # @option opts [String, nil] :hostname (nil) Assign a hostname to the service check.
       # @option opts [Array<String>, nil] :tags (nil) An array of tags
       # @option opts [String, nil] :message (nil) A message to associate with this service check status
     # @example Report a critical service check status
     #   $statsd.service_check('my.service.check', Statsd::CRITICAL, :tags=>['urgent'])
-    def service_check(name, status, opts={})
-      service_check_string = format_service_check(name, status, opts)
-      send_to_socket service_check_string
-    end
-
-    def format_service_check(name, status, opts={})
-      sc_string = "_sc|#{name}|#{status}"
-
-      SC_OPT_KEYS.each do |key, shorthand_key|
-        next unless opts[key]
-
-        if key == :tags
-          if tags_string = tags_as_string(opts)
-            sc_string << "|##{tags_string}"
-          end
-        elsif key == :message
-          message = remove_pipes(opts[:message])
-          escaped_message = escape_service_check_message(message)
-          sc_string << "|m:#{escaped_message}"
-        else
-          value = remove_pipes(opts[key])
-          sc_string << "|#{shorthand_key}#{value}"
-        end
-      end
-      return sc_string
+    def service_check(name, status, opts=EMPTY_OPTIONS)
+      send_stat format_service_check(name, status, opts)
     end
 
     # This end point allows you to post events to the stream. You can tag them, set priority and even aggregate them with other events.
@@ -302,9 +426,9 @@ module Datadog
     # it will be grouped with other events that don't have an event type.
     #
     # @param [String] title Event title
-    # @param [String] text Event text. Supports \n
+    # @param [String] text Event text. Supports newlines (+\n+)
     # @param [Hash] opts the additional data about the event
-    # @option opts [Integer, nil] :date_happened (nil) Assign a timestamp to the event. Default is now when none
+    # @option opts [Integer, String, nil] :date_happened (nil) Assign a timestamp to the event. Default is now when none
     # @option opts [String, nil] :hostname (nil) Assign a hostname to the event.
     # @option opts [String, nil] :aggregation_key (nil) Assign an aggregation key to the event, to group it with some others
     # @option opts [String, nil] :priority ('normal') Can be "normal" or "low"
@@ -313,11 +437,8 @@ module Datadog
     # @option opts [Array<String>] :tags tags to be added to every metric
     # @example Report an awful event:
     #   $statsd.event('Something terrible happened', 'The end is near if we do nothing', :alert_type=>'warning', :tags=>['end_of_times','urgent'])
-    def event(title, text, opts={})
-      event_string = format_event(title, text, opts)
-      raise "Event #{title} payload is too big (more that 8KB), event discarded" if event_string.length > 8 * 1024
-
-      send_to_socket event_string
+    def event(title, text, opts=EMPTY_OPTIONS)
+      send_stat format_event(title, text, opts)
     end
 
     # Send several metrics in the same UDP Packet
@@ -328,24 +449,73 @@ module Datadog
     #      s.gauge('users.online',156)
     #      s.increment('page.views')
     #    end
-    def batch()
-      @batch_nesting_depth += 1
-      yield self
-    ensure
-      @batch_nesting_depth -= 1
-      flush_buffer if @batch_nesting_depth == 0
+    def batch
+      @batch.open { yield self }
+    end
+
+    # Close the underlying socket
+    def close
+      @connection.close
+    end
+
+    private
+
+    NEW_LINE = "\n".freeze
+    ESC_NEW_LINE = "\\n".freeze
+    COMMA = ",".freeze
+    PIPE = "|".freeze
+    DOT = ".".freeze
+    DOUBLE_COLON = "::".freeze
+    UNDERSCORE = "_".freeze
+    PROCESS_TIME_SUPPORTED = (RUBY_VERSION >= "2.1.0")
+    EMPTY_OPTIONS = {}.freeze
+
+    private_constant :NEW_LINE, :ESC_NEW_LINE, :COMMA, :PIPE, :DOT,
+      :DOUBLE_COLON, :UNDERSCORE, :EMPTY_OPTIONS
+
+    def format_service_check(name, status, opts=EMPTY_OPTIONS)
+      sc_string = "_sc|#{name}|#{status}".dup
+
+      SC_OPT_KEYS.each do |key, shorthand_key|
+        next unless opts[key]
+
+        if key == :tags
+          if tags_string = tags_as_string(opts)
+            sc_string << "|##{tags_string}"
+          end
+        elsif key == :message
+          message = remove_pipes(opts[:message])
+          escaped_message = escape_service_check_message(message)
+          sc_string << "|m:#{escaped_message}"
+        else
+          if key == :timestamp && opts[key].is_a?(Integer)
+            value = opts[key]
+          else
+            value = remove_pipes(opts[key])
+          end
+          sc_string << "|#{shorthand_key}#{value}"
+        end
+      end
+      sc_string
     end
 
-    def format_event(title, text, opts={})
+    def format_event(title, text, opts=EMPTY_OPTIONS)
       escaped_title = escape_event_content(title)
       escaped_text = escape_event_content(text)
-      event_string_data = "_e{#{escaped_title.length},#{escaped_text.length}}:#{escaped_title}|#{escaped_text}"
+      event_string_data = "_e{#{escaped_title.bytesize},#{escaped_text.bytesize}}:#{escaped_title}|#{escaped_text}".dup
 
       # We construct the string to be sent by adding '|key:value' parts to it when needed
       # All pipes ('|') in the metadata are removed. Title and Text can keep theirs
       OPTS_KEYS.each do |key, shorthand_key|
         if key != :tags && opts[key]
-          value = remove_pipes(opts[key])
+          # :date_happened is the only key where the value is an Integer
+          # To not break backwards compatibility, we still accept a String
+          if key == :date_happened && opts[key].is_a?(Integer)
+              value = opts[key]
+          # All other keys only have String values
+          else
+              value = remove_pipes(opts[key])
+          end
           event_string_data << "|#{shorthand_key}:#{value}"
         end
       end
@@ -355,36 +525,25 @@ module Datadog
         event_string_data << "|##{tags_string}"
       end
 
-      raise "Event #{title} payload is too big (more that 8KB), event discarded" if event_string_data.length > 8192 # 8 * 1024 = 8192
-      return event_string_data
-    end
-
-    # Close the underlying socket
-    def close()
-      @socket.close
+      raise "Event #{title} payload is too big (more that 8KB), event discarded" if event_string_data.bytesize > MAX_EVENT_SIZE
+      event_string_data
     end
 
-    private
-
-    NEW_LINE = "\n".freeze
-    ESC_NEW_LINE = "\\n".freeze
-    COMMA = ",".freeze
-    PIPE = "|".freeze
-    DOT = ".".freeze
-    DOUBLE_COLON = "::".freeze
-    UNDERSCORE = "_".freeze
-    PROCESS_TIME_SUPPORTED = (RUBY_VERSION >= "2.1.0")
-
-    private_constant :NEW_LINE, :ESC_NEW_LINE, :COMMA, :PIPE, :DOT,
-      :DOUBLE_COLON, :UNDERSCORE
-
     def tags_as_string(opts)
-      tag_arr = opts[:tags] || []
-      tag_arr = tag_arr.map { |tag| escape_tag_content(tag) }
-      tag_arr = tags + tag_arr # @tags are normalized when set, so not need to normalize them again
+      if tag_arr = opts[:tags]
+        tag_arr = tag_hash_to_array(tag_arr) if tag_arr.is_a? Hash
+        tag_arr = tag_arr.map { |tag| escape_tag_content(tag) }
+        tag_arr = tags + tag_arr # @tags are normalized when set, so not need to normalize them again
+      else
+        tag_arr = tags
+      end
       tag_arr.join(COMMA) unless tag_arr.empty?
     end
 
+    def tag_hash_to_array(tag_hash)
+      tag_hash.to_a.map {|pair| pair.compact.join(":")}
+    end
+
     def escape_event_content(msg)
       msg.gsub NEW_LINE, ESC_NEW_LINE
     end
@@ -403,10 +562,10 @@ module Datadog
       escape_event_content(msg).gsub('m:'.freeze, 'm\:'.freeze)
     end
 
-    def send_stats(stat, delta, type, opts={})
-      sample_rate = opts[:sample_rate] || 1
-      if sample_rate == 1 or rand < sample_rate
-        full_stat = ''
+    def send_stats(stat, delta, type, opts=EMPTY_OPTIONS)
+      sample_rate = opts[:sample_rate] || @sample_rate || 1
+      if sample_rate == 1 or rand <= sample_rate
+        full_stat = ''.dup
         full_stat << @prefix if @prefix
 
         stat = stat.is_a?(String) ? stat.dup : stat.to_s
@@ -437,62 +596,11 @@ module Datadog
     end
 
     def send_stat(message)
-      if @batch_nesting_depth > 0
-        @buffer << message
-        flush_buffer if @buffer.length >= @max_buffer_size
+      if @batch.open?
+        @batch.add message
       else
-        send_to_socket(message)
-      end
-    end
-
-    def flush_buffer
-      return @buffer if @buffer.empty?
-      send_to_socket(@buffer.join(NEW_LINE))
-      @buffer = Array.new
-    end
-
-    def connect_to_socket
-      if !@socket_path.nil?
-        socket = Socket.new(Socket::AF_UNIX, Socket::SOCK_DGRAM)
-        socket.connect(Socket.pack_sockaddr_un(@socket_path))
-      else
-        socket = UDPSocket.new
-        socket.connect(@host, @port)
+        @connection.write(message)
       end
-      socket
-    end
-
-    def sock
-      @socket ||= connect_to_socket
-    end
-
-    def send_to_socket(message)
-      self.class.logger.debug { "Statsd: #{message}" } if self.class.logger
-      if @socket_path.nil?
-        sock.send(message, 0)
-      else
-        sock.sendmsg_nonblock(message)
-      end
-    rescue => boom
-      if @socket_path && (boom.is_a?(Errno::ECONNREFUSED) ||
-                          boom.is_a?(Errno::ECONNRESET) ||
-                          boom.is_a?(Errno::ENOENT))
-        return @socket = nil
-      end
-      # Try once to reconnect if the socket has been closed
-      retries ||= 1
-      if retries <= 1 && boom.is_a?(IOError) && boom.message =~ /closed stream/i
-        retries += 1
-        begin
-          @socket = connect_to_socket
-          retry
-        rescue => e
-          boom = e
-        end
-      end
-
-      self.class.logger.error { "Statsd: #{boom.class} #{boom}" } if self.class.logger
-      nil
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dogstatsd-ruby
 version: !ruby/object:Gem::Version
-  version: 3.3.0
+  version: 4.5.0
 platform: ruby
 authors:
 - Rein Henrichs
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-02-04 00:00:00.000000000 Z
+date: 2019-08-22 00:00:00.000000000 Z
 dependencies: []
 description: A Ruby DogStastd client
 email: code@datadoghq.com
@@ -33,7 +33,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -41,7 +41,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.7.1
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
 summary: A Ruby DogStatsd client