Loop wrk

loop_wrk.rb

# Usage:
# ruby ./bench.rb "wrk -c20 -t20 -d3m http://localhost:9292/request" "c20t20-jvm.dat"

PARSE_RPS = proc do |out|
  begin
    out.match(/Requests\/sec: (?<value>.*)/)[:value].gsub!(" ", "").to_f
  rescue
    puts "Wrong RPS parsing"
    0
  end
end

puts "Let's start!"

command = ARGV.shift
plot = ARGV.shift

unless command.include?("wrk")
  raise "Please, provide a wrk command"
end

i = 1
result = []

begin
  loop do
    out = `#{command}`
    rps = PARSE_RPS.call(out)
    puts "#{i}: #{rps} RPS"
    result << [i, rps]
    i += 1

    break if rps == 0

    sleep 10
  end
rescue SystemExit, Interrupt
  File.open(plot, "w") do |file|
    file.puts("# Iteration RPS")
    result.each do |(iteration, rps)|
      file.puts("#{iteration} #{rps}")
    end
  end
end

puts "Done. See #{plot} file"

# An Example:
# gnuplot> set boxwidth 0.6 relative
# gnuplot> set style fill solid
# gnuplot> plot 'out.dat' with boxes