Class: Mack::Utils::ContentLengthHandler

Inherits:
Object
  • Object
show all
Defined in:
lib/mack/utils/content_length_handler.rb

Overview

The whole purpose of this handler is to intercept the rack application process, calculate the size of the response body, and set the ‘Content-Length’ header if necessary.

From: thin.lighthouseapp.com/projects/7212/tickets/74 Content-Length should not be automatically added:

  • When the Status code is 1xx, 204 or 304

  • When the Transfer-Encoding header is set to chunked

  • When the body is neither a String or an Array

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ ContentLengthHandler

:nodoc:



16
17
18
 
# File 'lib/mack/utils/content_length_handler.rb', line 16

def initialize(app) # :nodoc:
  @app = app
end

Instance Attribute Details

#response ⇒ Object (readonly)

Returns the value of attribute response.



14
15
16
 
# File 'lib/mack/utils/content_length_handler.rb', line 14

def response
  @response
end

Instance Method Details

#call(env) ⇒ Object

:nodoc:



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/mack/utils/content_length_handler.rb', line 20

def call(env) # :nodoc:
  ret = @app.call(env)
          
  status        = ret[0]
  hdr           = ret[1]
  @response     = ret[2]
     
  unless self.response.is_a?(Rack::File) or 
         (!self.response.body.kind_of?Array and !self.response.body.kind_of?String) or 
         hdr["Transfer-Encoding"] == "chunked" or
         (status == 204 or status == 304 or (status >= 100 and status < 200))
    size = 0
    
    if self.response.body.respond_to?(:to_str)
      size = self.response.body.to_str.size
    elsif self.response.body.respond_to?(:each)
      self.response.body.each { |part| size += part.to_s.size }
    end
    
    hdr["Content-Length"] = size.to_s
  end
  return ret
end