ecin-rack-probe 0.0.2 → 0.0.3

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.0.3

data/lib/rack/probe.rb

@@ -24,15 +24,19 @@ module Rack
           p.probe :path,  :string     # Path visited 
           p.probe :referer, :string   # Referer
           p.probe :xhr                # AJAX request
+          
+          p.probe :request_start      # Start of a request
+          p.probe :request_finish     # End of a request
         end
-        
-        # Provider shortcut
-        @R = Dtrace::Probe::Rack
       end
+    
+      # Provider shortcut
+      @R = Dtrace::Probe::Rack
       @app = app
     end
 
     def call( env )
+      @R.request_start(&:fire)
       request = Rack::Request.new env
       @R.get(&:fire) if request.get?
       @R.post(&:fire) if request.post?
@@ -42,7 +46,9 @@ module Rack
       @R.path    { |p| p.fire(request.path) }
       @R.ip      { |p| p.fire(request.ip) }
       @R.referer { |p| p.fire(request.referer) }
-      @app.call env
+      response = @app.call(env)
+      @R.request_finish(&:fire)
+      response
     end
 
   end

data/rack-probe.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{rack-probe}
-  s.version = "0.0.2"
+  s.version = "0.0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["ecin"]
-  s.date = %q{2009-08-18}
+  s.date = %q{2009-08-21}
   s.description = %q{Rack::Probe provides a set of probes for Rack that fire with each request.}
   s.email = %q{ecin@copypastel.com}
   s.extra_rdoc_files = [

data/spec/rack/probe_spec.rb

@@ -24,11 +24,13 @@ describe Rack::Probe do
   end
 
   before :each do
+    @q = Proc.new {'hello'}
     @compiler = Dtrace.new
     @compiler.setopt('bufsize', '8m')
   end
 
   it 'should fire a probe with a request\'s ip' do
+    Rack::Probe.new @q
     consumer = compile_script ":::ip {trace(copyinstr((int) arg0));}"
     get '/'
     consumer.consume_once do |d|
@@ -110,5 +112,25 @@ describe Rack::Probe do
     end
     consumer.finished?.should be_true
   end
+  
+  it 'should fire a probe when a request is started' do
+    consumer = compile_script ":::request_start {}"
+    get '/'
+    consumer.consume_once do |d|
+      d.probe.name.should eql('request_start')
+      consumer.finish
+    end
+    consumer.finished?.should be_true
+  end
+  
+  it 'should fire a probe when a request finishes' do
+    consumer = compile_script ":::request_finish {}"
+    get '/'
+    consumer.consume_once do |d|
+      d.probe.name.should eql('request_finish')
+      consumer.finish
+    end
+    consumer.finished?.should be_true
+  end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ecin-rack-probe
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - ecin
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-18 00:00:00 -07:00
+date: 2009-08-21 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency