NIFI-13991 Fix GetAwsTextractJobStatus so that a ProvisionedThroughpu…

…tExceededException properly sends the flowfile to the "throttled" relationship. This closes #9505. Signed-off-by: Peter Turcsanyi <[email protected]>
apache · Nov 8, 2024 · 8acb704 · 8acb704
1 parent 4554aaf
commit 8acb704
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...ors/src/main/java/org/apache/nifi/processors/aws/ml/textract/GetAwsTextractJobStatus.java b/...ors/src/main/java/org/apache/nifi/processors/aws/ml/textract/GetAwsTextractJobStatus.java
@@ -36,6 +36,7 @@
 import software.amazon.awssdk.services.textract.model.GetDocumentTextDetectionRequest;
 import software.amazon.awssdk.services.textract.model.GetExpenseAnalysisRequest;
 import software.amazon.awssdk.services.textract.model.JobStatus;
+import software.amazon.awssdk.services.textract.model.ProvisionedThroughputExceededException;
 import software.amazon.awssdk.services.textract.model.TextractResponse;
 import software.amazon.awssdk.services.textract.model.ThrottlingException;
 
@@ -111,8 +112,9 @@ public void onTrigger(final ProcessContext context, final ProcessSession session
             } else {
                 throw new IllegalStateException("Unrecognized job status");
             }
-        } catch (final ThrottlingException e) {
+        } catch (final ThrottlingException | ProvisionedThroughputExceededException e) {
             getLogger().info("Request Rate Limit exceeded", e);
+            context.yield();
             session.transfer(flowFile, REL_THROTTLED);
         } catch (final Exception e) {
             getLogger().warn("Failed to get Textract Job status", e);