# Memory snapshot analyzer which parses the /proc file system on *nix
#
# Example (run in Heroku console):
#
#    ms = MemorySnapshot.new
#    1.upto(10000).map { |i| Array.new(i) }; nil
#    ms.snapshot!; nil
#    ms.diff 10
#    => {"lib/ld-2.11.1.so"=>156,"heap"=>2068,"all"=>2224}
#
class MemorySnapshot
 
  attr_reader :prevIoUs
  attr_reader :current
 
  def initialize
    snapshot!
    @prevIoUs = @current
  end
 
  # Generates a Hash of memory elements mapped to sizes of the elements in Kb
  def snapshot!
    @prevIoUs = @current
    @current = reduce(names_with_sizes)
  end
 
  # Calculates the difference between the prevIoUs and the current snapshot
  # Threshold is a minimum delta in kilobytes required to include an entry
  def diff(threshold = 0)
    self.class.diff_between prevIoUs,current,threshold
  end
 
  # Calculates the difference between two memory snapshots
  # Threshold is a minimum delta in kilobytes required to include an entry
  def self.diff_between(before,after,threshold)
    names = (before.keys + after.keys).uniq
    names.reduce({}) do |memo,name|
      delta = after.fetch(name) { 0 } - before.fetch(name) { 0 }
      memo[name] = delta if delta.abs >= threshold
      memo
    end
  end
 
  private
 
  def reduce(matches)
    total = 0
    current_name = nil
    matches.reduce(Hash.new { 0 }) do |memo,match|
      current_name = match[:name] || current_name
      size = match[:size].to_i
      total += size
      memo[current_name] += size
      memo
    end.tap { |snapshot| snapshot['all'] = total }
  end
 
  def names_with_sizes
    smap_entries.map do |line|
      /((^(\/|\[)(?<name>[^ \]]+)\]?\s+)|(^))(?<size>\d+)\s/.match(line)
    end
  end
 
  def smap_entries
    smaps.
        gsub(/^(([^Sa-f0-9])|(S[^i]))[^\n]+\n/m,'').
        gsub(/\nSize:/m,'').
        gsub(/[0-9a-f]+-[0-9a-f]+.{6}[0-9a-f]+ [0-9a-f]+:[0-9a-f]+ [0-9a-f]+\s+/i,'').
        split("\n")
  end
 
  def smaps
    File.read("/proc/#{Process.pid}/smaps")
  end
end