168 lines
4.3 KiB
Ruby
Executable File
168 lines
4.3 KiB
Ruby
Executable File
require 'optparse'
|
|
|
|
module ActionController
|
|
class RequestProfiler
|
|
# CGI with stubbed environment and standard input.
|
|
class StubCGI < CGI
|
|
attr_accessor :env_table, :stdinput
|
|
|
|
def initialize(env_table, stdinput)
|
|
@env_table = env_table
|
|
super
|
|
@stdinput = stdinput
|
|
end
|
|
end
|
|
|
|
# Stripped-down dispatcher.
|
|
class Sandbox
|
|
attr_accessor :env, :body
|
|
|
|
def self.benchmark(n, env, body)
|
|
Benchmark.realtime { n.times { new(env, body).dispatch } }
|
|
end
|
|
|
|
def initialize(env, body)
|
|
@env, @body = env, body
|
|
end
|
|
|
|
def dispatch
|
|
cgi = StubCGI.new(env, StringIO.new(body))
|
|
|
|
request = CgiRequest.new(cgi)
|
|
response = CgiResponse.new(cgi)
|
|
|
|
controller = Routing::Routes.recognize(request)
|
|
controller.process(request, response)
|
|
end
|
|
end
|
|
|
|
|
|
attr_reader :options
|
|
|
|
def initialize(options = {})
|
|
@options = default_options.merge(options)
|
|
end
|
|
|
|
|
|
def self.run(args = nil, options = {})
|
|
profiler = new(options)
|
|
profiler.parse_options(args) if args
|
|
profiler.run
|
|
end
|
|
|
|
def run
|
|
warmup
|
|
options[:benchmark] ? benchmark : profile
|
|
end
|
|
|
|
def profile
|
|
load_ruby_prof
|
|
|
|
results = RubyProf.profile { benchmark }
|
|
|
|
show_profile_results results
|
|
results
|
|
end
|
|
|
|
def benchmark
|
|
puts '%d req/sec' % (options[:n] / Sandbox.benchmark(options[:n], env, body))
|
|
end
|
|
|
|
def warmup
|
|
puts "#{options[:benchmark] ? 'Benchmarking' : 'Profiling'} #{options[:n]}x"
|
|
puts "\nrequest headers: #{env.to_yaml}"
|
|
|
|
response = Sandbox.new(env, body).dispatch
|
|
|
|
puts "\nresponse body: #{response.body[0...100]}#{'[...]' if response.body.size > 100}"
|
|
puts "\nresponse headers: #{response.headers.to_yaml}"
|
|
puts
|
|
end
|
|
|
|
|
|
def uri
|
|
URI.parse(options[:uri])
|
|
rescue URI::InvalidURIError
|
|
URI.parse(default_uri)
|
|
end
|
|
|
|
def default_uri
|
|
'/'
|
|
end
|
|
|
|
def env
|
|
@env ||= default_env
|
|
end
|
|
|
|
def default_env
|
|
defaults = {
|
|
'HTTP_HOST' => "#{uri.host || 'localhost'}:#{uri.port || 3000}",
|
|
'REQUEST_URI' => uri.path,
|
|
'REQUEST_METHOD' => method,
|
|
'CONTENT_LENGTH' => body.size }
|
|
|
|
if fixture = options[:fixture]
|
|
defaults['CONTENT_TYPE'] = "multipart/form-data; boundary=#{extract_multipart_boundary(fixture)}"
|
|
end
|
|
|
|
defaults
|
|
end
|
|
|
|
def method
|
|
options[:method] || (options[:fixture] ? 'POST' : 'GET')
|
|
end
|
|
|
|
def body
|
|
options[:fixture] ? File.read(options[:fixture]) : ''
|
|
end
|
|
|
|
|
|
def default_options
|
|
{ :n => 1000, :open => 'open %s &' }
|
|
end
|
|
|
|
# Parse command-line options
|
|
def parse_options(args)
|
|
OptionParser.new do |opt|
|
|
opt.banner = "USAGE: #{$0} uri [options]"
|
|
|
|
opt.on('-u', '--uri [URI]', 'Request URI. Defaults to http://localhost:3000/') { |v| options[:uri] = v }
|
|
opt.on('-n', '--times [0000]', 'How many requests to process. Defaults to 1000.') { |v| options[:n] = v.to_i }
|
|
opt.on('-b', '--benchmark', 'Benchmark instead of profiling') { |v| options[:benchmark] = v }
|
|
opt.on('--method [GET]', 'HTTP request method. Defaults to GET.') { |v| options[:method] = v.upcase }
|
|
opt.on('--fixture [FILE]', 'Path to POST fixture file') { |v| options[:fixture] = v }
|
|
opt.on('--open [CMD]', 'Command to open profile results. Defaults to "open %s &"') { |v| options[:open] = v }
|
|
opt.on('-h', '--help', 'Show this help') { puts opt; exit }
|
|
|
|
opt.parse args
|
|
end
|
|
end
|
|
|
|
protected
|
|
def load_ruby_prof
|
|
begin
|
|
require 'ruby-prof'
|
|
#RubyProf.measure_mode = RubyProf::ALLOCATED_OBJECTS
|
|
rescue LoadError
|
|
abort '`gem install ruby-prof` to use the profiler'
|
|
end
|
|
end
|
|
|
|
def extract_multipart_boundary(path)
|
|
File.open(path) { |f| f.readline }
|
|
end
|
|
|
|
def show_profile_results(results)
|
|
File.open "#{RAILS_ROOT}/tmp/profile-graph.html", 'w' do |file|
|
|
RubyProf::GraphHtmlPrinter.new(results).print(file)
|
|
`#{options[:open] % file.path}` if options[:open]
|
|
end
|
|
|
|
File.open "#{RAILS_ROOT}/tmp/profile-flat.txt", 'w' do |file|
|
|
RubyProf::FlatPrinter.new(results).print(file)
|
|
`#{options[:open] % file.path}` if options[:open]
|
|
end
|
|
end
|
|
end
|
|
end
|