diff --git a/memq-client/src/main/java/com/pinterest/memq/client/consumer/MemqConsumer.java b/memq-client/src/main/java/com/pinterest/memq/client/consumer/MemqConsumer.java index 50e092c..7011df9 100644 --- a/memq-client/src/main/java/com/pinterest/memq/client/consumer/MemqConsumer.java +++ b/memq-client/src/main/java/com/pinterest/memq/client/consumer/MemqConsumer.java @@ -99,6 +99,7 @@ import io.netty.buffer.ByteBufInputStream; import io.netty.buffer.ByteBufOutputStream; import io.netty.buffer.PooledByteBufAllocator; +import software.amazon.awssdk.core.exception.SdkClientException; /** * Consumer class for MemQ data. This consumer follows conventions to other @@ -787,7 +788,7 @@ public MemqLogMessage next() { try { loadNewNotification(poll); break; - } catch (IOException e) { + } catch (IOException | SdkClientException e) { if (retryCount == 1) { throw new RuntimeException(e); } else {