From 28bb59e247daa896aa2f60539bb66f4aa7caeeb9 Mon Sep 17 00:00:00 2001 From: Vladimir Kochnev Date: Mon, 29 Jun 2015 12:47:55 +0300 Subject: [PATCH] Unsubscribe notifier in ensure block In case of unhandled exception we should unsubscribe notifier anyway. Related to #7. --- lib/grape_logging/middleware/request_logger.rb | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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 {