lib/riemann/tools/haproxy.rb in riemann-tools-1.6.0 vs lib/riemann/tools/haproxy.rb in riemann-tools-1.7.0

@@ -10,10 +10,11 @@
       require 'net/http'
       require 'csv'
 
       opt :stats_url, 'Full url to haproxy stats (eg: https://user:[e-mail]:9999/stats)', required: true,
                                                                                                    type: :string
+      opt :user_agent, 'User-Agent header for HTTP requests', short: :none, default: "#{File.basename($PROGRAM_NAME)}/#{Riemann::Tools::VERSION} (+https://github.com/riemann/riemann-tools)"
 
       def initialize
         @uri = URI("#{opts[:stats_url]};csv")
       end
 
@@ -43,10 +44,10 @@
 
       def csv
         http = ::Net::HTTP.new(@uri.host, @uri.port)
         http.use_ssl = true if @uri.scheme == 'https'
         res = http.start do |h|
-          get = ::Net::HTTP::Get.new(@uri.request_uri)
+          get = ::Net::HTTP::Get.new(@uri.request_uri, { 'user-agent' => opts[:user_agent] })
           unless @uri.userinfo.nil?
             userinfo = @uri.userinfo.split(':')
             get.basic_auth userinfo[0], userinfo[1]
           end
           h.request get