Peak Memory Profiler

my_profiler.rb

class MyProfiler
  def self.profile
    p = Profiler.new
    p.event "Begin"

    yield p

    p.event "End"
    p
  end

  def initialize
    @events = []
    @max = 0
  end

  def event name=nil
    unless name.nil?
      @events.push([name, get_memory_usage])
    end

    @max = [@max, get_memory_usage].max
  end

  def get_memory_usage
    `ps -o rss= -p #{Process.pid}`.to_i
  end

  def print
    @events.each do |event|
      name, memory_usage = event
      puts "#{name}: #{memory_usage}"
    end

    puts "Max Usage: #{@eventmax}"
  end
end

=begin
  USAGE:
    profiler = MyProfiler.profile do |profiler|
      10000.times do
        profiler.event
        my_expensive_operations
      end
    end

    profiler.print
=end