vendor/rack/lib/rack/lobster.rb in relevance-castronaut-0.5.4 vs vendor/rack/lib/rack/lobster.rb in relevance-castronaut-0.6.0

@@ -20,15 +20,15 @@
       else
         lobster = LobsterString
         href = "?flip"
       end
 
-      [200, {"Content-Type" => "text/html"},
-       ["<title>Lobstericious!</title>",
-        "<pre>", lobster, "</pre>",
-        "<a href='#{href}'>flip!</a>"]
-      ]
+      content = ["<title>Lobstericious!</title>",
+                 "<pre>", lobster, "</pre>",
+                 "<a href='#{href}'>flip!</a>"]
+      length = content.inject(0) { |a,e| a+e.size }.to_s
+      [200, {"Content-Type" => "text/html", "Content-Length" => length}, content]
     }
 
     def call(env)
       req = Request.new(env)
       if req.GET["flip"] == "left"
@@ -41,17 +41,17 @@
       else
         lobster = LobsterString
         href = "?flip=left"
       end
 
-      Response.new.finish do |res|
-        res.write "<title>Lobstericious!</title>"
-        res.write "<pre>"
-        res.write lobster
-        res.write "</pre>"
-        res.write "<p><a href='#{href}'>flip!</a></p>"
-        res.write "<p><a href='?flip=crash'>crash!</a></p>"
-      end
+      res = Response.new
+      res.write "<title>Lobstericious!</title>"
+      res.write "<pre>"
+      res.write lobster
+      res.write "</pre>"
+      res.write "<p><a href='#{href}'>flip!</a></p>"
+      res.write "<p><a href='?flip=crash'>crash!</a></p>"
+      res.finish
     end
 
   end
 end