perf_check 0.0.2 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8d4485198a9057ea60ba1525b6ac00a1ef202aa3
-  data.tar.gz: fb4528654f7042955e990f6f267b3378a03b5138
+  metadata.gz: 1113bc3c586dc1e5f7e8d4d3ef14b8140151b486
+  data.tar.gz: e3f47dba96b00bc1061c4ff2b8f63e923d5a9fc9
 SHA512:
-  metadata.gz: 6771cc650d1712c2308f56625b7f177edd1484cfbd57b368e8ca977974f2a9b45003ad7b49647233c25756528fa4a288384c7580279784146f86c76f95db4a9c
-  data.tar.gz: 70e0bcf7aa68b5161eb09096c2047113515170fb63a6a2a4130e9e4ffc1c141909276bdc5101ea0058992ffef0bb9a7a66446cfe218196a3aead3f004169497f
+  metadata.gz: ceb56fbfa1a430635e5b172e0b268f0eaac51cbd5d323f73732822d8b5e823e14be511aae01b870684da415b43c6738b9b598a078ab13201d9b59910615469db
+  data.tar.gz: c44e78da727beef6a2f37a323c19edba6acfa2b3b07f147ba415a1d37d87627065c4d98b658b953986ab5b6e7993f7c8812aa9f4f762dcfb82fa551e5e73e04a

data/bin/perf_check

@@ -8,6 +8,7 @@ options.login = :admin
 options.number_of_requests = 10
 options.reference = 'master'
 
+
 opts = OptionParser.new do |opts|
   opts.banner = "Usage: perf_check [options] [route ...]"
 
@@ -49,6 +50,12 @@ opts = OptionParser.new do |opts|
     options.reference = nil
   end
 
+  opts.separator "\nOther"
+  opts.on('--server') do
+    PerfCheck::Server.start
+    exit!(0)
+  end
+
   opts.separator ''
   opts.separator <<EOF
 Usage examples:
@@ -67,19 +74,7 @@ EOF
 end
 opts.parse!
 
-app_root = Dir.pwd
-until app_root == '/' || File.exist?("#{app_root}/config/application.rb")
-  app_root = File.dirname(app_root)
-end
-
-unless File.exist?("#{app_root}/config/application.rb")
-  abort("perf_check should be run from a rails directory")
-end
-
-ENV['RAILS_ENV'] = 'development'
-ENV['PERF_CHECK'] = '1'
-require File.expand_path("#{app_root}/config/environment")
-
+PerfCheck.require_rails
 
 ARGV.each do |route|
   perf_check.add_test_case(route)

data/lib/perf_check/railtie.rb

@@ -0,0 +1,19 @@
+class PerfCheck
+  class Railtie < Rails::Railtie
+
+    config.before_initialize do
+
+      if defined?(Rack::MiniProfiler)
+        # Integrate with rack-mini-profiler
+        tmp = "#{Rails.root}/tmp/perf_check/miniprofiler"
+        FileUtils.mkdir_p(tmp)
+
+        Rack::MiniProfiler.config.storage_instance =
+          Rack::MiniProfiler::FileStore.new(:path => tmp)
+      end
+
+      # Force cache_classes = true .... :\
+      Rails::Application::Configuration.send(:define_method, :cache_classes){ true }
+    end
+  end
+end

data/lib/perf_check/server.rb

@@ -10,6 +10,17 @@ class PerfCheck
       define_method(:login, &block)
     end
 
+    def self.sign_cookie_data(key, data, opts={})
+      opts[:serializer] ||= Marshal
+      secret = Rails.application.config.secret_token
+
+      marshal = ActiveSupport::MessageVerifier.new(secret,
+                                                   :serializer => opts[:serializer])
+      marshal_value = marshal.generate(data)
+
+      "#{key}=#{marshal_value}"
+    end
+
     def initialize
       at_exit do
         exit

data/lib/perf_check/test_case.rb

@@ -19,8 +19,14 @@ class PerfCheck
     end
 
     def run(server, count)
+      print("\t"+'request #'.underline)
+      print("  "+'latency'.underline)
+      print("   "+'server rss'.underline)
+      puts("   "+'profiler data'.underline)
+
       (count+1).times do |i|
         errors = 0
+
         begin
           profile = server.profile do |http|
             http.get(resource, {'Cookie' => cookie})
@@ -29,8 +35,11 @@ class PerfCheck
           File.open("public/perf_check_failed_request.html", 'w') do |error_dump|
             error_dump.write(e.body)
           end
-          printf("\tRequest %2i: —— FAILURE (HTTP %s): %s\n",
-                 i, e.code, '/perf_check_failed_request.html')
+          error = sprintf("\t%2i:\tFAILED! (HTTP %s)", i, e.code)
+          puts(error.red.bold)
+          puts("\t   The server responded with a non-2xx status for this request.")
+          print("\t   The response has been written to public")
+          puts("/perf_check_failed_request.html".blue)
           exit(1)
         end
 
@@ -38,7 +47,7 @@ class PerfCheck
         # all the autoload overhead (?)
         next if i.zero?
 
-        printf("\tRequest %2i: %.1fms\t%4dMB\t%s\n",
+        printf("\t%2i:\t   %.1fms   %4dMB\t  %s\n",
                i, profile.latency, server.mem, profile.profile_url)
 
         self.latencies << profile.latency

data/lib/perf_check.rb

@@ -6,10 +6,26 @@ require 'digest'
 require 'fileutils'
 require 'benchmark'
 require 'ostruct'
+require 'colorize'
 
 class PerfCheck
   attr_accessor :options, :server, :test_cases
 
+  def self.require_rails
+    ENV['PERF_CHECK'] = '1'
+
+    app_root = Dir.pwd
+    until app_root == '/' || File.exist?("#{app_root}/config/application.rb")
+      app_root = File.dirname(app_root)
+    end
+
+    unless File.exist?("#{app_root}/config/application.rb")
+      abort("perf_check should be run from a rails directory")
+    end
+
+    require "#{app_root}/config/environment"
+  end
+
   def initialize
     self.options = OpenStruct.new
     self.server = Server.new
@@ -22,6 +38,10 @@ class PerfCheck
   end
 
   def sanity_check
+    if ENV['RAILS_ENV'] == 'production'
+      abort("perf_check cannot be run in the production environment")
+    end
+
     if Git.current_branch == "master"
       puts("Yo, profiling master vs. master isn't too useful, but hey, we'll do it")
     end
@@ -69,15 +89,17 @@ class PerfCheck
       if test.latency_difference < 0
         formatted_change = sprintf('%.1fx', test.latency_factor)
         formatted_change = "yours is #{formatted_change} faster!"
+        color = :green
       else
         formatted_change = sprintf('%.1fx', 1.0 / test.latency_factor)
         formatted_change = "yours is #{formatted_change} slower!!!"
+        color = :light_red
       end
       formatted_change = difference + " (#{formatted_change})"
 
       puts("master: ".rjust(15)     + "#{master_latency}")
       puts("your branch: ".rjust(15)+ "#{this_latency}")
-      puts("change: ".rjust(15)     + "#{formatted_change}")
+      puts(("change: ".rjust(15)     + "#{formatted_change}").bold.send(color))
     end
   end
 end
@@ -86,3 +108,7 @@ end
 require 'perf_check/server'
 require 'perf_check/test_case'
 require 'perf_check/git'
+
+if defined?(Rails)
+  require 'perf_check/railtie'
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: perf_check
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.7
 platform: ruby
 authors:
 - rubytune
@@ -9,7 +9,21 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2014-06-30 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.3
 description: 
 email: 
 executables:
@@ -20,6 +34,7 @@ files:
 - bin/perf_check
 - lib/perf_check.rb
 - lib/perf_check/git.rb
+- lib/perf_check/railtie.rb
 - lib/perf_check/server.rb
 - lib/perf_check/test_case.rb
 homepage: https://github.com/rubytune/perf_check