lib/rainbows/fiber/body.rb

   1 # -*- encoding: binary -*-
   2 # :enddoc:
   3 # non-portable body handling for Fiber-based concurrency goes here
   4 # this module is required and included in worker processes only
   5 # this is meant to be included _after_ Rainbows::Response::Body
   6 module Rainbows::Fiber::Body # :nodoc:
   7
   8   # the sendfile 1.1.0+ gem includes IO#trysendfile
   9   if IO.method_defined?(:trysendfile)
  10     def write_body_file(body, range)
  11       sock, n, body = to_io, nil, body_to_io(body)
  12       offset, count = range ? range : [ 0, body.stat.size ]
  13       case n = sock.trysendfile(body, offset, count)
  14       when Integer
  15         offset += n
  16         return if 0 == (count -= n)
  17       when :wait_writable
  18         kgio_wait_writable
  19       else # nil
  20         return
  21       end while true
  22     ensure
  23       close_if_private(body)
  24     end
  25   end
  26
  27   def self.included(klass)
  28     klass.__send__ :alias_method, :write_body_stream, :write_body_each
  29   end
  30 end