lib/rainbows/sendfile.rb

   1 # -*- encoding: binary -*-
   2 # This middleware handles X-\Sendfile headers generated by applications
   3 # or middlewares down the stack.  It should be placed at the top
   4 # (outermost layer) of the middleware stack to avoid having its
   5 # +to_path+ method clobbered by another middleware.
   6 #
   7 # This converts X-\Sendfile responses to bodies which respond to the
   8 # +to_path+ method which allows certain concurrency models to serve
   9 # efficiently using sendfile() or similar.  With multithreaded models
  10 # under Ruby 1.9, IO.copy_stream will be used.
  11 #
  12 # This middleware is the opposite of Rack::Sendfile as it
  13 # reverses the effect of Rack:::Sendfile.  Unlike many Ruby
  14 # web servers, some configurations of \Rainbows! are capable of
  15 # serving static files efficiently.
  16 #
  17 # === Compatibility (via IO.copy_stream in Ruby 1.9):
  18 # * ThreadSpawn
  19 # * ThreadPool
  20 # * WriterThreadPool
  21 # * WriterThreadSpawn
  22 #
  23 # === Compatibility (Ruby 1.8 and 1.9)
  24 # * EventMachine
  25 # * NeverBlock (using EventMachine)
  26 #
  27 # DO NOT use this middleware if you're proxying to \Rainbows! with a
  28 # server that understands X-\Sendfile (e.g. Apache, Lighttpd) natively.
  29 #
  30 # This does NOT understand X-Accel-Redirect headers intended for nginx.
  31 # X-Accel-Redirect requires the application to be highly coupled with
  32 # the corresponding nginx configuration, and is thus too complicated to
  33 # be worth supporting.
  34 #
  35 # Example config.ru:
  36 #
  37 #    use Rainbows::Sendfile
  38 #    run lambda { |env|
  39 #      path = "#{Dir.pwd}/random_blob"
  40 #      [ 200,
  41 #        {
  42 #          'X-Sendfile' => path,
  43 #          'Content-Type' => 'application/octet-stream'
  44 #        },
  45 #        []
  46 #      ]
  47 #    }
  48
  49 class Rainbows::Sendfile < Struct.new(:app)
  50
  51   # Body wrapper, this allows us to fall back gracefully to
  52   # +each+ in case a given concurrency model does not optimize
  53   # +to_path+ calls.
  54   class Body < Struct.new(:to_path) # :nodoc: all
  55     def self.new(path, headers)
  56       unless headers['Content-Length'.freeze]
  57         stat = File.stat(path)
  58         headers['Content-Length'.freeze] = stat.size.to_s if stat.file?
  59       end
  60       super(path)
  61     end
  62
  63     # fallback in case our +to_path+ doesn't get handled for whatever reason
  64     def each
  65       buf = ''
  66       File.open(to_path) do |fp|
  67         yield buf while fp.read(0x4000, buf)
  68       end
  69     end
  70   end
  71
  72   def call(env) # :nodoc:
  73     status, headers, body = app.call(env)
  74     headers = Rack::Utils::HeaderHash.new(headers) unless Hash === headers
  75     if path = headers.delete('X-Sendfile'.freeze)
  76       body = Body.new(path, headers) unless body.respond_to?(:to_path)
  77     end
  78     [ status, headers, body ]
  79   end
  80 end