Add location to RenderSubscriber

To know where from which controller action
the template was rendered.

Author: ChrisBr

lib/influxdb/rails/middleware/render_subscriber.rb

@@ -30,12 +30,20 @@ def call(_name, started, finished, _unique_id, payload)
 
         private
 
+        def location
+          [
+            Thread.current[:_influxdb_rails_controller],
+            Thread.current[:_influxdb_rails_action],
+          ].reject(&:blank?).join("#")
+        end
+
         def tags(payload)
           {
+            location:   location,
             filename:   payload[:identifier],
             count:      payload[:count],
             cache_hits: payload[:cache_hits],
-          }.reject { |_, value| value.nil? }
+          }.reject { |_, value| value.blank? }
         end
       end
     end

lib/influxdb/rails/middleware/request_subscriber.rb

@@ -24,6 +24,9 @@ def call(_name, start, finish, _id, payload)
             end
           rescue StandardError => e
             log :error, "[InfluxDB::Rails] Unable to write points: #{e.message}"
+          ensure
+            Thread.current[:_influxdb_rails_controller] = nil
+            Thread.current[:_influxdb_rails_action]     = nil
           end
         end
 

lib/influxdb/rails/railtie.rb

@@ -22,6 +22,11 @@ class Railtie < ::Rails::Railtie # :nodoc:
         ::ActionDispatch::DebugExceptions.prepend InfluxDB::Rails::Middleware::HijackRenderException
 
         if defined?(ActiveSupport::Notifications)
+          ActiveSupport::Notifications.subscribe "start_processing.action_controller" do |name, start, finish, id, payload|
+            Thread.current[:_influxdb_rails_controller] = payload[:controller]
+            Thread.current[:_influxdb_rails_action]     = payload[:action]
+          end
+
           config = InfluxDB::Rails.configuration
           request_subsriber = Middleware::RequestSubscriber.new(config)
           ActiveSupport::Notifications.subscribe "process_action.action_controller", request_subsriber

spec/unit/middleware/render_subscriber_spec.rb

@@ -24,6 +24,7 @@
         tags:
         {
           filename: "index.html",
+          location: "Foo#bar",
           count: 43,
           cache_hits: 42
         },
@@ -33,6 +34,16 @@
 
     subject { described_class.new(config, series_name) }
 
+    before do
+      Thread.current[:_influxdb_rails_controller] = "Foo"
+      Thread.current[:_influxdb_rails_action]     = "bar"
+    end
+
+    after do
+      Thread.current[:_influxdb_rails_action]     = nil
+      Thread.current[:_influxdb_rails_controller] = nil
+    end
+
     context 'successfully' do
       it "writes to InfluxDB" do
         expect_any_instance_of(InfluxDB::Client).to receive(:write_point).with(