after_response 0.9.2 → 0.9.3

data/after_response.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name            = "after_response"
-  s.version         = "0.9.2"
+  s.version         = "0.9.3"
   s.platform        = Gem::Platform::RUBY
   s.summary         = "Provides hooks to execute callbacks after the response has been delivered to the client."
 
@@ -11,7 +11,7 @@ the response to the client application sooner. This would mainly include logging
 event logging service, sending email and other tasks that do not affect the response body in any way.
 EOF
 
-  s.files           = Dir['{lib/*,rails/*}'] +
+  s.files           = Dir['{lib/**/*,rails/**/*}'] +
                         %w(after_response.gemspec CHANGELOG README)
   s.require_path    = 'lib'
   s.extra_rdoc_files = ['README', 'CHANGELOG']

data/lib/after_response.rb

@@ -57,6 +57,16 @@ module AfterResponse
     @after_response_attached
   end
 
+  # If a container adapter isn't available, this method can be called to enable the buffering of events,
+  # and Starboard::EventQueue.flush! must be called manually
+  def self.buffer_and_flush_manually!
+    @after_response_attached ||= begin
+      raise "Callback hook already installed for #{current_container.name}" if current_container
+      logger.info{ "[AfterResponse] => Will flush manually" }
+      true
+    end
+  end
+
   def self.logger
     @logger ||= begin
       if defined?(Rails)

data/lib/after_response/adapters/passenger2_2.rb

@@ -0,0 +1,72 @@
+# ACK... Passenger < 3.0 sucks because request loop can't be wrapped to do what we need.
+# https://github.com/FooBarWidget/passenger/blob/release-2.2.14/lib/phusion_passenger/abstract_request_handler.rb
+# We have to overwrite the entire AbstractRequestHandler#main_loop method. Here goes.
+
+class PhusionPassenger::AbstractRequestHandler
+  # Enter the request handler's main loop.
+	def main_loop
+		reset_signal_handlers
+		begin
+			@graceful_termination_pipe = IO.pipe
+			@graceful_termination_pipe[0].close_on_exec!
+			@graceful_termination_pipe[1].close_on_exec!
+
+			@main_loop_thread_lock.synchronize do
+				@main_loop_generation += 1
+				@main_loop_running = true
+				@main_loop_thread_cond.broadcast
+			end
+
+			install_useful_signal_handlers
+
+			while true
+				@iterations += 1
+				client = accept_connection
+				if client.nil?
+					break
+				end
+				begin
+					headers, input = parse_request(client)
+					if headers
+						if headers[REQUEST_METHOD] == PING
+							process_ping(headers, input, client)
+						else
+							process_request(headers, input, client)
+						end
+					end
+				rescue IOError, SocketError, SystemCallError => e
+					print_exception("Passenger RequestHandler", e)
+				ensure
+					# 'input' is the same as 'client' so we don't
+					# need to close that.
+					# The 'close_write' here prevents forked child
+					# processes from unintentionally keeping the
+					# connection open.
+					client.close_write rescue nil
+					client.close rescue nil
+          # All this crap to add one line...
+          AfterResponse::Callbacks.perform_after_response_callbacks!
+				end
+				@processed_requests += 1
+			end
+		rescue EOFError
+			# Exit main loop.
+		rescue Interrupt
+			# Exit main loop.
+		rescue SignalException => signal
+			if signal.message != HARD_TERMINATION_SIGNAL &&
+			   signal.message != SOFT_TERMINATION_SIGNAL
+				raise
+			end
+		ensure
+			revert_signal_handlers
+			@main_loop_thread_lock.synchronize do
+				@graceful_termination_pipe[0].close rescue nil
+				@graceful_termination_pipe[1].close rescue nil
+				@main_loop_generation += 1
+				@main_loop_running = false
+				@main_loop_thread_cond.broadcast
+			end
+		end
+	end
+end

data/lib/after_response/adapters/passenger3.rb

@@ -0,0 +1,11 @@
+class PhusionPassenger::AbstractRequestHandler
+
+protected
+  def accept_and_process_next_request_with_after_response(*args)
+    returning(accept_and_process_next_request_without_after_response(*args)) do
+      AfterResponse::Callbacks.perform_after_response_callbacks!
+    end
+  end
+  alias_method_chain :accept_and_process_next_request, :after_response
+
+end

data/lib/after_response/adapters/unicorn_middleware.rb

@@ -0,0 +1,33 @@
+module AfterResponse
+
+  module Adapters
+    #
+    # Example (in config.ru):
+    #
+    #     require 'after_response/adapters/unicorn'
+    #
+    #     use AfterResponse::Adapters::UnicornMiddleware
+    class UnicornMiddleware < Struct.new(:app, :body)
+
+      def initialize(app)
+        super(app)
+      end
+
+      def call(env)
+        status, headers, self.body = app.call(env)
+        [ status, headers, self ]
+      end
+
+      def each(&block)
+        body.each(&block)
+      end
+
+      # In Unicorn, this is called _after_ the socket is closed. (Not true for at least passenger3)
+      def close
+        body.close if body.respond_to?(:close)
+        AfterResponse::Callbacks.perform_after_response_callbacks!
+      end
+
+    end
+  end
+end

data/lib/after_response/adapters/unicorn_monkeypatch.rb

@@ -0,0 +1,9 @@
+class Unicorn::HttpServer
+  alias process_client_without_after_response process_client
+  undef_method :process_client
+  def process_client(client)
+    result = process_client_without_after_response(client)
+    AfterResponse::Callbacks.perform_after_response_callbacks!
+    result
+  end
+end

data/lib/after_response/callbacks.rb

@@ -0,0 +1,80 @@
+module AfterResponse
+
+  def self.callbacks
+    Callbacks.callbacks
+  end
+
+  def self.transient_callbacks
+    Callbacks.transient_callbacks
+  end
+
+  def self.append_after_response(&block)
+    Callbacks.append_after_response(&block)
+  end
+
+  def self.append_transient_after_response(&block)
+    Callbacks.append_transient_after_response(&block)
+  end
+
+  module Callbacks
+
+    module Helpers
+      def self.included(mod)
+        mod.send(:include, TransientHelpers)
+        mod.send(:extend, ModuleHelpers)
+      end
+    end
+
+    # ApplicationController should include this
+    module TransientHelpers
+      def after_response(&block)
+        AfterResponse.append_transient_after_response(&block)
+      end
+    end
+
+    # ApplicationController should extend this
+    module ModuleHelpers
+      def after_response(&block)
+        AfterResponse.append_after_response(&block)
+      end
+    end
+
+    def self.append_after_response(&block)
+      if AfterResponse.bufferable?
+        callbacks << block
+      else
+        block.call
+      end
+    end
+
+    def self.append_transient_after_response(&block)
+      if AfterResponse.bufferable?
+        transient_callbacks << block
+      else
+        block.call
+      end
+    end
+
+    def self.callbacks
+      Thread.current[:__after_response_callbacks__] ||= []
+    end
+
+    def self.transient_callbacks
+      Thread.current[:__transient_after_response_callbacks__] ||= []
+    end
+
+    def self.all_callbacks
+      transient_callbacks + callbacks
+    end
+
+    def self.perform_after_response_callbacks!
+      all_callbacks.each do |b|
+        b.call
+      end
+    ensure
+      AfterResponse.transient_callbacks.clear
+    end
+
+  end
+
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: after_response
 version: !ruby/object:Gem::Version 
-  hash: 63
-  prerelease: false
+  hash: 61
+  prerelease: 
   segments: 
   - 0
   - 9
-  - 2
-  version: 0.9.2
+  - 3
+  version: 0.9.3
 platform: ruby
 authors: 
 - Kevin E. Hunt
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-27 00:00:00 -08:00
+date: 2011-05-02 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -34,6 +34,11 @@ extra_rdoc_files:
 - README
 - CHANGELOG
 files: 
+- lib/after_response/adapters/passenger2_2.rb
+- lib/after_response/adapters/passenger3.rb
+- lib/after_response/adapters/unicorn_middleware.rb
+- lib/after_response/adapters/unicorn_monkeypatch.rb
+- lib/after_response/callbacks.rb
 - lib/after_response.rb
 - rails/init.rb
 - after_response.gemspec
@@ -69,7 +74,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.5.2
 signing_key: 
 specification_version: 3
 summary: Provides hooks to execute callbacks after the response has been delivered to the client.