aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/commands/performance/profiler.rb
blob: f599ad6635ec99bde32b115595368a149eedf32d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/usr/local/bin/ruby
if ARGV.empty?
  $stderr.puts "Usage: ./script/perform profiler 'Person.expensive_method(10)' [times]"
  exit(1)
end

# Keep the expensive require out of the profile.
$stderr.puts 'Loading Rails...'
require RAILS_ROOT + '/config/environment'

# Define a method to profile.
if ARGV[1] and ARGV[1].to_i > 1
  eval "def profile_me() #{ARGV[1]}.times { #{ARGV[0]} } end"
else
  eval "def profile_me() #{ARGV[0]} end"
end

# Use the ruby-prof extension if available.  Fall back to stdlib profiler.
begin
  require 'prof'
  $stderr.puts 'Using the ruby-prof extension.'
  Prof.clock_mode = Prof::GETTIMEOFDAY
  Prof.start
  profile_me
  results = Prof.stop
  require 'rubyprof_ext'
  Prof.print_profile(results, $stderr)
rescue LoadError
  require 'profiler'
  $stderr.puts 'Using the standard Ruby profiler.'
  Profiler__.start_profile
  profile_me
  Profiler__.stop_profile
  Profiler__.print_profile($stderr)
end