When processing uploads, Unicorn may expose a StreamInput object under “rack.input” of the (future) Rack (2.x) environment.
Initializes a new StreamInput object. You normally do not have to call this unless you are writing an HTTP server.
# File lib/unicorn/stream_input.rb, line 12 def initialize(socket, request) @chunked = request.content_length.nil? @socket = socket @parser = request @buf = request.buf @rbuf = '' @bytes_read = 0 filter_body(@rbuf, @buf) unless @buf.empty? end
Executes the block for every “line'' in ios, where lines are separated by the global record separator ($/, typically “n”).
# File lib/unicorn/stream_input.rb, line 95 def each while line = gets yield line end self # Rack does not specify what the return value is here end
Reads the next “line'' from the I/O stream; lines are separated by the global record separator ($/, typically “n”). A global record separator of nil reads the entire unread contents of ios. Returns nil if called at the end of file. This takes zero arguments for strict Rack::Lint compatibility, unlike IO#gets.
# File lib/unicorn/stream_input.rb, line 73 def gets sep = $/ if sep.nil? read_all(rv = '') return rv.empty? ? nil : rv end re = /\A(.*?#{Regexp.escape(sep)})/ begin @rbuf.sub!(re, '') and return $1 return @rbuf.empty? ? nil : @rbuf.slice!(0, @rbuf.size) if eof? @socket.kgio_read(@@io_chunk_size, @buf) or eof! filter_body(once = '', @buf) @rbuf << once end while true end
Reads at most length bytes from the I/O stream, or to the end of file if length is omitted or is nil. length must be a non-negative integer or nil. If the optional buffer argument is present, it must reference a String, which will receive the data.
At end of file, it returns nil or '' depend on length. ios.read() and ios.read(nil) returns ''. ios.read(length [, buffer]) returns nil.
If the Content-Length of the HTTP request is known (as is the common case for POST requests), then ios.read(length [, buffer]) will block until the specified length is read (or it is the last chunk). Otherwise, for uncommon “Transfer-Encoding: chunked” requests, ios.read(length [, buffer]) will return immediately if there is any data and only block when nothing is available (providing IO#readpartial semantics).
# File lib/unicorn/stream_input.rb, line 41 def read(length = nil, rv = '') if length if length <= @rbuf.size length < 0 and raise ArgumentError, "negative length #{length} given" rv.replace(@rbuf.slice!(0, length)) else to_read = length - @rbuf.size rv.replace(@rbuf.slice!(0, @rbuf.size)) until to_read == 0 || eof? || (rv.size > 0 && @chunked) @socket.kgio_read(to_read, @buf) or eof! filter_body(@rbuf, @buf) rv << @rbuf to_read -= @rbuf.size end @rbuf.replace('') end rv = nil if rv.empty? && length != 0 else read_all(rv) end rv end
# File lib/unicorn/stream_input.rb, line 136 def eof! # in case client only did a premature shutdown(SHUT_WR) # we do support clients that shutdown(SHUT_WR) after the # _entire_ request has been sent, and those will not have # raised EOFError on us. if @socket @socket.shutdown @socket.close end ensure raise Unicorn::ClientShutdown, "bytes_read=#{@bytes_read}", [] end
# File lib/unicorn/stream_input.rb, line 105 def eof? if @parser.body_eof? while @chunked && ! @parser.parse once = @socket.kgio_read(@@io_chunk_size) or eof! @buf << once end @socket = nil true else false end end
# File lib/unicorn/stream_input.rb, line 118 def filter_body(dst, src) rv = @parser.filter_body(dst, src) @bytes_read += dst.size rv end
# File lib/unicorn/stream_input.rb, line 124 def read_all(dst) dst.replace(@rbuf) @socket or return until eof? @socket.kgio_read(@@io_chunk_size, @buf) or eof! filter_body(@rbuf, @buf) dst << @rbuf end ensure @rbuf.replace('') end