corn 0.5.2 → 0.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 51b599d133e23dba5f4266e8fa347c1792f024aa
-  data.tar.gz: a33e7f54214b4506cb5887ba1403a4eca2645165
+  metadata.gz: a48acabdb1beaeceb6915bca571c759409f21812
+  data.tar.gz: f046e870baf89b719f4c4de7865ad8212f25e82f
 SHA512:
-  metadata.gz: 01206afb702625ba0bd4afce972670b04e92a36c4ecb712593155e53901b651fbc1d67e12b2828a359db571532dc4c42994e04ff80864d2e606dc73258057e20
-  data.tar.gz: 6b53aeb5b82e2fd905df1257662403f7c8a381d55186a6371c4ff05e0ef4186eedbef498aa2999687e8df19058066e7ebf12ac6431b498ef2962e4ad808d2d21
+  metadata.gz: 70b0f65086ebb697d15394f7c85df9e2b198c1d38ca4379dd5b79ee2c72894c67243f0feea02672db994f05734e37ef9ecaf1a12f77c5b3e4366623afa2488cf
+  data.tar.gz: b8a35e2f0d5c829b51db9f0b8cdcb2b0d98f0839d988447d557dc98f440238718f6d9976ff6fe12eb4a587cbc66b41d9d29850fc02555e88dd3b240eacf4ce72

data/lib/corn/config.rb

@@ -19,7 +19,8 @@ module Corn
             q = !!value == value ? '?' : ''
             self.class_eval <<-RUBY, __FILE__, __LINE__
               def self.#{key}#{q}
-                @config[:#{key}]
+                r = @config[:#{key}]
+                r.is_a?(Proc) ? r.call : r
               end
             RUBY
           end

data/lib/corn/post.rb

@@ -15,8 +15,8 @@ module Corn
     end
 
     def start_post_thread(interval)
-      if interval < 1
-        Corn.logger.info("Corn post interval < 1 sec, change it to 1 sec")
+      if interval <= 0
+        Corn.logger.info("Corn post interval <= 0 sec, change it to 1 sec")
         interval = 1
       else
         Corn.logger.info("Corn post interval #{interval} sec(s)")

data/lib/corn/rack/slow_request_profiler.rb

@@ -1,3 +1,4 @@
+require 'corn/config'
 require 'corn/post'
 require 'corn/rack/request_env'
 require 'sampling_prof'
@@ -5,21 +6,27 @@ require 'sampling_prof'
 module Corn
   module Rack
     class SlowRequestProfiler
+      include Config
+      config(:profiling => true,
+             :slow_request_threshold => 5,
+             :sampling_interval => 0.1,
+             :post_interval => 2)
+
       class ProfilingApp
-        def initialize(app,
-                       slow_request_threshold=5,
-                       sampling_interval=0.1,
-                       post_interval=2)
+        def initialize(app, config)
           @app = app
-          @slow_request_threshold = slow_request_threshold
-          @sampling_interval = sampling_interval
-          @post = Post.new(post_interval)
-          @prof = SamplingProf.new(@sampling_interval)
+          @config = config
+          @post = Post.new(config.post_interval)
+          @prof = SamplingProf.new(config.sampling_interval)
           at_exit { terminate }
         end
 
         def call(env)
-          @prof.profile(output_handler(env)) { @app.call(env) }
+          if @config.profiling?
+            @prof.profile(output_handler(env)) { @app.call(env) }
+          else
+            @app.call(env)
+          end
         end
 
         def terminate
@@ -30,15 +37,15 @@ module Corn
         def output_handler(env)
           request_env = RequestEnv.new(env)
           lambda do |data|
-            if request_env.time > @slow_request_threshold
+            if request_env.time > @config.slow_request_threshold
               @post.enqueue(request_env.to_report.merge("data" => data))
             end
           end
         end
       end
 
-      def initialize(app, *args)
-        @app = Corn.configured? ? ProfilingApp.new(app, *args) : app
+      def initialize(app)
+        @app = Corn.configured? ? ProfilingApp.new(app, self.class) : app
       end
 
       def call(env)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: corn
 version: !ruby/object:Gem::Version
-  version: 0.5.2
+  version: 0.5.3
 platform: ruby
 authors:
 - Xiao Li