onis 0.1

data/README.md

@@ -0,0 +1,81 @@
+# onis
+(pronounced onis)
+
+Onis gives you introspection into object counts of live running ruby process.
+
+## Install
+
+    $ sudo gem install onis
+
+## Example
+
+    $ cd onis/
+
+    $ cat test/stringpwd.rb &
+    require 'onis'
+    puts Process.pid
+    a = []
+    while true ; a << "foo" ; sleep(0.5) ; end
+
+    $ onis
+    PID        COMMAND
+    ------------------------------------------------------------
+    26584      ruby -I lib/ test/stringpwnd.rb
+
+    $ onis -f 26584
+    Class                                         Count
+    --------------------------------------------------------
+    String                                        3483
+    Class                                         156
+    Array                                         55
+    Module                                        25
+    Regexp                                        9
+    File                                          8
+    Hash                                          7
+    Float                                         6
+    Object                                        3
+    ...
+
+## Color!
+
+    $ onis -f -c 26584
+
+## How it works
+
+  1. Onis drops a tempfile named after the pid `onis.<pid>`
+  2. Onis then traps SIGUSR1
+  3. When SIGUSR1 is sent to a process that
+     has `require`d Onis, Onis will dump the ObjectSpace and counts
+     to that tempfile.
+
+## TODO:
+  * handle `fork`
+
+## Thanks
+  - Aman Gupta (tmm1) for inspiration and example
+  - Ryan Tomayko for playing that great music at Columbus Cafe
+    and telling me to take this one step further
+
+## License
+Copyright (c) 2009 Blake Mizerany
+
+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/bin/onis

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../lib"
+
+require 'optparse'
+require 'onis/base'
+
+# default options
+pid     = nil
+force   = false
+watch   = false
+color   = false
+
+# setup command
+command = `uname`.chomp == "Darwin" ? "command" : "cmd"
+psax    = `ps -axo pid,#{command}`
+format  = "%-10s %s\n"
+
+ARGV.options do |o|
+  o.banner = "Usage: onis [options] [pid]"
+  o.separator ""
+  o.separator "Run without params for process list"
+
+  o.on("-f", "--force", "force report") { force = true }
+  o.on("-w", "--watch", "watch report. -f is implied") { watch = true }
+  o.on("-c", "--color", "highlight every other line in report") { color = true }
+
+  o.on_tail("-h", "--help", "show this message") { abort(o.to_s) }
+  o.parse!
+end
+
+P = lambda {|line|
+  @n ||= 0
+  print "\e[7m" if color && ((@n += 1) % 2).zero?
+  puts line
+  print "\e[m" if color
+}
+
+if pid = ARGV.shift
+  if watch
+    exec "watch -n0 \"#{$0} -f #{pid}\""
+  elsif force
+    Process.kill("SIGUSR1", pid.to_i)
+    sleep(0.5)
+  end
+
+  pfp = Onis::Base.path_for_pid(pid)
+  abort("report for #{pid} not found") if !File.file?(pfp)
+
+  open(pfp) do |f|
+      format = "%-45s %-10s\n"
+      puts format % %w|Class Count|
+      puts "-" * 56
+      f.each_line do |line|
+        P[format % line.chomp.split(",")]
+      end
+  end
+else
+  print format % %w|PID COMMAND|
+  puts  "-" * 60 
+  psax.each_line do |line|
+    next unless line =~ /^\d+/
+    pid = line.slice!(/^\d+ /).strip
+    next if pid == Process.pid.to_s
+    pfp = Onis::Base.path_for_pid(pid)
+    next if !File.file?(pfp)
+    P[format % [pid, line]]
+  end
+end
+
+#vim: syn=ruby

data/lib/onis.rb

@@ -0,0 +1,3 @@
+require 'onis/base'
+
+Onis::Base.start!

data/lib/onis/base.rb

@@ -0,0 +1,35 @@
+require 'tempfile'
+require 'fileutils'
+
+module Onis
+  module Base
+    def self.drop_pid(pid = Process.pid)
+      path_for_pid(pid).tap {|pfp| FileUtils.touch(pfp)}
+    end
+
+    def self.path_for_pid(pid)
+      File.join(Dir.tmpdir, "onis.#{pid}")
+    end
+
+    def self.start!
+      pfp = drop_pid
+      trap("USR1") do
+        open(pfp, 'w') do |f|
+          ObjectInspector.all.each do |o|
+            f.puts o.join(",")
+          end
+        end
+      end
+    end
+  end
+
+  module ObjectInspector
+    def self.all
+      h = Hash.new(0)
+      ObjectSpace.each_object do |o|
+        h[o.class] += 1
+      end
+      h.sort {|a, b| b.last <=> a.last}
+    end
+  end
+end

data/onis.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+
+  s.name = 'onis'
+  s.version = '0.1'
+  s.date = '2009-10-27'
+
+  s.description = "Live ObjectSpace introspection"
+  s.summary     = s.description
+
+  s.authors = ["Blake Mizerany"]
+  s.email = "blake.mizerany@gmail.com"
+
+  s.files = %w[
+    README.md
+    onis.gemspec
+    lib/onis.rb
+    lib/onis/base.rb
+    bin/onis
+  ]
+  s.executables = ['onis']
+
+  s.extra_rdoc_files = %w[README.md]
+
+  s.homepage = "http://github.com/bmizerany/onis/"
+  s.require_paths = %w[lib]
+  s.rubygems_version = '1.1.1'
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification 
+name: onis
+version: !ruby/object:Gem::Version 
+  version: "0.1"
+platform: ruby
+authors: 
+- Blake Mizerany
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-27 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Live ObjectSpace introspection
+email: blake.mizerany@gmail.com
+executables: 
+- onis
+extensions: []
+
+extra_rdoc_files: 
+- README.md
+files: 
+- README.md
+- onis.gemspec
+- lib/onis.rb
+- lib/onis/base.rb
+- bin/onis
+has_rdoc: true
+homepage: http://github.com/bmizerany/onis/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.4
+signing_key: 
+specification_version: 2
+summary: Live ObjectSpace introspection
+test_files: []
+