Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
A load time profiler script
# = Description
# A load time profiler script, useful for identifying requires
# that slow down the loading of a script.
#
# Oftentimes we use but don't always need what we require in
# scripts. This can dramatically slow load times, which can be
# annoying. Use this script to identify slow-loading parts of
# a script and then consider using autoloading to defer
# the require until you really need it.
#
# #require 'stringio'
# autoload(:StringIO, 'stringio')
#
# = Usage
#
# Simply cut-n-paste this script before your first require
# statements (or add the requires at the end of this script):
#
# [profile_load_time.rb ]
# # ... this script ...
#
# require 'time'
#
# Then from the command line:
#
# % ruby profile_load_time.rb
# ================================================================================
# Require/Load Profile (time in ms)
# * Load times > 0.5 ms
# - duplicate requires
# ================================================================================
# * 9.1: time
# * 5.2: parsedate
# * 5.0: date/format
# * 1.2: rational
#
# Total time: 9.105 ms
#
# = Info
# Developer:: Simon Chiang (http://bahuvrihi.wordpress.com)
# Licence:: MIT-Style (http://en.wikipedia.org/wiki/MIT_License)
# Gist:: http://gist.github.com/5732
class RecursiveProfile
class << self
def instance
@@instance
end
def [](key)
@@variables[key]
end
def []=(key, value)
@@variables[key] = value
end
end
attr_reader :results
def initialize
@results = []
end
def profile
current = @results
@results = []
start = Time.now
target = [yield]
target << Time.now-start
target << @results
@results = current
@results << target
end
def each
recurse(@results, 0) do |value, time, nesting_depth|
yield(value, time, nesting_depth)
end
end
private
def recurse(parent, nesting_depth, &block)
parent.each do |value, time, children|
yield(value, time, nesting_depth)
recurse(children, nesting_depth + 1, &block)
end
end
@@instance = RecursiveProfile.new
@@variables = {}
end
def require(path)
RecursiveProfile.instance.profile do
begin
super
path
rescue(Exception)
"#{path} (error: #{$!.message})"
end
end
end
def load(path)
RecursiveProfile.instance.profile do
begin
super
path
rescue(Exception)
"#{path} (error: #{$!.message})"
end
end
end
at_exit do
total_time = (Time.now - RecursiveProfile[:start]) * 10**3
values = []
duplicates = []
RecursiveProfile.instance.each do |value, time, nesting_depth|
(values.include?(value) ? duplicates : values) << value
end
puts "=" * 80
puts "Require/Load Profile (time in ms)"
puts "* Load time > #{RecursiveProfile[:cutoff_in_ms]} ms"
puts "- Duplicate require"
puts "=" * 80
RecursiveProfile.instance.each do |value, time, nesting_depth|
time_in_ms = time * 10**3
flags = (time_in_ms > RecursiveProfile[:cutoff_in_ms] ? '*' : ' ')
flags = "-" if duplicates.include?(value) && !values.include?(value)
values.delete(value)
puts "#{flags} #{' ' * nesting_depth}#{"%.1f" % time_in_ms}: #{value}"
end
puts
puts "Total time: #{total_time} ms"
end
RecursiveProfile[:cutoff_in_ms] = 0.5
RecursiveProfile[:start] = Time.now
###############################################################################
# add requires here
###############################################################################
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment