2016-05-24 11:07:41 -04:00
|
|
|
# This Rack middleware is intended to measure the latency between
|
|
|
|
# gitlab-workhorse forwarding a request to the Rails application and the
|
|
|
|
# time this middleware is reached.
|
|
|
|
|
|
|
|
module Gitlab
|
|
|
|
module Middleware
|
2016-05-26 11:53:21 -04:00
|
|
|
class RailsQueueDuration
|
2016-05-24 11:07:41 -04:00
|
|
|
def initialize(app)
|
|
|
|
@app = app
|
|
|
|
end
|
|
|
|
|
|
|
|
def call(env)
|
2016-05-25 08:28:21 -04:00
|
|
|
trans = Gitlab::Metrics.current_transaction
|
2016-08-19 06:25:52 -04:00
|
|
|
proxy_start = env['HTTP_GITLAB_WORKHORSE_PROXY_START'].presence
|
2016-05-25 08:28:21 -04:00
|
|
|
if trans && proxy_start
|
|
|
|
# Time in milliseconds since gitlab-workhorse started the request
|
2016-05-26 11:53:21 -04:00
|
|
|
trans.set(:rails_queue_duration, Time.now.to_f * 1_000 - proxy_start.to_f / 1_000_000)
|
2016-05-24 11:07:41 -04:00
|
|
|
end
|
2016-05-25 08:28:21 -04:00
|
|
|
|
2016-05-24 11:07:41 -04:00
|
|
|
@app.call(env)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|