56 lines
1.3 KiB
Ruby
56 lines
1.3 KiB
Ruby
module Middleman
|
|
module Profiling
|
|
class << self
|
|
# The profiler instance. There can only be one!
|
|
attr_writer :profiler
|
|
def profiler
|
|
@profiler ||= NullProfiler.new
|
|
end
|
|
|
|
# Start the profiler
|
|
def start
|
|
profiler.start
|
|
end
|
|
|
|
# Stop the profiler and generate a report. Make sure to call start first
|
|
def report(report_name)
|
|
profiler.report(report_name)
|
|
end
|
|
end
|
|
|
|
# A profiler that does nothing. The default.
|
|
class NullProfiler
|
|
def start
|
|
end
|
|
|
|
def report(_)
|
|
end
|
|
end
|
|
|
|
# A profiler that uses ruby-prof
|
|
class RubyProfProfiler
|
|
def initialize
|
|
require 'ruby-prof'
|
|
rescue LoadError
|
|
raise "To use the --profile option, you must add the 'ruby-prof' gem to your Gemfile"
|
|
end
|
|
|
|
def start
|
|
RubyProf.start
|
|
end
|
|
|
|
def report(report_name)
|
|
result = RubyProf.stop
|
|
|
|
printer = RubyProf::GraphHtmlPrinter.new(result)
|
|
outfile = File.join('profile', report_name)
|
|
outfile = (outfile + '.html') unless outfile.end_with? '.html'
|
|
FileUtils.mkdir_p(File.dirname(outfile))
|
|
File.open(outfile, 'w') do |f|
|
|
printer.print(f, min_percent: 1)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|