memory_profiler 0.9.8 → 0.9.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 87e903eb12fc8a885066fe621d8254a4297555fa
-  data.tar.gz: 179a6ccb31c2b71ae9d1ef620be16ce1fa9b4d70
+  metadata.gz: d7bad0d25de1214584652c298193da15c2306470
+  data.tar.gz: 58b4c27c8e49262703918d27f08e9a2d25b7ba90
 SHA512:
-  metadata.gz: ac3255e9b9623c580c3c7cdba776a530907936f8d4ff1c78b46ce42b3d7f97b446082d190c74d7a7f050ec7036d2c662f6736ce1db104ccd0cf17cc466f0ad67
-  data.tar.gz: c490d5aa5e024119839aaf8948c27d992c6b40325f8ded2447c228d726145c5668d9ee29d3ac13f8f7d0a8d9c0bc064130fcf67209103e493ad8e4ad5acb73ad
+  metadata.gz: 724ecac4311bd0080037b9694d0fcf01d5ccce32886257d5fcdaf7a18fd5a1391d2296878b791b2ab30e7fc6086afc8381878ffa097e041b1eab1f983ce8164a
+  data.tar.gz: 9ddb1527672a92743baa9c29f0cb3ec2db1fda0336b3502197a614decd8d547a3bf2a8c1946b7809b666a98f27da7f344e2b6634d13e9fe08b396d0a8eec329a

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 # Changelog
 
+## 0.9.9
+- Add options for pretty printer to customize report
+
 ## 0.9.8
 - Add optional start/stop sematics to memory profiler api @nicklamuro @dgynn
 

data/README.md

@@ -1,9 +1,10 @@
+[![Build Status](https://travis-ci.org/SamSaffron/memory_profiler.svg?branch=master)](https://travis-ci.org/SamSaffron/memory_profiler)
+[![Gem Version](https://badge.fury.io/rb/memory_profiler.svg)](https://rubygems.org/gems/memory_profiler)
+
 # MemoryProfiler
 
 A memory profiler for Ruby 
 
-[![Build Status](https://travis-ci.org/SamSaffron/memory_profiler.png?branch=master)](https://travis-ci.org/SamSaffron/memory_profiler)
-
 ## Requirements 
 
 Ruby(MRI) Version 2.1.0 and above. 

data/lib/memory_profiler/helpers.rb

@@ -5,6 +5,7 @@ module MemoryProfiler
       @gem_guess_cache = Hash.new
       @location_cache = Hash.new { |h,k| h[k] = Hash.new.compare_by_identity }
       @class_name_cache = Hash.new.compare_by_identity
+      @string_cache = Hash.new
     end
 
     def guess_gem(path)
@@ -28,5 +29,8 @@ module MemoryProfiler
       @class_name_cache[klass] ||= ((klass.is_a?(Class) && klass.name) || '<<Unknown>>').to_s
     end
 
+    def lookup_string(obj)
+      @string_cache[obj] ||= String.new << obj
+    end
   end
 end

data/lib/memory_profiler/reporter.rb

@@ -64,8 +64,15 @@ module MemoryProfiler
     # Collects object allocation and memory of ruby code inside of passed block.
     def run(&block)
       start
-      block.call
-      stop
+      begin
+        block.call
+      rescue Exception
+        ObjectSpace.trace_object_allocations_stop
+        GC.enable
+        raise
+      else
+        stop
+      end
     end
 
     private
@@ -100,7 +107,7 @@ module MemoryProfiler
           class_name = helper.lookup_class_name(klass)
           gem        = helper.guess_gem(file)
 
-          string     = '' << obj  if klass == String
+          string     = klass == String ? helper.lookup_string(obj) : nil
 
           memsize = ObjectSpace.memsize_of(obj) + rvalue_size_adjustment
           # compensate for API bug

data/lib/memory_profiler/results.rb

@@ -63,6 +63,9 @@ module MemoryProfiler
     # @param [Hash] options the options for output
     # @option opts [String] :to_file a path to your log file
     # @option opts [Boolean] :color_output a flag for whether to colorize output
+    # @option opts [Integer] :retained_strings how many retained strings to print
+    # @option opts [Integer] :allocated_strings how many allocated strings to print
+    # @option opts [Boolean] :detailed_report should report include detailed information
     def pretty_print(io = $stdout, **options)
       # Handle the special case that Ruby PrettyPrint expects `pretty_print`
       # to be a customized pretty printing function for a class
@@ -76,26 +79,34 @@ module MemoryProfiler
       io.puts "Total allocated: #{total_allocated_memsize} bytes (#{total_allocated} objects)"
       io.puts "Total retained:  #{total_retained_memsize} bytes (#{total_retained} objects)"
 
-      io.puts
-      ["allocated", "retained"]
-          .product(["memory", "objects"])
-          .product(["gem", "file", "location", "class"])
-          .each do |(type, metric), name|
-            dump "#{type} #{metric} by #{name}", self.send("#{type}_#{metric}_by_#{name}"), io
-          end
+      if options[:detailed_report] != false
+        io.puts
+        ["allocated", "retained"]
+            .product(["memory", "objects"])
+            .product(["gem", "file", "location", "class"])
+            .each do |(type, metric), name|
+              dump "#{type} #{metric} by #{name}", self.send("#{type}_#{metric}_by_#{name}"), io
+            end
+      end
 
       io.puts
-      dump_strings(io, "Allocated", strings_allocated)
+      dump_strings(io, "Allocated", strings_allocated, limit: options[:allocated_strings])
       io.puts
-      dump_strings(io, "Retained", strings_retained)
+      dump_strings(io, "Retained", strings_retained, limit: options[:retained_strings])
 
       io.close if io.is_a? File
     end
 
     private
 
-    def dump_strings(io, title, strings)
+    def dump_strings(io, title, strings, limit: nil)
       return unless strings
+
+      if limit
+        return if limit == 0
+        strings = strings[0...limit]
+      end
+
       io.puts "#{title} String Report"
       io.puts @colorize.line("-----------------------------------")
       strings.each do |string, stats|

data/lib/memory_profiler/version.rb

@@ -1,3 +1,3 @@
 module MemoryProfiler
-  VERSION = "0.9.8"
+  VERSION = "0.9.9"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: memory_profiler
 version: !ruby/object:Gem::Version
-  version: 0.9.8
+  version: 0.9.9
 platform: ruby
 authors:
 - Sam Saffron
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-02-27 00:00:00.000000000 Z
+date: 2018-02-14 00:00:00.000000000 Z
 dependencies: []
 description: Memory profiling routines for Ruby 2.1+
 email:
@@ -50,7 +50,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.2
+rubygems_version: 2.6.13
 signing_key: 
 specification_version: 4
 summary: Memory profiling routines for Ruby 2.1+