aboutsummaryrefslogtreecommitdiffstats
path: root/tools/profile_requires
diff options
context:
space:
mode:
authorJeremy Kemper <jeremy@bitsweat.net>2009-05-12 17:55:45 -0700
committerJeremy Kemper <jeremy@bitsweat.net>2009-05-13 11:43:03 -0700
commite9c5750bd1bb15b2d551ee014f06bd5019f2357d (patch)
tree10360fb32dd0f312265a355b7dcfc5c9e0c79126 /tools/profile_requires
parent4c963972fd0fb696e857540351f7558f027ec962 (diff)
downloadrails-e9c5750bd1bb15b2d551ee014f06bd5019f2357d.tar.gz
rails-e9c5750bd1bb15b2d551ee014f06bd5019f2357d.tar.bz2
rails-e9c5750bd1bb15b2d551ee014f06bd5019f2357d.zip
Preserve ordering
Diffstat (limited to 'tools/profile_requires')
-rwxr-xr-xtools/profile_requires81
1 files changed, 81 insertions, 0 deletions
diff --git a/tools/profile_requires b/tools/profile_requires
new file mode 100755
index 0000000000..be7770b3e5
--- /dev/null
+++ b/tools/profile_requires
@@ -0,0 +1,81 @@
+#!/usr/bin/env ruby
+# Example:
+# ruby -Iactivesupport/lib tools/profile_requires.rb active_support
+# ruby -Iactionpack/lib tools/profile_requires.rb action_controller
+abort 'Use REE so you can profile memory and object allocation' unless GC.respond_to?(:enable_stats)
+
+GC.enable_stats
+require 'rubygems'
+Gem.source_index
+require 'benchmark'
+
+module TrackHeapGrowth
+ class << self
+ attr_accessor :indent
+ attr_accessor :stats
+ end
+ self.indent = 0
+ self.stats = []
+
+ def track_growth(file)
+ TrackHeapGrowth.stats << [file, TrackHeapGrowth.indent]
+ TrackHeapGrowth.indent += 1
+ heap_before, objects_before = GC.allocated_size, ObjectSpace.allocated_objects
+ result = nil
+ elapsed = Benchmark.realtime { result = yield }
+ heap_after, objects_after = GC.allocated_size, ObjectSpace.allocated_objects
+ TrackHeapGrowth.indent -= 1
+ TrackHeapGrowth.stats.pop if TrackHeapGrowth.stats.last.first == file
+ TrackHeapGrowth.stats << [file, TrackHeapGrowth.indent, elapsed, heap_after - heap_before, objects_after - objects_before] if result
+ result
+ end
+
+ def require(file, *args)
+ track_growth(file) { super }
+ end
+
+ def load(file, *args)
+ track_growth(file) { super }
+ end
+end
+
+Object.instance_eval { include TrackHeapGrowth }
+
+GC.start
+before = GC.allocated_size
+before_rss = `ps -o rss= -p #{Process.pid}`.to_i
+before_live_objects = ObjectSpace.live_objects
+
+path = ARGV.shift
+
+if mode = ARGV.shift
+ require 'ruby-prof'
+ RubyProf.measure_mode = RubyProf.const_get(mode.upcase)
+ RubyProf.start
+end
+
+ENV['NO_RELOAD'] ||= '1'
+ENV['RAILS_ENV'] ||= 'development'
+elapsed = Benchmark.realtime { require path }
+results = RubyProf.stop if mode
+
+GC.start
+after_live_objects = ObjectSpace.live_objects
+after_rss = `ps -o rss= -p #{Process.pid}`.to_i
+after = GC.allocated_size
+usage = (after - before) / 1024.0
+
+if mode
+ File.open("profile_startup.#{mode}.tree", 'w') do |out|
+ RubyProf::CallTreePrinter.new(results).print(out)
+ end
+end
+
+TrackHeapGrowth.stats.each do |file, indent, sec, bytes, objects|
+ if sec
+ puts "%10.2f KB %10d obj %8.1f ms %s%s" % [bytes / 1024.0, objects, sec * 1000, ' ' * indent, file]
+ else
+ puts "#{' ' * (42 + indent)}#{file}"
+ end
+end
+puts "%10.2f KB %10d obj %8.1f ms %d KB RSS" % [usage, after_live_objects - before_live_objects, elapsed * 1000, after_rss - before_rss]