Option to disable process timeout for consumer (#19)

avast · Dec 11, 2018 · 3e7908a · 3e7908a
1 parent e6a8562
commit 3e7908a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/core/src/main/resources/reference.conf b/core/src/main/resources/reference.conf
@@ -38,7 +38,7 @@ avastRabbitMQConsumerDefaults {
 
   //    queueName = "test"
 
-  processTimeout = 10s
+  processTimeout = 10s // 0 for disable the timeout
 
   prefetchCount = 100
 

diff --git a/core/src/main/scala/com/avast/clients/rabbitmq/DefaultRabbitMQClientFactory.scala b/core/src/main/scala/com/avast/clients/rabbitmq/DefaultRabbitMQClientFactory.scala
@@ -520,8 +520,13 @@ private[rabbitmq] object DefaultRabbitMQClientFactory extends LazyLogging {
           .map(Task.fromEffect[F, DeliveryResult])
           .flatten
 
-        action
-          .timeout(ScalaDuration(processTimeout.toMillis, TimeUnit.MILLISECONDS))
+        val timedOutAction = if (processTimeout == Duration.ZERO) {
+          action
+        } else {
+          action.timeout(ScalaDuration(processTimeout.toMillis, TimeUnit.MILLISECONDS))
+        }
+
+        timedOutAction
           .onErrorRecoverWith {
             case e: TimeoutException =>
               timeoutsMeter.mark()