request_profiler 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+pkg/*
+*.gem
+.bundle

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in request_profiler.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,30 @@
+PATH
+  remote: .
+  specs:
+    request_profiler (0.0.1)
+      ruby-prof
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    mocha (0.9.10)
+      rake
+    rack (1.2.1)
+    rack-test (0.5.6)
+      rack (>= 1.0)
+    rake (0.8.7)
+    ruby-prof (0.9.2)
+    sinatra (1.1.0)
+      rack (~> 1.1)
+      tilt (~> 1.1)
+    tilt (1.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  mocha
+  rack-test
+  request_profiler!
+  ruby-prof
+  sinatra

data/README.rdoc

@@ -0,0 +1,3 @@
+= Request Profiler
+
+- A Rack middleware wrapper for ruby-prof

data/Rakefile

@@ -0,0 +1,13 @@
+require 'bundler'
+require 'rake/testtask'
+Bundler::GemHelper.install_tasks
+
+task :default => :test
+task :build => :test
+task :release => :test
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.test_files = FileList['test/*_test.rb']
+  t.verbose = true
+end

data/lib/rack/request_profiler.rb

@@ -0,0 +1,69 @@
+require 'ruby-prof'
+
+module Rack
+  class RequestProfiler
+    def initialize(app, options = {})
+      @app = app
+      @printer = options[:printer] || ::RubyProf::GraphHtmlPrinter
+      @mode = options[:mode] || ::RubyProf::PROCESS_TIME
+      @eliminations = options[:eliminations]
+
+      @path = options[:path]
+      @path ||= Rails.root + 'tmp/performance' if defined?(Rails)
+      @path ||= ::File.join(ENV["TMPDIR"] + 'performance')
+      @path = Pathname(@path)
+    end
+
+    def call(env)
+      request = Rack::Request.new(env)
+      profile_request = request.params.delete("profile_request") == "true"
+
+      if profile_request
+        ::RubyProf.measure_mode = @mode
+        ::RubyProf.start
+      end
+      status, headers, body = @app.call(env)
+
+      if profile_request
+        result = ::RubyProf.stop
+        write_result(result, request)
+      end
+      
+      [status, headers, body]
+    end
+
+    def format(printer)
+      case printer
+      when ::RubyProf::FlatPrinter
+        'txt'
+      when ::RubyProf::FlatPrinterWithLineNumbers
+        'txt'
+      when ::RubyProf::GraphPrinter
+        'txt'
+      when ::RubyProf::GraphHtmlPrinter
+        'html'
+      when ::RubyProf::DotPrinter
+        'dot'
+      when ::RubyProf::CallTreePrinter
+        "out.#{Process.pid}"
+      when ::RubyProf::CallStackPrinter
+        'html'
+      else
+        'txt'
+      end
+    end
+
+    def write_result(result, request)
+      printer = @printer.new(result)
+      Dir.mkdir(@path) unless ::File.exists?(@path)
+      url = request.fullpath.gsub(/[?\/]/, '-')
+      filename = "#{Time.now.strftime('%Y-%m-%d-%H-%M-%S')}-#{url}.#{format(printer)}"
+      ::File.open(@path + filename, 'w+') do |f|
+        # HACK to keep this from crashing under patched 1.9.2
+        GC.disable
+        printer.print(f)
+        GC.enable
+      end
+    end
+  end
+end

data/lib/request_profiler/version.rb

@@ -0,0 +1,3 @@
+module RequestProfiler
+  VERSION = "0.0.1"
+end

data/lib/request_profiler.rb

@@ -0,0 +1 @@
+require 'rack/request_profiler'

data/request_profiler.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "request_profiler/version"
+
+Gem::Specification.new do |s|
+  s.name        = "request_profiler"
+  s.version     = RequestProfiler::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Justin Weiss"]
+  s.email       = ["justin@uberweiss.org"]
+  s.homepage    = "https://github.com/justinweiss/request_profiler"
+  s.summary     = %q{Profile Rack requests with ruby-prof}
+  s.description = %q{Request Profiler is a Rack middleware that allows optionally profiling requests with ruby-prof.}
+
+  s.rubyforge_project = "request_profiler"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'ruby-prof'
+  s.add_development_dependency 'rack-test'
+  s.add_development_dependency 'sinatra'
+  s.add_development_dependency 'mocha'
+end

data/test/fake_app.rb

@@ -0,0 +1,7 @@
+require 'sinatra'
+
+class FakeApp < Sinatra::Base
+  get "/" do
+    "foo"
+  end
+end

data/test/request_profiler_test.rb

@@ -0,0 +1,44 @@
+require 'test_helper'
+require 'fake_app'
+require 'mocha'
+
+class RequestProfilerTest < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  attr_accessor :app
+
+  def teardown
+    @app = nil
+  end
+
+  def test_basic_request_isnt_profiled
+    self.app = Rack::RequestProfiler.new(FakeApp.new)
+    RubyProf.expects(:start).never
+    RubyProf.expects(:stop).never
+    get "/"
+    assert last_response.ok?
+
+    get "/?profile_request=false"
+    assert last_response.ok?
+  end
+
+  def test_basic_request_is_profiled
+    self.app = Rack::RequestProfiler.new(FakeApp.new)
+    RubyProf.expects(:start).once
+    RubyProf.expects(:stop).once
+    app.expects(:write_result).once
+    get "/?profile_request=true"
+    assert last_response.ok?
+  end
+
+  def test_file_format
+    self.app = Rack::RequestProfiler.new(FakeApp.new)
+    RubyProf.start
+    results = RubyProf.stop
+    printer = self.app.instance_variable_get(:@printer).new(results)
+    
+    assert_equal ::RubyProf::GraphHtmlPrinter, printer.class
+    assert_equal 'html', self.app.format(printer)
+  end
+  
+end

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'request_profiler'
+require 'test/unit'
+require 'rack/test'

metadata

@@ -0,0 +1,129 @@
+--- !ruby/object:Gem::Specification 
+name: request_profiler
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Justin Weiss
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-15 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ruby-prof
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rack-test
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: sinatra
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+description: Request Profiler is a Rack middleware that allows optionally profiling requests with ruby-prof.
+email: 
+- justin@uberweiss.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- README.rdoc
+- Rakefile
+- lib/rack/request_profiler.rb
+- lib/request_profiler.rb
+- lib/request_profiler/version.rb
+- request_profiler.gemspec
+- test/fake_app.rb
+- test/request_profiler_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: https://github.com/justinweiss/request_profiler
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: request_profiler
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Profile Rack requests with ruby-prof
+test_files: 
+- test/fake_app.rb
+- test/request_profiler_test.rb
+- test/test_helper.rb