Avoid re-wrapping QueueExceptions

commercetools · Apr 5, 2024 · 44acbb6 · 44acbb6
1 parent 5368344
commit 44acbb6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/aws/sqs/src/main/scala/com/commercetools/queue/aws/sqs/package.scala b/aws/sqs/src/main/scala/com/commercetools/queue/aws/sqs/package.scala
@@ -32,9 +32,15 @@ package object sqs {
     new CannotPushException(queueName, makeQueueException(t, queueName))
 
   def makePullQueueException(t: Throwable, queueName: String): QueueException =
-    new CannotPullException(queueName, makeQueueException(t, queueName))
+    t match {
+      case t: QueueException => t
+      case _ => new CannotPullException(queueName, makeQueueException(t, queueName))
+    }
 
   def makeMessageException(t: Throwable, queueName: String, msgId: String, action: Action): QueueException =
-    new MessageException(msgId = msgId, action = action, inner = makeQueueException(t, queueName))
+    t match {
+      case t: QueueException => t
+      case _ => new MessageException(msgId = msgId, action = action, inner = makeQueueException(t, queueName))
+    }
 
 }
diff --git a/azure/service-bus/src/main/scala/com/commercetools/queue/azure/servicebus/package.scala b/azure/service-bus/src/main/scala/com/commercetools/queue/azure/servicebus/package.scala
@@ -33,9 +33,15 @@ package object servicebus {
     new CannotPushException(queueName, makeQueueException(t, queueName))
 
   def makePullQueueException(t: Throwable, queueName: String): QueueException =
-    new CannotPullException(queueName, makeQueueException(t, queueName))
+    t match {
+      case t: QueueException => t
+      case _ => new CannotPullException(queueName, makeQueueException(t, queueName))
+    }
 
   def makeMessageException(t: Throwable, queueName: String, msgId: String, action: Action): QueueException =
-    new MessageException(msgId = msgId, action = action, inner = makeQueueException(t, queueName))
+    t match {
+      case t: QueueException => t
+      case _ => new MessageException(msgId = msgId, action = action, inner = makeQueueException(t, queueName))
+    }
 
 }