Require memory tree

require_memory_tree.rb

$current = {:path => "*ROOT*", :children => [], :mem => 2 ** 30}

if GC.respond_to?(:enable_stats)
  module RequireTracking
    def require(*args)
      node = {:path => args, :children => []}
      $current[:children] << node
      prev, $current = $current, node

      start_allocated_size = GC.allocated_size
      super
    ensure
      node[:mem] = (GC.allocated_size - start_allocated_size)
      $current = prev
    end

    def numeric_thousands_indicators(number)
      number.to_s.gsub(/(\d)(?=\d{3}+(?:\.|$))(\d{3}\..*)?/,'\1,\2')
    end
  end

  Object.send(:include, RequireTracking)
  Kernel.send(:include, RequireTracking)

  $require_stats       = Hash.new { |h,k| h[k] = 0 }
  $require_stats_top ||= 20
  GC.enable_stats
  GC.clear_stats

  at_exit do
    include ActionView::Helpers::NumberHelper
    def print_node(node, indent = 0)
      if node[:mem] > (2**20)
        puts %(#{" " * indent}#{node[:path]}: #{number_to_human_size node[:mem]})
        node[:children].each do |child|
          print_node child, indent + 2
        end
      end
    end
    print_node $current
  end
else
  puts "Warning: Not running with an interpreter with GC statistics"
end