64 lines
1.3 KiB
Ruby
64 lines
1.3 KiB
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
||
|
require 'socket'
|
||
|
|
||
|
INTERVAL = ENV['COLLECTD_INTERVAL'].to_i
|
||
|
HOSTNAME = ENV['COLLECTD_HOSTNAME']
|
||
|
|
||
|
|
||
|
def recv sock
|
||
|
results = []
|
||
|
loop do
|
||
|
line = sock.gets
|
||
|
if line.start_with? "| OK:"
|
||
|
break
|
||
|
else
|
||
|
results << line.chomp
|
||
|
end
|
||
|
end
|
||
|
results
|
||
|
end
|
||
|
|
||
|
sock = TCPSocket.new "localhost", 5582
|
||
|
loop do
|
||
|
stats = {}
|
||
|
|
||
|
sock.puts "c2s:show()"
|
||
|
recv(sock).each do |line|
|
||
|
if line.start_with? "| "
|
||
|
encrypted = line.index " (encrypted)"
|
||
|
ipv6 = line.index " (IPv6)"
|
||
|
k = "c2s:#{encrypted ? 'encrypted' : 'unencrypted'}-#{ipv6 ? 'ipv6' : 'ipv4'}"
|
||
|
stats[k] = 0 unless stats[k]
|
||
|
stats[k] += 1
|
||
|
end
|
||
|
end
|
||
|
|
||
|
sock.puts "s2s:show()"
|
||
|
recv(sock).each do |line|
|
||
|
if line.start_with? "| "
|
||
|
out = if line.index "<-"
|
||
|
false
|
||
|
elsif line.index "->"
|
||
|
true
|
||
|
else
|
||
|
next
|
||
|
end
|
||
|
encrypted = line.index " (encrypted)"
|
||
|
ipv6 = line.index " (IPv6)"
|
||
|
k = "s2s-#{out ? 'out' : 'in'}:#{encrypted ? 'encrypted' : 'unencrypted'}-#{ipv6 ? 'ipv6' : 'ipv4'}"
|
||
|
stats[k] = 0 unless stats[k]
|
||
|
stats[k] += 1
|
||
|
end
|
||
|
end
|
||
|
|
||
|
stats.each do |k,v|
|
||
|
k1, k2 = k.split(":", 2)
|
||
|
puts "PUTVAL \"#{HOSTNAME}/exec-prosody-#{k1}/current_connections-#{k2}\" interval=#{INTERVAL} N:#{v}"
|
||
|
end
|
||
|
|
||
|
ensure
|
||
|
sleep INTERVAL
|
||
|
|
||
|
end
|