stackflame 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a371b3ca76216b8729ee4a3486fc141f196207ae
-  data.tar.gz: 399a853e4ee66393aa5b52291c7bd0ef84d9d731
+  metadata.gz: ea1dc3c56d307e41f5c8eda061d0b18a58b08f62
+  data.tar.gz: 2c2d7b2febbb53b2d6cb57746e3eed2271141b58
 SHA512:
-  metadata.gz: a1180e73b1756377ee1c7f945fb1508beb11d3adea98c714afef251ab6d378e47b537da974b8b914d2096344e5bdba08e5abf38be7f9f39979f853ca674e9be4
-  data.tar.gz: 1a8076a8822f667b6daa6c19e6d0b9db43f01a2524575701fe91c3521248dba6a7d10bfc863e9f238677e55e53885b48cd88d23adc9d90f33730e5fa34e79819
+  metadata.gz: b2d7313472890057a3f6e7193712d102c04581d06fb3a6ae60e53902b795121d3e3960d27b4bb705ac5d9564645f7b573c086c231d3195737d02850f2a4ef654
+  data.tar.gz: aa2f2242da84638b4fe079cf1f0ffbd768674cb14a102f4a834144720c894d09733c514ee8bbcbb726119bdda547838b53348c6e80c78579f6b1de9ac4153ebe

data/lib/rack/stackflame.rb

@@ -1,16 +1,41 @@
 module Rack
   class Stackflame
-    def initialize(app, options = {})
-      @app = app
-      @options = options
+    ASSET_EXTNAMES = %w[js css png jpg jpeg].freeze
+
+    def initialize(app, options = {}, &block)
+      @app     = app
+      @options = options.dup
+
+      if block_given?
+        @block = block
+      else
+        @block = -> (env) { not_asset_request?(env) }
+      end
     end
 
     def call(env)
       result = nil
-      ::Stackflame.profile(@options) do
+
+      stackflame = ::Stackflame.new
+      stackflame.run(@options) do
         result = @app.call(env)
       end
+
+      if @block.call(env)
+        stackflame.open_flamegraph
+      end
+
       result
     end
+
+    private
+
+    def not_asset_request?(env)
+      path = env['PATH_INFO']
+      return false unless path
+
+      extname = Pathname.new(path).extname.gsub(/\A\./, '')
+      !ASSET_EXTNAMES.include?(extname)
+    end
   end
 end

data/lib/stackflame/version.rb

@@ -1,3 +1,3 @@
 class Stackflame
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stackflame
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Takashi Kokubun