Merge pull request #123 from BobDu/fix-validation-exception

fix: no retry ValidationException
awslabs · Jan 9, 2024 · 8b58dd6 · 8b58dd6
2 parents 33430c2 + b937507
commit 8b58dd6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...n/java/com/amazonaws/samples/connectors/timestream/DefaultWriteRequestFailureHandler.java b/...n/java/com/amazonaws/samples/connectors/timestream/DefaultWriteRequestFailureHandler.java
@@ -199,15 +199,15 @@ public void onWriteError(final List<Record> requestEntries,
                              final Consumer<List<Record>> dropCompletionConsumer) {
         final Class<? extends Exception> exceptionClass = exception.getClass();
         LOG.debug("Sending WriteRecordsRequest failed. Starting handling exception: {}", exceptionClass.getName());
-        if (checkIsRetryableException(exception)) {
-            handleRetryableException(requestEntries, writeRecordsRequest, exception,
-                    retryOrSuccessCompletionConsumer, dropCompletionConsumer);
-        } else if (exceptionTypeToExceptionHandleMethod.containsKey(exceptionClass)) {
+        if (exceptionTypeToExceptionHandleMethod.containsKey(exceptionClass)) {
             LOG.debug("Found designated exception handler method.");
             exceptionTypeToExceptionHandleMethod
                     .get(exceptionClass)
                     .accept(requestEntries, writeRecordsRequest, exception,
                             retryOrSuccessCompletionConsumer, dropCompletionConsumer);
+        } else if (checkIsRetryableException(exception)) {
+            handleRetryableException(requestEntries, writeRecordsRequest, exception,
+                    retryOrSuccessCompletionConsumer, dropCompletionConsumer);
         } else {
             LOG.debug("No designated exception handler method found. Launching the default handler.");
             handleDefaultException(requestEntries, writeRecordsRequest, exception,