Add optional ports

github · Jan 20, 2015 · 116c12e · 116c12e
1 parent affe1b9
commit 116c12e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/statsd.rb b/lib/statsd.rb
@@ -48,8 +48,9 @@ def initialize(host, port=8125, key=nil)
     add_host(host, port, key)
   end
 
-  def add_host(host, port, key = nil)
-    @hosts << Host.new(host, port, key)
+  def add_host(host, port = nil, key = nil)
+    host, port = host.split(':') if host.include?(':')
+    @hosts << Host.new(host, port.to_i, key)
   end
 
   # Sends an increment (count = 1) for the given stat to the statsd server.