lib/fluent/plugin/out_statsd.rb in fluent-plugin-statsd-output-1.2.0 vs lib/fluent/plugin/out_statsd.rb in fluent-plugin-statsd-output-1.3.0

@@ -24,11 +24,11 @@
       super
     end
 
     def configure(conf)
       super
-      @statsd = Statsd.new(host, port)
+      @statsd = Statsd::Batch.new(Statsd.new(host, port))
       @statsd.namespace = namespace if namespace
 
       if batch_byte_size
         @statsd.batch_size = nil
         @statsd.batch_byte_size = batch_byte_size
@@ -43,10 +43,11 @@
       super
     end
 
     def shutdown
       super
+      @statsd.flush
     end
 
     def format(tag, time, record)
       [tag, record].to_msgpack
     end
@@ -58,9 +59,10 @@
         @metrics.each do |metric|
           arg_names = %w{statsd_type statsd_key statsd_val}
           send_to_statsd(*metric.values_at(*arg_names).map {|str| parser.parse(str) })
         end
       end
+      @statsd.flush
     end
 
 
     private