lib/memprof2.rb in memprof2-0.0.1 vs lib/memprof2.rb in memprof2-0.0.2

@@ -1,8 +1,8 @@
 require 'objspace'
 
-module Memprof2
+class Memprof2
   class << self
     def start
       ObjectSpace.trace_object_allocations_start
     end
 
@@ -13,41 +13,60 @@
 
     def run(&block)
       ObjectSpace.trace_object_allocations(&block)
     end
 
-    def report(opts={})
+    def report(opts = {})
       ObjectSpace.trace_object_allocations_stop
-      @rvalue_size = GC::INTERNAL_CONSTANTS[:RVALUE_SIZE]
-      if @trace = opts[:trace]
-        raise ArgumentError, "`trace` option must be a Regexp object" unless @trace.is_a?(Regexp)
-      end
-      if @ignore = opts[:ignore]
-        raise ArgumentError, "`trace` option must be a Regexp object" unless @ignore.is_a?(Regexp)
-      end
-      results = {}
-      ObjectSpace.each_object do |o|
-        next unless (file = ObjectSpace.allocation_sourcefile(o))
-        next if file == __FILE__
-        next if (@trace  and @trace !~ file)
-        next if (@ignore and @ignore =~ file)
-        line = ObjectSpace.allocation_sourceline(o)
-        memsize  = ObjectSpace.memsize_of(o) + @rvalue_size
-        memsize = @rvalue_size if memsize > 100_000_000_000 # compensate for API bug
-        klass = o.class.name rescue "BasicObject"
-        location = "#{file}:#{line}:#{klass}"
-        results[location] ||= 0
-        results[location] += memsize
-      end
-      @out = opts[:out] || "/dev/stdout"
-      File.open(@out, 'w') do |io|
-        results.each do |location, memsize|
-          io.puts "#{memsize} #{location}"
-        end
-      end
+      self.new.report(opts)
     ensure
       ObjectSpace.trace_object_allocations_start
     end
+
+    def report!(opts = {})
+      report(opts)
+      ObjectSpace.trace_object_allocations_clear
+    end
+  end
+
+  def report(opts={})
+    configure(opts)
+    results = collect_info
+    File.open(@out, 'w') do |io|
+      results.each do |location, memsize|
+        io.puts "#{memsize} #{location}"
+      end
+    end
+  end
+
+  def configure(opts = {})
+    @rvalue_size = GC::INTERNAL_CONSTANTS[:RVALUE_SIZE]
+    if @trace = opts[:trace]
+      raise ArgumentError, "`trace` option must be a Regexp object" unless @trace.is_a?(Regexp)
+    end
+    if @ignore = opts[:ignore]
+      raise ArgumentError, "`ignore` option must be a Regexp object" unless @ignore.is_a?(Regexp)
+    end
+    @out = opts[:out] || "/dev/stdout"
+    self
+  end
+
+  def collect_info
+    results = {}
+    ObjectSpace.each_object do |o|
+      next unless (file = ObjectSpace.allocation_sourcefile(o))
+      next if file == __FILE__
+      next if (@trace  and @trace !~ file)
+      next if (@ignore and @ignore =~ file)
+      line = ObjectSpace.allocation_sourceline(o)
+      memsize = ObjectSpace.memsize_of(o) + @rvalue_size
+      memsize = @rvalue_size if memsize > 100_000_000_000 # compensate for API bug
+      klass = o.class.name rescue "BasicObject"
+      location = "#{file}:#{line}:#{klass}"
+      results[location] ||= 0
+      results[location] += memsize
+    end
+    results
   end
 end