@@ -14,7 +14,7 @@ require 'rack/cache' Rails::Initializer.run do |config| - config.middleware.use "ResponseTimer" + config.middleware.use "ResponseTimer", "Load Time" config.middleware.use Rack::Cache, :verbose => true, config/environment.rb @@ -1,17 +1,22 @@ class ResponseTimer - def initialize(app) + def initialize(app, message = "Response Time") @app = app + @message = message end def call(env) + dup._call(env) + end + + def _call(env) @start = Time.now @status, @headers, @response = @app.call(env) @stop = Time.now - return [@status, @headers, self] + [@status, @headers, self] end def each(&block) - block.call("<!-- Response Time: #{@stop - @start} -->\n") if @headers["Content-Type"].include? "text/html" + block.call("<!-- #{@message}: #{@stop - @start} -->\n") if @headers["Content-Type"].include? "text/html" @response.each(&block) end -end \ No newline at end of file +end lib/response_timer.rb f04fdcad75e759d5049a529c60a1b165dc44df70 Anil Wadghule anildigital@gmail.com http://github.com/anildigital/planet/commit/4d321a7a30ff158cdc7c3da211a270c0c82c45a1 4d321a7a30ff158cdc7c3da211a270c0c82c45a1 2009-03-23T13:09:42-07:00 2009-03-23T13:09:42-07:00 Made it threadsafe a781bb2e7d2f27e6d1ec1c4c685879bbaf2bd8f6 Anil Wadghule anildigital@gmail.com