Memory Leak Analyzer

memory_leak_analyzer.rb

#########################################################################################################
# Dependencies: rbtrace (https://github.com/tmm1/rbtrace) in the ruby process that will be analyzed     #
# Reference: https://samsaffron.com/archive/2015/03/31/debugging-memory-leaks-in-ruby                   #
#########################################################################################################

require 'json'
class MemoryLeakAnalyzer
  attr_reader :filename

  def initialize
    @filename = "/tmp/leak-analyzer-dump"
  end

  def data_parsed
    # TODO: Read with chunk
    File.new(@filename).readlines.map { |line| JSON.parse(line) }
  end

  def analyze
    data_parsed
      .group_by{ |row| row["generation"] }
      .sort{ |a,b| a[0].to_i <=> b[0].to_i }
      .each do |k,v|
        puts "generation #{k} objects #{v.count}"
      end

    nil
  end

  def analyze_generation(generation)
    data_parsed
       .filter { |line| line["generation"] == generation }
      .group_by{ |row| "#{row["file"]}:#{row["line"]}" }
      .sort{ |a,b| b[1].count <=> a[1].count }
      .each do |k,v|
        puts "#{k} * #{v.count}"
      end

    nil
  end

  def dump(pid)
    `bundle exec rbtrace -p #{pid} -e 'Thread.new{GC.start; i=File.open("#{filename}", "w"); ObjectSpace.dump_all(output: i); i.close}'`
  end
end