Don't handle TimeoutException if processTimeout is disabled (#20)

avast · Dec 13, 2018 · f374e54 · f374e54
1 parent 3e7908a
commit f374e54
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/core/src/main/scala/com/avast/clients/rabbitmq/DefaultRabbitMQClientFactory.scala b/core/src/main/scala/com/avast/clients/rabbitmq/DefaultRabbitMQClientFactory.scala
@@ -523,16 +523,18 @@ private[rabbitmq] object DefaultRabbitMQClientFactory extends LazyLogging {
         val timedOutAction = if (processTimeout == Duration.ZERO) {
           action
         } else {
-          action.timeout(ScalaDuration(processTimeout.toMillis, TimeUnit.MILLISECONDS))
+          action
+            .timeout(ScalaDuration(processTimeout.toMillis, TimeUnit.MILLISECONDS))
+            .onErrorRecoverWith {
+              case e: TimeoutException =>
+                timeoutsMeter.mark()
+                logger.warn(s"[$name] Task timed-out, applying DeliveryResult.${consumerConfig.timeoutAction}", e)
+                Task.now(consumerConfig.timeoutAction)
+            }
         }
 
         timedOutAction
           .onErrorRecoverWith {
-            case e: TimeoutException =>
-              timeoutsMeter.mark()
-              logger.warn(s"[$name] Task timed-out, applying DeliveryResult.${consumerConfig.timeoutAction}", e)
-              Task.now(consumerConfig.timeoutAction)
-
             case NonFatal(e) =>
               fatalFailuresMeter.mark()
               logger.warn(s"[$name] Error while executing callback, applying DeliveryResult.${consumerConfig.failureAction}", e)