From 5451f0c20d2382ac126e923179c096357d20cf0a Mon Sep 17 00:00:00 2001
From: Eric Wong <normalperson@yhbt.net>
Date: Mon, 19 Jul 2010 10:09:58 +0000
Subject: event_machine: split out uncommonly used modules

Some applications may not use Response*Pipe and TryDefer at all,
so there's no reason to pollute the runtime with extra nodes to
mark during GC.
---
 lib/rainbows/event_machine/response_chunk_pipe.rb | 25 ++++++++++++++++++++
 lib/rainbows/event_machine/response_pipe.rb       | 28 +++++++++++++++++++++++
 lib/rainbows/event_machine/try_defer.rb           | 27 ++++++++++++++++++++++
 3 files changed, 80 insertions(+)
 create mode 100644 lib/rainbows/event_machine/response_chunk_pipe.rb
 create mode 100644 lib/rainbows/event_machine/response_pipe.rb
 create mode 100644 lib/rainbows/event_machine/try_defer.rb

(limited to 'lib/rainbows/event_machine')

diff --git a/lib/rainbows/event_machine/response_chunk_pipe.rb b/lib/rainbows/event_machine/response_chunk_pipe.rb
new file mode 100644
index 0000000..ee8d260
--- /dev/null
+++ b/lib/rainbows/event_machine/response_chunk_pipe.rb
@@ -0,0 +1,25 @@
+# -*- encoding: binary -*-
+# :enddoc:
+module Rainbows::EventMachine::ResponseChunkPipe
+  include Rainbows::EventMachine::ResponsePipe
+
+  def unbind
+    @client.write("0\r\n\r\n")
+    super
+  end
+
+  def notify_readable
+    begin
+      data = @io.read_nonblock(16384, BUF)
+      @client.write(sprintf("%x\r\n", data.size))
+      @client.write(data)
+      @client.write("\r\n")
+    rescue Errno::EINTR
+    rescue Errno::EAGAIN
+      return
+    rescue EOFError
+      detach
+      return
+    end while true
+  end
+end
diff --git a/lib/rainbows/event_machine/response_pipe.rb b/lib/rainbows/event_machine/response_pipe.rb
new file mode 100644
index 0000000..88d6e5a
--- /dev/null
+++ b/lib/rainbows/event_machine/response_pipe.rb
@@ -0,0 +1,28 @@
+# -*- encoding: binary -*-
+# :enddoc:
+module Rainbows::EventMachine::ResponsePipe
+  # garbage avoidance, EM always uses this in a single thread,
+  # so a single buffer for all clients will work safely
+  BUF = ''
+
+  def initialize(client, alive)
+    @client, @alive = client, alive
+  end
+
+  def notify_readable
+    begin
+      @client.write(@io.read_nonblock(16384, BUF))
+    rescue Errno::EINTR
+    rescue Errno::EAGAIN
+      return
+    rescue EOFError
+      detach
+      return
+    end while true
+  end
+
+  def unbind
+    @client.quit unless @alive
+    @io.close
+  end
+end
diff --git a/lib/rainbows/event_machine/try_defer.rb b/lib/rainbows/event_machine/try_defer.rb
new file mode 100644
index 0000000..97e384d
--- /dev/null
+++ b/lib/rainbows/event_machine/try_defer.rb
@@ -0,0 +1,27 @@
+# -*- encoding: binary -*-
+# :enddoc:
+
+# Middleware that will run the app dispatch in a separate thread.
+# This middleware is automatically loaded by Rainbows! when using
+# EventMachine and if the app responds to the +deferred?+ method.
+class Rainbows::EventMachine::TryDefer < Struct.new(:app)
+  # shortcuts
+  ASYNC_CALLBACK = Rainbows::EvCore::ASYNC_CALLBACK
+
+  def initialize(app)
+    # the entire app becomes multithreaded, even the root (non-deferred)
+    # thread since any thread can share processes with others
+    Rainbows::Const::RACK_DEFAULTS['rack.multithread'] = true
+    super
+  end
+
+  def call(env)
+    if app.deferred?(env)
+      EM.defer(proc { catch(:async) { app.call(env) } }, env[ASYNC_CALLBACK])
+      # all of the async/deferred stuff breaks Rack::Lint :<
+      nil
+    else
+      app.call(env)
+    end
+  end
+end
-- 
cgit v1.2.3-24-ge0c7