lib/measures/client.rb in measures-1.0.0 vs lib/measures/client.rb in measures-2.0.0

@@ -1,15 +1,13 @@
-require "socket"
 require "json"
 require "benchmark"
 
 module Measures
   class Client
-    def initialize(client, host, port)
+    def initialize(transport, client)
       @client = client
-      @socket = UDPSocket.new
-      @socket.connect(host, port)
+      @transport = transport
     end
 
     def count(metric, data = {})
       data.merge!({ "count" => 1 })
 
@@ -34,9 +32,9 @@
         "metric" => metric,
       }
 
       default.merge!(data)
 
-      @socket.send(default.to_json, 0)
+      @transport.send(default.to_json)
     end
   end
 end