Merge pull request #2 from ncareol/fix-ruby19-string-length

Content-Length: handle body w/ UTF8 characters for Ruby 1.9+ to avoid Rack::Lint::LintError
bittersweet · Feb 9, 2017 · 413b527 · 413b527
2 parents 5930446 + e05dfde
commit 413b527
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/load_speed.rb b/lib/load_speed.rb
@@ -14,7 +14,10 @@ def call(env)
         index = body.rindex("</body>")
         if index
           body.insert(index, performance_code)
-          headers["Content-Length"] = body.length.to_s
+          #
+          # handle body w/ UTF8 characters for Ruby 1.9+ to avoid Rack::Lint::LintError
+          #
+          headers["Content-Length"] = (body.respond_to?(:bytesize) ? body.bytesize : body.length).to_s
           response = [body]
         end
       end