batman clients counter for sending data to elasticsearch / logstash

batmanclients.rb

#!/usr/bin/env ruby
require 'getoptlong'

# The name of the collectd plugin, something like apache, memory, mysql, interface, ...
PLUGIN_NAME = 'batmanclients'

def usage
  puts("#{$0} -h <host_id> [-i <sampling_interval>]")
  exit
end

# Main
begin
  # Sync stdout so that it will flush to collectd properly.
  $stdout.sync = true

  # Parse command line options
  hostname = nil
  sampling_interval = 60 # in seconds, Default value
  opts = GetoptLong.new(
    ['--hostid', '-h', GetoptLong::REQUIRED_ARGUMENT],
    ['--sampling-interval', '-i', GetoptLong::OPTIONAL_ARGUMENT]
  )
  opts.each do |opt, arg|
    case opt
      when '--hostid'
        hostname = arg
      when '--sampling-interval'
        sampling_interval = arg.to_i
    end
  end
  usage if !hostname

  # Collection loop
  while true do
    start_run = Time.now.to_i
    next_run = start_run + sampling_interval

    # collectd data and print the values
    data = `/usr/bin/alfred-json -r 159 -z|/bin/grep clients -A 3|/bin/grep 'total'|/usr/bin/cut -d ' ' -f 8|sed 's/,$//'|/usr/bin/awk '{s+=$1} END {print s}'`
    puts("PUTVAL #{hostname}/#{PLUGIN_NAME}/gauge-clients_total #{start_run}:#{data}")

    # sleep to make the interval
    while((time_left = (next_run - Time.now.to_i)) > 0) do
      sleep(time_left)
    end
  end
end