zenprofile 1.0.2 → 1.1.0

data/History.txt

@@ -1,3 +1,13 @@
+=== 1.1.0 / 2009-06-23
+
+* 1 major enhancement:
+
+  * Added memory_profiler.rb. It is rad.
+
+* 1 minor enhancement:
+
+  * Updated rakefile for new hoe capabilities.
+
 === 1.0.0 / 2008-09-17
 
 * 1 major enhancement

data/Manifest.txt

@@ -3,6 +3,7 @@ Manifest.txt
 README.txt
 Rakefile
 bin/zenprofile
+lib/memory_profiler.rb
 lib/spy_on.rb
 lib/zenprofile.rb
 lib/zenprofiler.rb

data/Rakefile

@@ -2,15 +2,17 @@
 
 require 'rubygems'
 require 'hoe'
-require './lib/zenprofiler.rb'
 
-Hoe.new('zenprofile', ZenProfiler::VERSION) do |p|
-  p.rubyforge_name = 'seattlerb'
-  p.developer('Ryan Davis', 'ryand-ruby@zenspider.com')
+Hoe.plugin :seattlerb
+Hoe.plugin :inline
 
-  p.extra_deps << 'ZenTest'
-  p.extra_deps << 'event_hook'
-  p.extra_deps << 'RubyInline'
+Hoe.spec 'zenprofile' do
+  developer 'Ryan Davis', 'ryand-ruby@zenspider.com'
+
+  self.rubyforge_name = 'seattlerb'
+
+  extra_deps << 'ZenTest'
+  extra_deps << 'event_hook'
 end
 
-# vim: syntax=Ruby
+# vim: syntax=ruby

data/lib/memory_profiler.rb

@@ -0,0 +1,108 @@
+
+class MemoryProfiler
+
+  # profile is a CLEAN, totally minimal memory profiler:
+  #
+  # * 1 thread
+  # * 2 hashes (curr, prev)
+  # * classes as keys
+  # * fixnums as values.
+  # 
+  # No extra memory required if I can help it. Keep it as simple as
+  # possible. The code at the link above is WAY to complex for such a
+  # simple task.  It does use a sort_by, so that produces a bit extra
+  # junk... but that shouldn't be more than an extra array and hash
+  # per iteration.
+  # 
+  # Just fire it up as MemoryProfiler.profile or use it with a block
+  # of code:
+  # 
+  #   MemoryProfiler.profile do
+  #     # ...
+  #   end
+
+  def self.profile(delay = 2, limit=10, output=$stderr)
+    Thread.abort_on_exception = true
+    totals = Hash.new(0)
+    t = Thread.new do
+      prev = Hash.new(0)
+      curr = Hash.new(0)
+      loop do
+        curr.clear
+        ObjectSpace.each_object do |o|
+          curr[o.class] += 1
+        end
+
+        curr.each do |k,v|
+          curr[k] -= prev[k]
+        end
+
+        puts
+        curr.sort_by { |k,v| -v }.first(limit).each do |k,v|
+          output.printf "%+5d: %s\n", v, k.name unless v == 0
+        end
+
+        prev.clear
+        prev.update curr
+        curr.each do |k,n| totals[k] += n end
+        sleep delay
+      end
+    end
+    if block_given? then
+      yield
+      t.exit
+    end
+    totals
+  end
+
+  @@locations = {}
+  @@allocations = Hash.new { |h,k| h[k] = Hash.new(0) }
+
+  def self.log(o)
+    loc = caller[1]
+    unless @@locations.has_key? loc then
+      @@locations[loc] = @@locations.size
+    end
+    loc = @@locations[loc]
+    @@allocations[o.class][loc] += 1
+  end
+
+  def self.spy_on *klasses
+    klasses.each do |klass|
+      hook klass, :initialize
+      if klass == String then
+        hook klass, :%
+        hook klass, :strip
+      end
+    end
+  end
+
+  def self.hook klass, meth
+    klass.module_eval do
+      old = meth.to_s.sub(/%/, 'pct').sub(/^/, 'old_')
+      alias_method old, meth
+      eval "def #{meth}(*args, &block)
+        MemoryProfiler.log(self)
+        #{old}(*args, &block)
+      end"
+    end
+  end
+
+  def self.report
+    keys = @@locations.invert
+
+    puts
+    puts "############################################################"
+    puts "Spy Report:"
+
+    @@allocations.each do |klass, locations|
+      puts
+      puts klass
+      puts
+      locations.sort_by { |k,v| -v }.first(10).each do |location, count|
+        printf "%6d: %s\n", count, keys[location]
+      end
+    end
+    p @@locations, @@allocations, keys if $DEBUG
+  end
+end

data/lib/zenprofiler.rb

@@ -17,12 +17,12 @@ require 'event_hook'
 #    % ruby -rzenprofile misc/factorial.rb
 
 class ZenProfiler < EventHook
-  VERSION = '1.0.2'
+  VERSION = '1.1.0'
 
-  @@start = nil
-  @@stack = [[0, 0, [nil, :toplevel]], [0, 0, [nil, :dummy]]]
-  @@map   = Hash.new { |h,k| h[k] = [0, 0.0, 0.0, k] }
-  @@map["#toplevel"] = [1, 0.0, 0.0, [nil, "#toplevel"]]
+  @@start                  = nil
+  @@stack                  = [[0, 0, [nil, :toplevel]], [0, 0, [nil, :dummy]]]
+  @@map                    = Hash.new { |h,k| h[k] = [0, 0.0, 0.0, k] }
+  @@map["#toplevel"]       = [1, 0.0, 0.0, [nil, "#toplevel"]]
   @@percent_time_threshold = 0.5
 
   def self.run(fp = $stdout, opts = {})

data/misc/factorial.rb

@@ -17,7 +17,7 @@ end
 if $0 == __FILE__ then
   f = Factorial.new()
 
-  max = ARGV.shift || 1000000
+  max = ARGV.shift || 1_000_000
   max = max.to_i
 
   tstart = Time.now

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: zenprofile
 version: !ruby/object:Gem::Version 
-  version: 1.0.2
+  version: 1.1.0
 platform: ruby
 authors: 
 - Ryan Davis
@@ -30,7 +30,7 @@ cert_chain:
   FBHgymkyj/AOSqKRIpXPhjC6
   -----END CERTIFICATE-----
 
-date: 2009-05-20 00:00:00 -07:00
+date: 2009-06-23 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -53,16 +53,6 @@ dependencies:
       - !ruby/object:Gem::Version 
         version: "0"
     version: 
-- !ruby/object:Gem::Dependency 
-  name: RubyInline
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        version: "0"
-    version: 
 - !ruby/object:Gem::Dependency 
   name: hoe
   type: :development
@@ -71,7 +61,7 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        version: 1.12.2
+        version: 2.3.0
     version: 
 description: |-
   zenprofiler helps answer WHAT is being called the most. spy_on helps
@@ -120,6 +110,7 @@ files:
 - README.txt
 - Rakefile
 - bin/zenprofile
+- lib/memory_profiler.rb
 - lib/spy_on.rb
 - lib/zenprofile.rb
 - lib/zenprofiler.rb
@@ -150,7 +141,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: seattlerb
-rubygems_version: 1.3.2
+rubygems_version: 1.3.4
 signing_key: 
 specification_version: 3
 summary: zenprofiler helps answer WHAT is being called the most