rack/middleware: fix performance stat sending

CHANGELOG.md

@@ -3,6 +3,9 @@ Airbrake Changelog
 
 ### master
 
+* Fixed performance stats not being sent
+  ([#901](https://github.com/airbrake/airbrake/pull/901))
+
 ### [v8.1.1][v8.1.1] (February 14, 2019)
 
 * Fixed `uninitialized constant Airbrake::Rack::Middleware::ActiveRecord` for

lib/airbrake/rack/middleware.rb

@@ -34,6 +34,10 @@ def initialize(app, notifier_name = :default)
         @notice_notifier = Airbrake.notifiers[:notice][notifier_name]
         @performance_notifier = Airbrake.notifiers[:performance][notifier_name]
 
+        # This object is shared among hooks. ActionControllerRouteSubscriber
+        # writes it and other hooks read it.
+        @routes = {}
+
         # Prevent adding same filters to the same notifier.
         return if @@known_notifiers.include?(notifier_name)
         @@known_notifiers << notifier_name
@@ -57,6 +61,9 @@ def call(env)
         rescue Exception => ex
           notify_airbrake(ex, env)
           raise ex
+        ensure
+          # Clear routes for the next request.
+          @routes.clear
         end
         # rubocop:enable Lint/RescueException
 
@@ -101,13 +108,13 @@ def framework_exception(env)
       def install_action_controller_hooks
         ActiveSupport::Notifications.subscribe(
           'start_processing.action_controller',
-          Airbrake::Rails::ActionControllerRouteSubscriber.new
+          Airbrake::Rails::ActionControllerRouteSubscriber.new(@routes)
         )
 
         ActiveSupport::Notifications.subscribe(
           'process_action.action_controller',
           Airbrake::Rails::ActionControllerNotifySubscriber.new(
-            @performance_notifier
+            @performance_notifier, @routes
           )
         )
       end
@@ -120,7 +127,9 @@ def install_active_record_hooks
         )
         ActiveSupport::Notifications.subscribe(
           'sql.active_record',
-          Airbrake::Rails::ActiveRecordSubscriber.new(@performance_notifier)
+          Airbrake::Rails::ActiveRecordSubscriber.new(
+            @performance_notifier, @routes
+          )
         )
       end
     end

lib/airbrake/rails/action_controller_notify_subscriber.rb

@@ -5,25 +5,28 @@ module Rails
     #
     # @since v8.0.0
     class ActionControllerNotifySubscriber
-      def initialize(notifier)
+      def initialize(notifier, routes)
         @notifier = notifier
+        @routes = routes
       end
 
       def call(*args)
-        return unless (route = Thread.current[:airbrake_rails_route])
+        return if @routes.none?
 
         event = ActiveSupport::Notifications::Event.new(*args)
         payload = event.payload
 
-        @notifier.notify(
-          Airbrake::Request.new(
-            method: payload[:method],
-            route: route,
-            status_code: find_status_code(payload),
-            start_time: event.time,
-            end_time: Time.new
+        @routes.each do |route, method|
+          @notifier.notify(
+            Airbrake::Request.new(
+              method: method,
+              route: route,
+              status_code: find_status_code(payload),
+              start_time: event.time,
+              end_time: Time.new
+            )
           )
-        )
+        end
       end
 
       private

lib/airbrake/rails/action_controller_route_subscriber.rb

@@ -5,7 +5,8 @@ module Rails
     #
     # @since v8.0.0
     class ActionControllerRouteSubscriber
-      def initialize
+      def initialize(routes)
+        @routes = routes
         @all_routes = nil
       end
 
@@ -17,8 +18,7 @@ def call(*args)
         event = ActiveSupport::Notifications::Event.new(*args)
         payload = event.payload
 
-        Thread.current[:airbrake_rails_route] = find_route(payload[:params])
-        Thread.current[:airbrake_rails_method] = payload[:method]
+        @routes[find_route(payload[:params])] = payload[:method]
       end
 
       private

lib/airbrake/rails/active_record_subscriber.rb

@@ -4,21 +4,24 @@ module Rails
     #
     # @since v8.1.0
     class ActiveRecordSubscriber
-      def initialize(notifier)
+      def initialize(notifier, routes)
         @notifier = notifier
+        @routes = routes
       end
 
       def call(*args)
         event = ActiveSupport::Notifications::Event.new(*args)
-        @notifier.notify(
-          Airbrake::Query.new(
-            route: Thread.current[:airbrake_rails_route],
-            method: Thread.current[:airbrake_rails_method],
-            query: event.payload[:sql],
-            start_time: event.time,
-            end_time: event.end
+        @routes.each do |route, method|
+          @notifier.notify(
+            Airbrake::Query.new(
+              route: route,
+              method: method,
+              query: event.payload[:sql],
+              start_time: event.time,
+              end_time: event.end
+            )
           )
-        )
+        end
       end
     end
   end