Check for around_action

vectordotdev · Feb 23, 2017 · 5479f82 · 5479f82
1 parent 5342658
commit 5479f82
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/timber/probes/action_controller_user_context.rb b/lib/timber/probes/action_controller_user_context.rb
@@ -6,7 +6,11 @@ class ActionControllerUserContext < Probe
       module AroundFilter
         def self.included(klass)
           klass.class_eval do
-            around_filter :_timber_capture_user_context
+            if klass.respond_to?(:around_action)
+              around_action :_timber_capture_user_context
+            else
+              around_filter :_timber_capture_user_context
+            end
 
             private
               def _timber_capture_user_context