Class: Rack::Chunked

Inherits:
Object
  • Object
show all
Includes:
Utils
Defined in:
lib/rack/chunked.rb

Overview

Middleware that applies chunked transfer encoding to response bodies when the response does not include a Content-Length header.

Constant Summary collapse

TERM = 
"\r\n"
TAIL = 
"0#{TERM}#{TERM}"

Constants included from Utils

Utils::DEFAULT_SEP, Utils::ESCAPE_HTML, Utils::ESCAPE_HTML_PATTERN, Utils::HTTP_STATUS_CODES, Utils::STATUS_WITH_NO_ENTITY_BODY, Utils::SYMBOL_TO_STATUS_CODE

Instance Method Summary collapse

Methods included from Utils

build_nested_query, build_query, byte_ranges, bytesize, delete_cookie_header!, escape, escape_html, normalize_params, parse_nested_query, parse_query, rfc2822, select_best_encoding, set_cookie_header!, status_code, unescape

Constructor Details

#initialize(app) ⇒ Chunked

Returns a new instance of Chunked.



13
14
15
 
# File 'lib/rack/chunked.rb', line 13

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/rack/chunked.rb', line 17

def call(env)
  status, headers, body = @app.call(env)
  headers = HeaderHash.new(headers)

  if env['HTTP_VERSION'] == 'HTTP/1.0' ||
     STATUS_WITH_NO_ENTITY_BODY.include?(status) ||
     headers['Content-Length'] ||
     headers['Transfer-Encoding']
    [status, headers, body]
  else
    dup.chunk(status, headers, body)
  end
end

#chunk(status, headers, body) ⇒ Object 



31
32
33
34
35
36
 
# File 'lib/rack/chunked.rb', line 31

def chunk(status, headers, body)
  @body = body
  headers.delete('Content-Length')
  headers['Transfer-Encoding'] = 'chunked'
  [status, headers, self]
end

#close ⇒ Object 



48
49
50
 
# File 'lib/rack/chunked.rb', line 48

def close
  @body.close if @body.respond_to?(:close)
end

#each {|TAIL| ... } ⇒ Object

Yields:



38
39
40
41
42
43
44
45
46
 
# File 'lib/rack/chunked.rb', line 38

def each
  term = TERM
  @body.each do |chunk|
    size = bytesize(chunk)
    next if size == 0
    yield [size.to_s(16), term, chunk, term].join
  end
  yield TAIL
end