oink 0.1.0

data/spec/oinked_request/oinked_request_spec.rb

@@ -0,0 +1,20 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe OinkedRequest do
+  it "should be comparable" do
+    lr1 = OinkedRequest.new("Controller#Action", "February 1 10:20", [], 10)
+    lr2 = OinkedRequest.new("Controller#Action", "February 1 10:20", [], 5)
+
+    (lr1 > lr2).should == true
+    (lr1 == lr2).should == false
+  end
+
+  it "should sort by memory used" do
+    lr1 = OinkedRequest.new("Controller#Action", "February 1 10:20", [], 10)
+    lr2 = OinkedRequest.new("Controller#Action", "February 1 10:20", [], 5)
+    lr3 = OinkedRequest.new("Controller#Action", "February 1 10:20", [], 30)
+
+    [lr1, lr2, lr3].sort.should == [lr2, lr1, lr3]
+  end
+
+end

data/spec/priority_queue/priority_queue_spec.rb

@@ -0,0 +1,75 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe PriorityQueue do
+  
+  describe "size" do
+    
+    it "should report the right size" do
+      pq = PriorityQueue.new(5)
+      pq.push(1)
+      pq.size.should == 1
+      pq.push(2)
+      pq.size.should == 2
+    end
+    
+    it "should be limited to the size initialized with" do
+      pq = PriorityQueue.new(5)
+      pq.push(1)
+      pq.push(2)
+      pq.push(3)
+      pq.push(4)
+      pq.push(5)
+      pq.push(6)
+      pq.size.should == 5
+    end
+    
+  end
+  
+  describe "order" do
+    
+    it "should be in order from highest to lowest" do
+      pq = PriorityQueue.new(5)
+      pq.push(1)
+      pq.push(2)
+      pq.push(3)
+      pq.to_a.should == [3,2,1]
+    end
+    
+    it "should throw out the lower value when adding a new value" do
+      pq = PriorityQueue.new(3)
+      pq.push(1)
+      pq.push(2)
+      pq.push(3)
+      pq.push(4)
+      pq.to_a.should == [4,3,2]
+    end
+    
+    it "should not make it into the queue if it's smaller than the items in the queue" do
+      pq = PriorityQueue.new(3)
+      pq.push(2)
+      pq.push(3)
+      pq.push(4)
+      pq.push(1)
+      pq.to_a.should == [4,3,2]
+    end
+    
+  end
+  
+  describe "each" do
+    it "should return each item in turn" do
+      arr = []
+      pq = PriorityQueue.new(5)
+      pq.push(2)
+      pq.push(3)
+      pq.push(4)
+      pq.push(1)
+      pq.push(5)
+      pq.each do |i|
+        arr << i
+      end
+      arr.should == [5,4,3,2,1]
+    end
+  end
+  
+  
+end

data/spec/rails/instance_type_counter_spec.rb

@@ -0,0 +1,45 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+describe Oink::OinkInstanceTypeCounterInstanceMethods do
+  before :each do
+    ActiveRecord::Base.reset_instance_type_count
+  end
+  
+  describe "hash" do
+    it "should not count objects not instantiated" do
+      ActiveRecord::Base.instantiated_hash["Pig"].should == nil
+    end
+    
+    it "should include the objects instantiated" do
+      Pig.create(:name => "Babe")
+      Pig.first
+      ActiveRecord::Base.instantiated_hash["Pig"].should == 2
+    end
+    
+    it "should count instantiations for multiple classes" do
+      Pig.create(:name => "Babe")
+      Pen.create(:location => "Backyard")
+      Pig.first
+      ActiveRecord::Base.instantiated_hash["Pen"].should == 1
+    end
+    
+    it "should report the total number of objects instantiated" do
+      Pig.create(:name => "Babe")
+      Pen.create(:location => "Backyard")
+      Pig.first
+      ActiveRecord::Base.total_objects_instantiated.should == 3
+    end    
+  end
+  
+  describe "reset" do
+    it "should reset the total count" do
+      Pig.create(:name => "Babe")
+      ActiveRecord::Base.instantiated_hash["Pig"].should == 1
+      ActiveRecord::Base.reset_instance_type_count
+      ActiveRecord::Base.total_objects_instantiated.should == 0
+      Pig.create(:name => "Babe")
+      ActiveRecord::Base.total_objects_instantiated.should == 0
+    end
+  end
+
+end

data/spec/rails/memory_usage_logger_spec.rb

@@ -0,0 +1,74 @@
+require File.expand_path(File.dirname(__FILE__) + "/../spec_helper")
+
+class ApplicationController
+  attr_accessor :backtick
+
+  class Logger
+    attr_accessor :log
+
+    def info(*args)
+      (@log ||= []) << [:info, *args]
+    end
+  end
+
+  class << self
+    attr_accessor :after_filters
+
+    def after_filter method
+      (@after_filters ||= []) << method
+    end
+  end
+
+  include Oink::MemoryUsageLogger
+
+  def index
+    run_after_filters
+  end
+
+  def logger
+    @logger ||= Logger.new
+  end
+
+  def `(*args)
+    (@backtick ||= []) << args
+    '915'
+  end
+
+  protected
+  def run_after_filters
+    self.class.after_filters.each { |filter| self.send(filter) }
+  end
+end
+
+describe Oink::MemoryUsageLogger do
+  unless defined? WIN32OLE
+    describe "get_memory_usage" do
+      it "should work on linux" do
+        proc_file = <<-STR
+            Header
+
+            Size: 25
+            Size: 13 trailing
+
+      leading Size: 4
+
+            Footer
+
+            STR
+
+        File.should_receive(:new).with("/proc/#{$$}/smaps").and_return(proc_file)
+        controller = ApplicationController.new
+        controller.index
+        [[:info, "Memory usage: 42 | PID: #{$$}"]].should == controller.logger.log
+      end
+
+      it "should work on non-linux" do
+        File.stub!(:new).and_raise(Errno::ENOENT.new("No such file or directory"))
+        controller = ApplicationController.new
+        controller.index
+        [["ps -o vsz= -p #{$$}"]].should == controller.backtick
+        [[:info, "Memory usage: 915 | PID: #{$$}"]].should == controller.logger.log
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,37 @@
+require "rubygems"
+require "spec"
+require 'ostruct'
+
+dir = File.dirname(__FILE__)
+require File.join(dir, "/../lib/oink.rb")
+require "oink/rails/instance_type_counter"
+require "oink/rails/memory_usage_logger"
+require File.join(dir, '../config/environment')
+
+class PsuedoOutput < Array
+  
+  def puts(line)
+    self << line
+  end
+  
+end
+
+Spec::Runner.configure do |config|
+
+  config.before :suite do
+    load File.join(dir, "../db/schema.rb")
+  end
+
+  config.before :each do
+    Pig.delete_all
+    Pen.delete_all
+  end
+
+  config.before :suite do
+    ActiveRecord::Base.send(:include, Oink::OinkInstanceTypeCounterInstanceMethods)
+    Pig = Class.new(ActiveRecord::Base)
+    Pen = Class.new(ActiveRecord::Base)
+    Pig.belongs_to :pen
+  end
+
+end

metadata

@@ -0,0 +1,84 @@
+--- !ruby/object:Gem::Specification 
+name: oink
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Noah Davis
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-08-10 00:00:00 -04:00
+default_executable: oink
+dependencies: []
+
+description: Log parser to identify actions which significantly increase VM heap size
+email: noahd1@yahoo.com
+executables: 
+- oink
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- History.txt
+- MIT-LICENSE
+- README.rdoc
+- Rakefile
+- bin/oink
+- lib/oink.rb
+- lib/oink/active_record_instantiation_reporter.rb
+- lib/oink/base.rb
+- lib/oink/cli.rb
+- lib/oink/memory_usage_reporter.rb
+- lib/oink/oinked_request/oinked_ar_request.rb
+- lib/oink/oinked_request/oinked_memory_request.rb
+- lib/oink/oinked_request/oinked_request.rb
+- lib/oink/priority_queue.rb
+- lib/oink/rails.rb
+- lib/oink/rails/instance_type_counter.rb
+- lib/oink/rails/memory_usage_logger.rb
+has_rdoc: true
+homepage: http://github.com/noahd1/oink
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Log parser to identify actions which significantly increase VM heap size
+test_files: 
+- spec/oink/active_record_instantiation_reporter_spec.rb
+- spec/oink/memory_usage_reporter_spec.rb
+- spec/oinked_request/oinked_request_spec.rb
+- spec/priority_queue/priority_queue_spec.rb
+- spec/rails/instance_type_counter_spec.rb
+- spec/rails/memory_usage_logger_spec.rb
+- spec/spec_helper.rb