1 files changed, 24 insertions, 0 deletions
diff --git a/lib/rainbows/timed_read.rb b/lib/rainbows/timed_read.rb
new file mode 100644
index 0000000..4a4e027
--- /dev/null
+++ b/lib/rainbows/timed_read.rb
@@ -0,0 +1,24 @@
+# -*- encoding: binary -*-
+# :enddoc:
+module Rainbows::TimedRead
+  G = Rainbows::G # :nodoc:
+
+  def wait_readable
+    IO.select([self], nil, nil, G.kato)
+  end
+
+  # used for reading headers (respecting keepalive_timeout)
+  def timed_read(buf)
+    expire = nil
+    begin
+      case rv = kgio_tryread(16384, buf)
+      when :wait_readable
+        return if expire && expire < Time.now
+        expire ||= Time.now + G.kato
+        wait_readable
+      else
+        return rv
+      end
+    end while true
+  end
+end