diff --git a/lib/grape_logging/middleware/request_logger.rb b/lib/grape_logging/middleware/request_logger.rb index 92a7517..e5491ef 100644 --- a/lib/grape_logging/middleware/request_logger.rb +++ b/lib/grape_logging/middleware/request_logger.rb @@ -16,10 +16,15 @@ module GrapeLogging def after stop_time logger.info parameters - ActiveSupport::Notifications.unsubscribe(@subscription) if @subscription nil end + def call!(env) + super + ensure + ActiveSupport::Notifications.unsubscribe(@subscription) if @subscription + end + protected def parameters {