module Airbrake::Rack

def self.capture_timing(label)
  return yield unless Airbrake::Config.instance.performance_stats
  routes = Airbrake::Rack::RequestStore[:routes]
  if !routes || routes.none?
    result = yield
  else
    timed_trace = Airbrake::TimedTrace.span(label) do
      result = yield
    end
    routes.each do |_route_path, params|
      params[:groups].merge!(timed_trace.spans)
    end
  end
  result
end