rack.git  about / heads / tags
a modular Ruby webserver interface
blob d2bca9e5e4992eb0088769c685d9a12a5f63f4ac 867 bytes (raw)
$ git show no-unicorn:lib/rack/runtime.rb	# shows this blob on the CLI

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# frozen_string_literal: true

require 'rack/utils'

module Rack
  # Sets an "X-Runtime" response header, indicating the response
  # time of the request, in seconds
  #
  # You can put it right before the application to see the processing
  # time, or before all the other middlewares to include time for them,
  # too.
  class Runtime
    FORMAT_STRING = "%0.6f" # :nodoc:
    HEADER_NAME = "X-Runtime" # :nodoc:

    def initialize(app, name = nil)
      @app = app
      @header_name = HEADER_NAME
      @header_name += "-#{name}" if name
    end

    def call(env)
      start_time = Utils.clock_time
      status, headers, body = @app.call(env)
      request_time = Utils.clock_time - start_time

      unless headers.has_key?(@header_name)
        headers[@header_name] = FORMAT_STRING % request_time
      end

      [status, headers, body]
    end
  end
end

git clone https://yhbt.net/rack.git