mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
55f9b8129a
Style/SpaceBeforeBlockBraces Style/SpaceInsideBlockBraces Style/SpaceInsideHashLiteralBraces Fix all violations in the repository.
136 lines
4.4 KiB
Bash
Executable file
136 lines
4.4 KiB
Bash
Executable file
#!/usr/bin/env ruby
|
|
# Profile require calls giving information about the time and the files that are called
|
|
# when loading the provided file.
|
|
#
|
|
# Example:
|
|
# tools/profile activesupport/lib/active_support.rb [ruby-prof mode] [ruby-prof printer]
|
|
ENV["NO_RELOAD"] ||= "1"
|
|
ENV["RAILS_ENV"] ||= "development"
|
|
|
|
module CodeTools
|
|
class Profiler
|
|
Error = Class.new(StandardError)
|
|
|
|
attr_reader :path, :mode
|
|
def initialize(path, mode=nil)
|
|
assert_ruby_file_exists(path)
|
|
@path, @mode = path, mode
|
|
require "benchmark"
|
|
end
|
|
|
|
def profile_requires
|
|
GC.start
|
|
before_rss = `ps -o rss= -p #{Process.pid}`.to_i
|
|
|
|
if mode
|
|
require "ruby-prof"
|
|
RubyProf.measure_mode = RubyProf.const_get(mode.upcase)
|
|
RubyProf.start
|
|
else
|
|
Object.instance_eval { include RequireProfiler }
|
|
end
|
|
|
|
elapsed = Benchmark.realtime { require path }
|
|
results = RubyProf.stop if mode
|
|
|
|
GC.start
|
|
after_rss = `ps -o rss= -p #{Process.pid}`.to_i
|
|
|
|
if mode
|
|
if printer = ARGV.shift
|
|
puts "RubyProf outputting to stderr with printer #{printer}"
|
|
RubyProf.const_get("#{printer.to_s.classify}Printer").new(results).print($stdout)
|
|
elsif RubyProf.const_defined?(:CallStackPrinter)
|
|
filename = "#{File.basename(path, '.rb')}.#{mode}.html"
|
|
puts "RubyProf outputting to #{filename}"
|
|
File.open(filename, "w") do |out|
|
|
RubyProf::CallStackPrinter.new(results).print(out)
|
|
end
|
|
else
|
|
filename = "#{File.basename(path, '.rb')}.#{mode}.callgrind"
|
|
puts "RubyProf outputting to #{filename}"
|
|
File.open(filename, "w") do |out|
|
|
RubyProf::CallTreePrinter.new(results).print(out)
|
|
end
|
|
end
|
|
end
|
|
|
|
RequireProfiler.stats.each do |file, depth, sec|
|
|
if sec
|
|
puts "%8.1f ms %s%s" % [sec * 1000, " " * depth, file]
|
|
else
|
|
puts "#{' ' * (13 + depth)}#{file}"
|
|
end
|
|
end
|
|
puts "%8.1f ms %d KB RSS" % [elapsed * 1000, after_rss - before_rss]
|
|
end
|
|
|
|
private
|
|
|
|
def assert_ruby_file_exists(path)
|
|
fail Error.new("No such file") unless File.exist?(path)
|
|
fail Error.new("#{path} is a directory") if File.directory?(path)
|
|
ruby_extension = File.extname(path) == ".rb"
|
|
ruby_executable = File.open(path, "rb") { |f| f.readline } =~ [/\A#!.*ruby/]
|
|
fail Error.new("Not a ruby file") unless ruby_extension or ruby_executable
|
|
end
|
|
|
|
module RequireProfiler
|
|
private
|
|
def require(file, *args) RequireProfiler.profile(file) { super } end
|
|
def load(file, *args) RequireProfiler.profile(file) { super } end
|
|
|
|
@depth, @stats = 0, []
|
|
class << self
|
|
attr_accessor :depth
|
|
attr_accessor :stats
|
|
|
|
def profile(file)
|
|
stats << [file, depth]
|
|
self.depth += 1
|
|
result = nil
|
|
elapsed = Benchmark.realtime { result = yield }
|
|
self.depth -= 1
|
|
stats.pop if stats.last.first == file
|
|
stats << [file, depth, elapsed] if result
|
|
result
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
# ruby-prof printer name causes the third arg to be sent :classify
|
|
# which is probably overkill if you already know the name of the ruby-prof
|
|
# printer you want to use, e.g. Graph
|
|
begin
|
|
require "active_support/inflector"
|
|
require "active_support/core_ext/string/inflections"
|
|
rescue LoadError
|
|
STDERR.puts $!.message
|
|
class String
|
|
# File activesupport/lib/active_support/inflector/methods.rb, line 150
|
|
def classify
|
|
# strip out any leading schema name
|
|
camelize(sub(/.*\./, ""))
|
|
end
|
|
# File activesupport/lib/active_support/inflector/methods.rb, line 68
|
|
def camelize(uppercase_first_letter = true)
|
|
string = self
|
|
if uppercase_first_letter
|
|
string = string.sub(/^[a-z\d]*/) { |match| match.capitalize }
|
|
else
|
|
string = string.sub(/^(?:(?=\b|[A-Z_])|\w)/) { |match| match.downcase }
|
|
end
|
|
string.gsub(/(?:_|(\/))([a-z\d]*)/) { "#{$1}#{$2.capitalize}" }.gsub("/", "::")
|
|
end
|
|
end
|
|
end
|
|
if $0 == __FILE__
|
|
if (filename = ARGV.shift)
|
|
path = File.expand_path(filename)
|
|
mode = ARGV.shift
|
|
CodeTools::Profiler.new(path, mode).profile_requires
|
|
else
|
|
STDERR.puts "No file path entered. Usage is tools/profile path/to/file.rb [ruby-prof mode] [ruby-prof printer]"
|
|
end
|
|
end
|