Class: Rack::ContentLength

Inherits:
Object show all
Defined in:
lib/gems/rack-0.9.1/lib/rack/content_length.rb

Overview

Sets the Content-Length header on responses with fixed-length bodies.

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ ContentLength

Returns a new instance of ContentLength.



4
5
6
 
# File 'lib/gems/rack-0.9.1/lib/rack/content_length.rb', line 4

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/gems/rack-0.9.1/lib/rack/content_length.rb', line 8

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

  if !Utils::STATUS_WITH_NO_ENTITY_BODY.include?(status) &&
     !headers['Content-Length'] &&
     !headers['Transfer-Encoding'] &&
     (body.respond_to?(:to_ary) || body.respond_to?(:to_str))

    body = [body] if body.respond_to?(:to_str) # rack 0.4 compat
    length = body.to_ary.inject(0) { |len, part| len + part.length }
    headers['Content-Length'] = length.to_s
  end

  [status, headers, body]
end