lib/rorvswild.rb in rorvswild-0.0.9 vs lib/rorvswild.rb in rorvswild-0.1.0

@@ -55,18 +55,18 @@
       setup_callbacks
       RorVsWild.register_default_client(self)
     end
 
     def setup_callbacks
+      client = self
       ActiveSupport::Notifications.subscribe("sql.active_record", &method(:after_sql_query))
       ActiveSupport::Notifications.subscribe("render_template.action_view", &method(:after_view_rendering))
       ActiveSupport::Notifications.subscribe("process_action.action_controller", &method(:after_http_request))
       ActiveSupport::Notifications.subscribe("start_processing.action_controller", &method(:before_http_request))
-
-      client = self
       ActionController::Base.rescue_from(StandardError) { |exception| client.after_exception(exception, self) }
 
+      Resque::Job.send(:extend, ResquePlugin) if defined?(Resque::Job)
       Delayed::Worker.lifecycle.around(:invoke_job, &method(:around_delayed_job)) if defined?(Delayed::Worker)
     end
 
     def before_http_request(name, start, finish, id, payload)
       @request = {controller: payload[:controller], action: payload[:action], path: payload[:path]}
@@ -287,8 +287,14 @@
     end
 
     def log_error(exception)
       logger.error("[RorVsWild] " + exception.inspect)
       logger.error("[RorVsWild] " + exception.backtrace.join("\n[RorVsWild] "))
+    end
+  end
+
+  module ResquePlugin
+    def around_perform_rorvswild(*args, &block)
+      RorVsWild.measure_block(to_s + "#perform", &block)
     end
   end
 end