devver-rack-contrib 0.9.3 → 0.9.4

data/lib/rack/contrib/perftools_profiler.rb

@@ -0,0 +1,119 @@
+require 'perftools'
+require 'rbconfig'
+
+# You'll need graphviz to generate call graphs using dot:
+#
+#    sudo port install graphviz     # osx
+#    sudo apt-get install graphviz  # debian/ubuntu
+
+# You'll need ps2pdf to generate PDFs (comes with ghostscript)
+# sudo port install ghostscript
+
+module Rack
+
+  # Pass the :printer option to pick a different result format.
+  class PerftoolsProfiler
+    MODES = %w(
+      process_time
+      #wall_time
+    )
+
+    DEFAULT_PRINTER = :text
+    #DEFAULT_CONTENT_TYPE = 'application/octet-stream'
+
+    PRINTER_CONTENT_TYPE = {
+      :text => 'text/plain',
+      :gif => 'image/gif',
+      :pdf => 'application/pdf'
+    }
+
+    # Accepts a :printer => [:call_tree|:graph_html|:graph|:flat] option
+    # defaulting to :call_tree.
+    def initialize(app, options = {})
+      @app = app
+      @printer = parse_printer(options[:printer])
+      @times = (options[:times] || 1).to_i
+    end
+
+    def call(env)
+      if mode = profiling?(env)
+        profile(env, mode)
+      else
+        @app.call(env)
+      end
+    end
+
+    private
+      def profiling?(env)
+        unless PerfTools::CpuProfiler.running?
+          request = Rack::Request.new(env)
+          if mode = request.params.delete('profile')
+            true
+            #if RubyProf.const_defined?(mode.upcase)
+            #  mode
+            #else
+              #env['rack.errors'].write "Invalid RubyProf measure_mode: " +
+              #  "#{mode}. Use one of #{MODES.to_a.join(', ')}"
+            #  false
+            #end
+          end
+        end
+      end
+
+      def profile(env, mode)
+        #RubyProf.measure_mode = RubyProf.const_get(mode.upcase)
+
+        #GC.enable_stats if GC.respond_to?(:enable_stats)
+        #result = RubyProf.profile do
+        temp_file = "/tmp/perftools_file"
+        PerfTools::CpuProfiler.start(temp_file) do 
+          @times.times { @app.call(env) }
+        end
+        #end
+        #GC.disable_stats if GC.respond_to?(:disable_stats)
+
+        [200, headers(@printer, env, mode), print(@printer, temp_file)]
+      end
+
+      def print(printer, temp_file)
+        cmd = "pprof.rb --#{printer} #{temp_file}"
+        `#{cmd}`
+        #body = StringIO.new
+        #printer.new(result).print(body, :min_percent => 0.01)
+        #body.rewind
+        #body
+      end
+
+      def headers(printer, env, mode)
+        headers = { 'Content-Type' => PRINTER_CONTENT_TYPE[printer] || DEFAULT_CONTENT_TYPE }
+        if printer==:pdf
+          filetype = printer
+          filename='profile'
+          headers['Content-Disposition'] = %(attachment; filename="#{filename}.#{mode}.#{filetype}")
+        end
+        headers
+      end
+
+      def parse_printer(printer)
+        if printer.nil?
+          DEFAULT_PRINTER
+        else
+          printer.to_sym
+        end
+        #elsif printer.is_a?(Class)
+        #  printer
+        #else
+        #  name = "#{camel_case(printer)}Printer"
+        #  if RubyProf.const_defined?(name)
+        #    RubyProf.const_get(name)
+        #  else
+        #    DEFAULT_PRINTER
+        #  end
+        #end
+      end
+
+      def camel_case(word)
+        word.to_s.gsub(/(?:^|_)(.)/) { $1.upcase }
+      end
+  end
+end

data/rack-contrib.gemspec

@@ -3,7 +3,7 @@ Gem::Specification.new do |s|
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
 
   s.name = 'devver-rack-contrib'
-  s.version = '0.9.3'
+  s.version = '0.9.4'
   s.date = '2010-01-10'
 
   s.description = "The Devver fork of rack-contrib"
@@ -35,6 +35,7 @@ Gem::Specification.new do |s|
     lib/rack/contrib/mailexceptions.rb
     lib/rack/contrib/nested_params.rb
     lib/rack/contrib/not_found.rb
+    lib/rack/contrib/perftools_profiler.rb
     lib/rack/contrib/post_body_content_type_parser.rb
     lib/rack/contrib/proctitle.rb
     lib/rack/contrib/profiler.rb

metadata

@@ -1,7 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: devver-rack-contrib
 version: !ruby/object:Gem::Version 
-  version: 0.9.3
+  prerelease: false
+  segments: 
+  - 0
+  - 9
+  - 4
+  version: 0.9.4
 platform: ruby
 authors: 
 - rack-devel
@@ -14,44 +19,58 @@ default_executable:
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rack
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 9
+        - 1
         version: 0.9.1
-    version: 
+  type: :runtime
+  version_requirements: *id001
 - !ruby/object:Gem::Dependency 
   name: test-spec
-  type: :development
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
     requirements: 
     - - ~>
       - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 9
+        - 0
         version: 0.9.0
-    version: 
+  type: :development
+  version_requirements: *id002
 - !ruby/object:Gem::Dependency 
   name: tmail
-  type: :development
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 2
         version: "1.2"
-    version: 
+  type: :development
+  version_requirements: *id003
 - !ruby/object:Gem::Dependency 
   name: json
-  type: :development
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 1
         version: "1.1"
-    version: 
+  type: :development
+  version_requirements: *id004
 description: The Devver fork of rack-contrib
 email: rack-devel@googlegroups.com
 executables: []
@@ -83,6 +102,7 @@ files:
 - lib/rack/contrib/mailexceptions.rb
 - lib/rack/contrib/nested_params.rb
 - lib/rack/contrib/not_found.rb
+- lib/rack/contrib/perftools_profiler.rb
 - lib/rack/contrib/post_body_content_type_parser.rb
 - lib/rack/contrib/proctitle.rb
 - lib/rack/contrib/profiler.rb
@@ -135,18 +155,20 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 required_rubygems_version: !ruby/object:Gem::Requirement 
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.5
+rubygems_version: 1.3.6
 signing_key: 
 specification_version: 2
 summary: This gem is the Devver fork of rack-contrib (http://github.com/rack/rack-contrib) so we can play around with new features we're building. You should not use this version and use the official rack-contrib gem instead.