Merge pull request #29 from pinterest/aws_sdk_try_catch

Catch SdkClientException during loadNewNotification to retry AWS credentials refresh errors; Bump version to 0.2.21-SNAPSHOT
pinterest · Apr 30, 2024 · b2773d5 · b2773d5
2 parents 1965675 + 99763ce
commit b2773d5
Showing 1 changed file with 2 additions and 1 deletion.
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
@@ -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<K, V> next() {
           try {
             loadNewNotification(poll);
             break;
-          } catch (IOException e) {
+          } catch (IOException | SdkClientException e) {
             if (retryCount == 1) {
               throw new RuntimeException(e);
             } else {