Simple class to wrap a profile run around some code

development_profiler.rb

class DevelopmentProfiler
  
  def self.prof(file_name)
    
    RubyProf.start
    yield
    results = RubyProf.stop
    
    # Print a flat profile to text
    File.open "#{Rails.root}/tmp/performance/#{file_name}-graph.html", 'w' do |file|
      RubyProf::GraphHtmlPrinter.new(results).print(file)
    end
 
    File.open "#{Rails.root}/tmp/performance/#{file_name}-flat.txt", 'w' do |file|
      # RubyProf::FlatPrinter.new(results).print(file)
      RubyProf::FlatPrinterWithLineNumbers.new(results).print(file)
    end

    File.open "#{Rails.root}/tmp/performance/#{file_name}-stack.html", 'w' do |file|
      RubyProf::CallStackPrinter.new(results).print(file)
    end
 
  end
  
end

# This version keeps output files together in the same directory suffixed with the time profiling completed,
# and also handles stopping the profiler if exceptions are raised within the code being profiled.
#
# E.g.:
#   tmp/profilers/my-profiled-code-1519860540/graph.html
#                                             flat.html
#                                             stack.html
#
# You can also pass a `false` argument to toggle when the profiler runs.
#
# E.g.: 
#   RubyProfiler.prof('slow-method', id == 1) do ...

module RubyProfiler

  DIR = File.join(Rails.root, 'tmp', 'profiler' )

  def self.prof(file_name, enabled=true)
    return yield unless enabled

    # create dir automatically
    FileUtils.mkdir_p DIR unless File.exist? DIR

    require 'ruby-prof'

    RubyProf.start

    begin
      yield
    rescue => e
      RubyProf.stop
      raise e
    end

    results = RubyProf.stop

    subdir = File.join(DIR, "#{file_name}-#{Time.now.to_i}")
    FileUtils.mkdir_p subdir unless File.exist? subdir

    # Print a flat profile to text
    File.open(File.join(subdir, "graph.html"), 'w') do |file|
      RubyProf::GraphHtmlPrinter.new(results).print(file)
    end

    File.open(File.join(subdir, "flat.txt"), 'w') do |file|
      # RubyProf::FlatPrinter.new(results).print(file)
      RubyProf::FlatPrinterWithLineNumbers.new(results).print(file)
    end

    File.open(File.join(subdir, "stack.html"), 'w') do |file|
      RubyProf::CallStackPrinter.new(results).print(file)
    end
  end
end