corn 0.5.3 → 0.5.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a48acabdb1beaeceb6915bca571c759409f21812
-  data.tar.gz: f046e870baf89b719f4c4de7865ad8212f25e82f
+  metadata.gz: 350b0a06c165a783afcb852799e10c9b1e0c7773
+  data.tar.gz: 35cdfb9da1153beda11dd19f841e7f3b78e498d3
 SHA512:
-  metadata.gz: 70b0f65086ebb697d15394f7c85df9e2b198c1d38ca4379dd5b79ee2c72894c67243f0feea02672db994f05734e37ef9ecaf1a12f77c5b3e4366623afa2488cf
-  data.tar.gz: b8a35e2f0d5c829b51db9f0b8cdcb2b0d98f0839d988447d557dc98f440238718f6d9976ff6fe12eb4a587cbc66b41d9d29850fc02555e88dd3b240eacf4ce72
+  metadata.gz: 147290199a7b42dd8d2a73b478c5a644e115ba51b0ce03a3211b42278750bc56ff5500741a89662b185710d6eb5c2337d78b3b8b92890be20bacea15b10402e9
+  data.tar.gz: 7d8450a6461dd85f0d7ddda2ab7a4525f2876ffb3ac8c26f3884b4cb5e51f40249c0ff1c1df5dd83ef74452c2889a2d25b0aae465ddedd08b4ce03f9f1554c24

data/lib/corn/post.rb

@@ -24,7 +24,7 @@ module Corn
       Thread.start do
         begin
           loop do
-            http_post(*@queue.pop)
+            http_post(@queue.pop)
             sleep interval
           end
         rescue => e
@@ -33,8 +33,9 @@ module Corn
       end
     end
 
-    def enqueue(*args)
-      @queue << args
+    def enqueue(data)
+      return if data.nil?
+      @queue << data
     end
 
     def http_post(data)

data/lib/corn/profiler.rb

@@ -0,0 +1,21 @@
+require 'corn/post'
+require 'sampling_prof'
+
+module Corn
+  class Profiler
+    def initialize(post_interval, sampling_interval)
+      @post = Post.new(post_interval)
+      @prof = SamplingProf.new(sampling_interval)
+      at_exit { terminate }
+    end
+
+    def profile(handler, &block)
+      @prof.profile(lambda {|data| @post.enqueue(handler.call(data))}, &block)
+    end
+
+    def terminate
+      @prof.terminate rescue nil
+      @post.terminate rescue nil
+    end
+  end
+end

data/lib/corn/rack/slow_request_profiler.rb

@@ -1,7 +1,6 @@
 require 'corn/config'
-require 'corn/post'
+require 'corn/profiler'
 require 'corn/rack/request_env'
-require 'sampling_prof'
 
 module Corn
   module Rack
@@ -14,31 +13,30 @@ module Corn
 
       class ProfilingApp
         def initialize(app, config)
+          @@prof ||= Profiler.new(config.post_interval,
+                                  config.sampling_interval)
           @app = app
           @config = config
-          @post = Post.new(config.post_interval)
-          @prof = SamplingProf.new(config.sampling_interval)
-          at_exit { terminate }
         end
 
         def call(env)
           if @config.profiling?
-            @prof.profile(output_handler(env)) { @app.call(env) }
+            @@prof.profile(output_handler(env)) { @app.call(env) }
           else
             @app.call(env)
           end
         end
 
         def terminate
-          @prof.terminate rescue nil
-          @post.terminate rescue nil
+          @@prof.terminate
+          @@prof = nil
         end
 
         def output_handler(env)
           request_env = RequestEnv.new(env)
           lambda do |data|
             if request_env.time > @config.slow_request_threshold
-              @post.enqueue(request_env.to_report.merge("data" => data))
+              request_env.to_report.merge("data" => data)
             end
           end
         end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: corn
 version: !ruby/object:Gem::Version
-  version: 0.5.3
+  version: 0.5.4
 platform: ruby
 authors:
 - Xiao Li
@@ -36,6 +36,7 @@ files:
 - lib/corn.rb
 - lib/corn/config.rb
 - lib/corn/post.rb
+- lib/corn/profiler.rb
 - lib/corn/rack.rb
 - lib/corn/rack/request_env.rb
 - lib/corn/rack/slow_request_profiler.rb