profmem 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in profmem.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Phil Hofmann <phil@branch14.org>
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,54 @@
+                      __
+     _ __  _ __ ___  / _|_ __ ___   ___ _ __ ___
+    | '_ \| '__/ _ \| |_| '_ ` _ \ / _ \ '_ ` _ \
+    | |_) | | | (_) |  _| | | | | |  __/ | | | | |
+    | .__/|_|  \___/|_| |_| |_| |_|\___|_| |_| |_|
+    |_|
+
+# Welcome to profmem
+
+A simple tool to profile memory, built to identify memory leaks.
+
+## Usage
+
+Let's say you want to profile the memory consumption of your Rspec
+suite...
+
+First in your Gemfile's test group add profmem
+
+    group :test do
+      gem 'profmem', require: false
+      ...
+
+Secondly in your `spec_helper` file require profmem and start it right
+at the top. (It is safe to do this unconditionally as it will only
+trigger if the env var PROFMEM is set.)
+
+    require 'profmem'
+    Profmem.start
+
+In an after-suite-callback of Rspec tell memprof to summarize it's
+findings
+
+    config.after(:suite) do
+      Profmem.summarize
+    end
+
+Run you Rspec suite with env var PROFMEM set to the destination of the
+profiling data
+
+    $ PROFMEM=pm.dump rspec
+
+Finally run profmem to do the number crunching (this might take a
+while) and study the results
+
+    $ profmem pm.dump pm.txt
+    $ less pm.txt
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/lib/profmem.rb

@@ -0,0 +1,88 @@
+#!/usr/bin/env ruby
+
+require 'json'
+
+module Profmem
+
+  extend self
+
+  def setup
+    if ENV['PROFMEM']
+      require 'objspace'
+      ObjectSpace.trace_object_allocations_start
+    end
+  end
+
+  def finalize
+    if path = ENV['PROFMEM']
+      io = File.open(path, "w")
+      ObjectSpace.dump_all(output: io)
+      io.close
+      puts "est. total memory consumed: #{ObjectSpace.memsize_of_all}"
+      puts "memory profiling data written to #{path} (#{File.size(path)})"
+    end
+  end
+
+  def run(args)
+
+    puts
+    puts 'Reading input file...'
+    total = `wc -l "#{args[0]}"`.strip.split(' ')[0].to_i
+    puts "Data points:\t#{total}"
+    width = total.to_s.length
+
+    data = []
+    counter = 0
+    File.open(args[0]) do |f|
+      f.each_line do |line|
+        data << (parsed=JSON.parse(line))
+        counter += 1
+        print "Progress:\t% #{width}s\r" % counter
+      end
+    end
+
+    puts
+    puts
+    puts 'Grouping by generation...'
+
+    generations = data.group_by { |row| row["generation"] }
+
+    puts 'Ranking...'
+    ranked = {}
+    generations.each do |gen, allocs|
+      gen ||= 1 # sometimes gen is nil
+      ranked[gen] = {
+        rank: gen * allocs.size,
+        allocs: allocs
+      }
+    end
+    generations = nil # free some memory
+
+    puts 'Sorting...'
+    ranked = ranked.sort_by { |k, v| -v[:rank] }
+
+    puts 'Writing output file...'
+    puts "Generations:\t#{ranked.size}"
+    width = ranked.size.to_s.length
+
+    File.open(args[1], 'w') do |f|
+      counter = 0
+      ranked.each do |k,v|
+        counter += 1
+        print "Progress:\t% #{width}s\r" % counter
+        f.puts "generation #{k} objects #{v[:allocs].count} rank #{v[:rank]}"
+        locs = v[:allocs].group_by { |a| [a['file'], a['line']] * ':' }
+        locs = locs.sort_by { |loc, allocs| -allocs.size }
+        locs.each do |loc, allocs|
+          f.puts "  #{allocs.size} #{loc}"
+        end
+      end
+    end
+
+    puts
+    puts
+
+  end
+end
+
+Profmem.run(ARGV) if $0 == __FILE__

data/lib/profmem/version.rb

@@ -0,0 +1,3 @@
+module Profmem
+  VERSION = "0.0.1"
+end

data/profmem.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'profmem/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "profmem"
+  spec.version       = Profmem::VERSION
+  spec.authors       = ["phil"]
+  spec.email         = ["phil@branch14.org"]
+  spec.description   = %q{A simple tool to profile memory, built to identify memory leaks.}
+  spec.summary       = %q{A simple tool to profile memory, built to identify memory leaks.}
+  spec.homepage      = "http://github.com/branch14/profmem"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+end

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification
+name: profmem
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- phil
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-09-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A simple tool to profile memory, built to identify memory leaks.
+email:
+- phil@branch14.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/profmem.rb
+- lib/profmem/version.rb
+- profmem.gemspec
+homepage: http://github.com/branch14/profmem
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1021228832940894519
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1021228832940894519
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: A simple tool to profile memory, built to identify memory leaks.
+test_files: []