源码：GitHub - SamSaffron/memory_profiler: memory_profiler for ruby

入口可以是CLI监控命令：ruby-memory-profiler --pretty run -- rubocop --cache false

  • 使用[[optparse.rb]]解析选项，转为env
  • 然后RUBYOPT引入memory_profiler/autorun
  • Kernel.exec
或者CLI监控文件：ruby-memory-profiler --max=10 main.rb

begin
  MemoryProfiler.start(options)
  load(script)
ensure
  report = MemoryProfiler.stop
  report.pretty_print(**options)
end

又或者是侵入式

require 'memory_profiler'
report = MemoryProfiler.report do
  # run your code here
end

report.pretty_print

最终都是创建一个MemoryProfiler::Reporter然后start和stop：

  • 开始MemoryProfiler::Reporter#start
    • 先GC.start清理干净，然后GC.disable停用GC，再GC.count记录当前GC代数
    • 接着开启对象分配跟踪ObjectSpace.trace_object_allocations_start
  • 执行待跟踪的代码yield
  • 停止MemoryProfiler::Reporter#stop
    • 停止对象分配跟踪ObjectSpace.trace_object_allocations_stop
    • 遍历ObjectSpace.each_object，针对每个当前代数（ObjectSpace.allocation_generation(obj)）的对象，生成MemoryProfiler::Stat.new(class_name, gem, ObjectSpace.allocation_sourcefile(obj), ObjectSpace.allocation_sourceline(obj), ObjectSpace.memsize_of(obj), string)，记作“过程中分配的对象”
    • 开启GCGC.enable，尝试GC.start清理能清的
    • 遍历ObjectSpace.each_object，针对每个之前代数的对象，记作“过程中分配且没法清理的对象”
    • 清理跟踪信息ObjectSpace.trace_object_allocations_clear