44 lines
973 B
Ruby
44 lines
973 B
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Sherlock
|
|
# Rack middleware used for tracking request metrics.
|
|
class Middleware
|
|
CONTENT_TYPES = %r{text/html|application/json}i.freeze
|
|
|
|
IGNORE_PATHS = %r{^/sherlock}.freeze
|
|
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
# env - A Hash containing Rack environment details.
|
|
def call(env)
|
|
if instrument?(env)
|
|
call_with_instrumentation(env)
|
|
else
|
|
@app.call(env)
|
|
end
|
|
end
|
|
|
|
def call_with_instrumentation(env)
|
|
trans = transaction_from_env(env)
|
|
retval = trans.run { @app.call(env) }
|
|
|
|
Sherlock.collection.add(trans)
|
|
|
|
retval
|
|
end
|
|
|
|
def instrument?(env)
|
|
!!(env['HTTP_ACCEPT'] =~ CONTENT_TYPES &&
|
|
env['REQUEST_URI'] !~ IGNORE_PATHS)
|
|
end
|
|
|
|
def transaction_from_env(env)
|
|
Transaction.new(env['REQUEST_METHOD'], env['REQUEST_URI'])
|
|
end
|
|
end
|
|
end
|
|
end
|