discourse/script/bench.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

340 lines
8.5 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
2013-08-15 01:19:23 -04:00
require "socket"
require "csv"
2013-08-15 02:35:57 -04:00
require "yaml"
2013-12-10 18:32:23 -05:00
require "optparse"
require "fileutils"
2013-12-10 18:32:23 -05:00
@include_env = false
@result_file = nil
@iterations = 500
2014-01-08 23:56:03 -05:00
@best_of = 1
@mem_stats = false
@unicorn = false
@dump_heap = false
@concurrency = 1
@skip_asset_bundle = false
@unicorn_workers = 3
2013-12-10 18:32:23 -05:00
opts = OptionParser.new do |o|
o.banner = "Usage: ruby bench.rb [options]"
o.on("-n", "--with_default_env", "Include recommended Discourse env") do
@include_env = true
end
o.on("-o", "--output [FILE]", "Output results to this file") do |f|
@result_file = f
end
o.on("-i", "--iterations [ITERATIONS]", "Number of iterations to run the bench for") do |i|
@iterations = i.to_i
end
2014-01-08 23:56:03 -05:00
o.on("-b", "--best_of [NUM]", "Number of times to run the bench taking best as result") do |i|
@best_of = i.to_i
end
o.on("-d", "--heap_dump") do
@dump_heap = true
# We need an env var for config/boot.rb to enable allocation tracing prior to framework init
ENV['DISCOURSE_DUMP_HEAP'] = "1"
end
o.on("-m", "--memory_stats") do
@mem_stats = true
end
o.on("-u", "--unicorn", "Use unicorn to serve pages as opposed to puma") do
@unicorn = true
end
o.on("-c", "--concurrency [NUM]", "Run benchmark with this number of concurrent requests (default: 1)") do |i|
@concurrency = i.to_i
end
o.on("-w", "--unicorn_workers [NUM]", "Run benchmark with this number of unicorn workers (default: 3)") do |i|
@unicorn_workers = i.to_i
end
o.on("-s", "--skip-bundle-assets", "Skip bundling assets") do
@skip_asset_bundle = true
end
2013-12-10 18:32:23 -05:00
end
opts.parse!
2013-08-15 02:35:57 -04:00
2014-01-08 23:56:03 -05:00
def run(command, opt = nil)
exit_status =
if opt == :quiet
system(command, out: "/dev/null", err: :out)
else
system(command, out: $stdout, err: :out)
end
2017-04-14 12:58:35 -04:00
abort("Command '#{command}' failed with exit status #{$?}") unless exit_status
2013-08-15 01:32:07 -04:00
end
begin
require 'facter'
raise LoadError if Gem::Version.new(Facter.version) < Gem::Version.new("4.0")
rescue LoadError
run "gem install facter"
2014-01-02 21:03:58 -05:00
puts "please rerun script"
exit
end
@timings = {}
2013-08-15 02:35:57 -04:00
def measure(name)
start = Time.now
yield
@timings[name] = ((Time.now - start) * 1000).to_i
end
2013-08-15 01:37:33 -04:00
def prereqs
puts "Be sure to following packages are installed:
2014-05-29 00:10:34 -04:00
sudo apt-get -y install build-essential libssl-dev libyaml-dev git libtool libxslt-dev libxml2-dev libpq-dev gawk curl pngcrush python-software-properties software-properties-common tasksel
2013-08-15 01:37:33 -04:00
sudo tasksel install postgresql-server
OR
apt-get install postgresql-server^
2013-08-15 01:37:33 -04:00
sudo apt-add-repository -y ppa:rwky/redis
sudo apt-get update
sudo apt-get install redis-server
"
end
2013-08-15 01:19:23 -04:00
puts "Running bundle"
if run("bundle", :quiet)
2013-08-15 01:32:07 -04:00
puts "Quitting, some of the gems did not install"
2013-08-15 01:37:33 -04:00
prereqs
2013-08-15 01:32:07 -04:00
exit
end
2013-08-15 01:19:23 -04:00
puts "Ensuring config is setup"
%x{which ab > /dev/null 2>&1}
unless $? == 0
2013-08-15 01:19:23 -04:00
abort "Apache Bench is not installed. Try: apt-get install apache2-utils or brew install ab"
end
unless File.exists?("config/database.yml")
puts "Copying database.yml.development.sample to database.yml"
`cp config/database.yml.development-sample config/database.yml`
end
ENV["RAILS_ENV"] = "profile"
2013-10-12 17:06:45 -04:00
discourse_env_vars = %w(
DISCOURSE_DUMP_HEAP
RUBY_GC_HEAP_INIT_SLOTS
RUBY_GC_HEAP_FREE_SLOTS
RUBY_GC_HEAP_GROWTH_FACTOR
RUBY_GC_HEAP_GROWTH_MAX_SLOTS
RUBY_GC_MALLOC_LIMIT
RUBY_GC_OLDMALLOC_LIMIT
RUBY_GC_MALLOC_LIMIT_MAX
RUBY_GC_OLDMALLOC_LIMIT_MAX
RUBY_GC_MALLOC_LIMIT_GROWTH_FACTOR
RUBY_GC_OLDMALLOC_LIMIT_GROWTH_FACTOR
RUBY_GC_HEAP_OLDOBJECT_LIMIT_FACTOR
RUBY_GLOBAL_METHOD_CACHE_SIZE
2018-05-03 01:50:45 -04:00
LD_PRELOAD
)
2013-12-10 18:32:23 -05:00
if @include_env
puts "Running with tuned environment"
discourse_env_vars.each do |v|
ENV.delete v
end
ENV['RUBY_GLOBAL_METHOD_CACHE_SIZE'] = '131072'
ENV['RUBY_GC_HEAP_GROWTH_MAX_SLOTS'] = '40000'
ENV['RUBY_GC_HEAP_INIT_SLOTS'] = '400000'
ENV['RUBY_GC_HEAP_OLDOBJECT_LIMIT_FACTOR'] = '1.5'
2013-12-10 18:32:23 -05:00
else
# clean env
puts "Running with the following custom environment"
end
discourse_env_vars.each do |w|
puts "#{w}: #{ENV[w]}" if ENV[w].to_s.length > 0
2013-10-12 17:06:45 -04:00
end
2013-08-15 02:35:57 -04:00
2013-08-15 01:19:23 -04:00
def port_available?(port)
server = TCPServer.open("0.0.0.0", port)
2013-08-15 01:19:23 -04:00
server.close
true
rescue Errno::EADDRINUSE
false
end
2013-08-15 03:13:05 -04:00
@port = 60079
2013-08-15 01:19:23 -04:00
2013-08-15 03:13:05 -04:00
while !port_available? @port
@port += 1
2013-08-15 01:19:23 -04:00
end
puts "Ensuring profiling DB exists and is migrated"
puts `bundle exec rake db:create`
`bundle exec rake db:migrate`
2013-08-15 02:59:38 -04:00
puts "Timing loading Rails"
2013-08-15 02:35:57 -04:00
measure("load_rails") do
`bundle exec rake middleware`
end
2013-08-15 01:19:23 -04:00
2013-08-15 02:59:38 -04:00
puts "Populating Profile DB"
run("bundle exec ruby script/profile_db_generator.rb")
2013-08-15 01:19:23 -04:00
2013-09-10 02:03:11 -04:00
puts "Getting api key"
api_key = `bundle exec rake api_key:create_master[bench]`.split("\n")[-1]
2013-09-10 02:03:11 -04:00
2017-04-14 12:58:35 -04:00
def bench(path, name)
2013-08-15 01:19:23 -04:00
puts "Running apache bench warmup"
add = ""
add = "-c #{@concurrency} " if @concurrency > 1
`ab #{add} -n 20 -l "http://127.0.0.1:#{@port}#{path}"`
2017-04-14 12:58:35 -04:00
puts "Benchmarking #{name} @ #{path}"
`ab #{add} -n #{@iterations} -l -e tmp/ab.csv "http://127.0.0.1:#{@port}#{path}"`
2013-08-15 01:19:23 -04:00
percentiles = Hash[*[50, 75, 90, 99].zip([]).flatten]
CSV.foreach("tmp/ab.csv") do |percent, time|
percentiles[percent.to_i] = time.to_i if percentiles.key? percent.to_i
end
2013-08-15 03:13:05 -04:00
percentiles
end
begin
2013-09-10 02:03:11 -04:00
# critical cause cache may be incompatible
unless @skip_asset_bundle
puts "precompiling assets"
run("bundle exec rake assets:precompile")
end
2013-09-10 02:03:11 -04:00
pid =
if @unicorn
ENV['UNICORN_PORT'] = @port.to_s
ENV['UNICORN_WORKERS'] = @unicorn_workers.to_s
FileUtils.mkdir_p(File.join('tmp', 'pids'))
spawn("bundle exec unicorn -c config/unicorn.conf.rb")
else
spawn("bundle exec puma -p #{@port} -e production")
end
2013-08-15 03:13:05 -04:00
while port_available? @port
sleep 1
end
puts "Starting benchmark..."
headers = { 'Api-Key' => api_key,
'Api-Username' => "admin1" }
# asset precompilation is a dog, wget to force it
run "curl -s -o /dev/null http://127.0.0.1:#{@port}/"
2013-08-15 03:13:05 -04:00
redirect_response = `curl -s -I "http://127.0.0.1:#{@port}/t/i-am-a-topic-used-for-perf-tests"`
if redirect_response !~ /301 Moved Permanently/
raise "Unable to locate topic for perf tests"
end
topic_url = redirect_response.match(/^location: .+(\/t\/i-am-a-topic-used-for-perf-tests\/.+)$/i)[1].strip
2014-01-08 23:56:03 -05:00
tests = [
["categories", "/categories"],
2014-01-08 23:56:03 -05:00
["home", "/"],
["topic", topic_url]
2017-03-28 14:27:54 -04:00
# ["user", "/u/admin1/activity"],
2014-01-08 23:56:03 -05:00
]
tests.concat(tests.map { |k, url| ["#{k}_admin", "#{url}", headers] })
tests.each do |_, path, headers_for_path|
header_string = headers_for_path&.map { |k, v| "-H \"#{k}: #{v}\"" }&.join(" ")
if `curl -s -I "http://127.0.0.1:#{@port}#{path}" #{header_string}` !~ /200 OK/
raise "#{path} returned non 200 response code"
end
end
# NOTE: we run the most expensive page first in the bench
2014-01-08 23:56:03 -05:00
def best_of(a, b)
return a unless b
return b unless a
a[50] < b[50] ? a : b
end
results = {}
@best_of.times do
tests.each do |name, url|
2017-04-14 12:58:35 -04:00
results[name] = best_of(bench(url, name), results[name])
2014-01-08 23:56:03 -05:00
end
end
2013-08-15 03:48:11 -04:00
puts "Your Results: (note for timings- percentile is first, duration is second in millisecs)"
2013-08-15 01:19:23 -04:00
if @unicorn
puts "Unicorn: (workers: #{@unicorn_workers})"
else
# TODO we want to also bench puma clusters
puts "Puma: (single threaded)"
end
puts "Include env: #{@include_env}"
puts "Iterations: #{@iterations}, Best of: #{@best_of}"
puts "Concurrency: #{@concurrency}"
puts
# Prevent using external facts because it breaks when running in the
# discourse/discourse_bench docker container.
Facter.reset
2013-08-29 07:34:32 -04:00
facts = Facter.to_hash
facts.delete_if { |k, v|
!["operatingsystem", "architecture", "kernelversion",
"memorysize", "physicalprocessorcount", "processor0",
"virtual"].include?(k)
}
2013-09-10 02:03:11 -04:00
run("RAILS_ENV=profile bundle exec rake assets:clean")
def get_mem(pid)
YAML.load `ruby script/memstats.rb #{pid} --yaml`
end
mem = get_mem(pid)
2014-01-10 00:11:10 -05:00
results = results.merge("timings" => @timings,
"ruby-version" => "#{RUBY_VERSION}-p#{RUBY_PATCHLEVEL}",
"rss_kb" => mem["rss_kb"],
"pss_kb" => mem["pss_kb"]).merge(facts)
2013-12-10 18:32:23 -05:00
if @unicorn
child_pids = `ps --ppid #{pid} | awk '{ print $1; }' | grep -v PID`.split("\n")
child_pids.each do |child|
mem = get_mem(child)
results["rss_kb_#{child}"] = mem["rss_kb"]
results["pss_kb_#{child}"] = mem["pss_kb"]
end
end
2014-01-08 23:56:03 -05:00
puts results.to_yaml
2013-12-10 18:32:23 -05:00
if @mem_stats
puts
puts open("http://127.0.0.1:#{@port}/admin/memory_stats", headers).read
end
if @dump_heap
puts
puts open("http://127.0.0.1:#{@port}/admin/dump_heap", headers).read
end
2013-12-10 18:32:23 -05:00
if @result_file
File.open(@result_file, "wb") do |f|
f.write(results)
end
end
2013-08-15 01:19:23 -04:00
ensure
Process.kill "KILL", pid
end