diff --git a/lib/riemann/tools.rb b/lib/riemann/tools.rb index 2cd138cf..c7574461 100644 --- a/lib/riemann/tools.rb +++ b/lib/riemann/tools.rb @@ -63,7 +63,7 @@ def report(event) event[:tags] = [*event.fetch(:tags, [])] + options[:tag].map(&:dup) end - event[:ttl] ||= (options[:ttl] || (options[:interval] * 2)) + event[:ttl] ||= options[:ttl] || (options[:interval] * 2) event[:host] = options[:event_host].dup if options[:event_host] diff --git a/lib/riemann/tools/bench.rb b/lib/riemann/tools/bench.rb index 4e371531..6e27d798 100644 --- a/lib/riemann/tools/bench.rb +++ b/lib/riemann/tools/bench.rb @@ -15,7 +15,7 @@ def initialize @hosts = %w[a b c d e f g h i j] @services = %w[test1 test2 test3 foo bar baz xyzzy attack cat treat] @states = {} - @client = Riemann::Client.new(host: (ARGV.first || 'localhost')) + @client = Riemann::Client.new(host: ARGV.first || 'localhost') end def evolve(state) diff --git a/lib/riemann/tools/haproxy.rb b/lib/riemann/tools/haproxy.rb index a3848561..0728da78 100644 --- a/lib/riemann/tools/haproxy.rb +++ b/lib/riemann/tools/haproxy.rb @@ -44,6 +44,10 @@ def tick end def csv + CSV.parse(body.split('# ')[1], headers: true) + end + + def body http = ::Net::HTTP.new(@uri.host, @uri.port) http.use_ssl = true if @uri.scheme == 'https' res = http.start do |h| @@ -53,8 +57,8 @@ def csv get.basic_auth userinfo[0], userinfo[1] end h.request get + res.body end - CSV.parse(res.body.split('# ')[1], { headers: true }) end end end diff --git a/lib/riemann/tools/net.rb b/lib/riemann/tools/net.rb index 38b406b3..3fd92120 100644 --- a/lib/riemann/tools/net.rb +++ b/lib/riemann/tools/net.rb @@ -108,7 +108,7 @@ def tick if @old_state # Report services from `@old_state` that don't exist in `state` as expired - @old_state.reject { |k| state.key?(k) }.each do |service, _metric| + @old_state.reject { |k| state.key?(k) }.each_key do |service| report(service: service.dup, state: 'expired') end diff --git a/spec/riemann/tools/haproxy_spec.rb b/spec/riemann/tools/haproxy_spec.rb new file mode 100644 index 00000000..c3ce67c6 --- /dev/null +++ b/spec/riemann/tools/haproxy_spec.rb @@ -0,0 +1,24 @@ +# frozen_string_literal: true + +require 'riemann/tools/haproxy' + +RSpec.describe Riemann::Tools::Haproxy do + context('#tick') do + before do + ARGV.replace(['--stats-url', 'http://localhost']) + + allow(subject).to receive(:body).and_return(<<~DATA) + # pxname,svname,qcur,qmax,scur,smax,slim,stot,bin,bout,dreq,dresp,ereq,econ,eresp,wretr,wredis,status,weight,act,bck,chkfail,chkdown,lastchg,downtime,qlimit,pid,iid,sid,throttle,lbtot,tracked,type,rate,rate_lim,rate_max,check_status,check_code,check_duration,hrsp_1xx,hrsp_2xx,hrsp_3xx,hrsp_4xx,hrsp_5xx,hrsp_other,hanafail,req_rate,req_rate_max,req_tot,cli_abrt,srv_abrt,comp_in,comp_out,comp_byp,comp_rsp,lastsess,last_chk,last_agt,qtime,ctime,rtime,ttime,agent_status,agent_code,agent_duration,check_desc,agent_desc,check_rise,check_fall,check_health,agent_rise,agent_fall,agent_health,addr,cookie,mode,algo,conn_rate,conn_rate_max,conn_tot,intercepted,dcon,dses,wrew,connect,reuse,cache_lookups,cache_hits,srv_icur,src_ilim,qtime_max,ctime_max,rtime_max,ttime_max,eint,idle_conn_cur,safe_conn_cur,used_conn_cur,need_conn_est, + ft-http,FRONTEND,,,0,24,8000,911,225427,282808,0,0,142,,,,,OPEN,,,,,,,,,1,2,0,,,,0,0,0,18,,,,0,0,86,142,1076,0,,0,46,1304,,,0,0,0,0,,,,,,,,,,,,,,,,,,,,,http,,0,18,911,0,0,0,0,,,0,0,,,,,,,0,,,,, + ft-https,FRONTEND,,,4,23,8000,442138,1437985213,6885481212,0,0,49,,,,,OPEN,,,,,,,,,1,3,0,,,,0,2,0,41,,,,,,,,,,,0,0,0,,,0,0,0,0,,,,,,,,,,,,,,,,,,,,,tcp,,2,41,442138,,0,0,0,,,,,,,,,,,0,,,,, + stats,FRONTEND,,,1,1,8000,15939,3235327,99009201,0,0,0,,,,,OPEN,,,,,,,,,1,4,0,,,,0,1,0,1,,,,0,15938,0,0,0,0,,1,1,15939,,,0,0,0,0,,,,,,,,,,,,,,,,,,,,,http,,1,1,15939,15939,0,0,0,,,0,0,,,,,,,0,,,,, + DATA + end + + it 'reports ok state' do + allow(subject).to receive(:report) + subject.tick + expect(subject).to have_received(:report).exactly(294).times + end + end +end