diff options
Diffstat (limited to 'railties/lib/commands/perform/profiler.rb')
-rw-r--r-- | railties/lib/commands/perform/profiler.rb | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/railties/lib/commands/perform/profiler.rb b/railties/lib/commands/perform/profiler.rb new file mode 100644 index 0000000000..d84c8b739f --- /dev/null +++ b/railties/lib/commands/perform/profiler.rb @@ -0,0 +1,35 @@ +#!/usr/local/bin/ruby +if ARGV.empty? + $stderr.puts "Usage: profiler 'Person.expensive_method(10)' [times]" + exit(1) +end + +# Keep the expensive require out of the profile. +$stderr.puts 'Loading Rails...' +require File.dirname(__FILE__) + '/../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 |