NIFI-11259 Used asAllowableValue() from Failure Strategy

apache · Jun 24, 2024 · b249739 · b249739
1 parent 24925ed
commit b249739
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...le/nifi-kafka-processors/src/main/java/org/apache/nifi/kafka/processors/PublishKafka.java b/...le/nifi-kafka-processors/src/main/java/org/apache/nifi/kafka/processors/PublishKafka.java
@@ -406,8 +406,8 @@ private void publishFlowFiles(final ProcessContext context, final ProcessSession
     }
 
     private void routeFailureStrategy(final ProcessContext context, final ProcessSession session, final List<FlowFile> flowFiles) {
-        final String strategy = context.getProperty(FAILURE_STRATEGY).getValue();
-        if (FailureStrategy.ROLLBACK.getValue().equals(strategy)) {
+        final FailureStrategy strategy = context.getProperty(FAILURE_STRATEGY).asAllowableValue(FailureStrategy.class);
+        if (FailureStrategy.ROLLBACK == strategy) {
             session.rollback();
             context.yield();
         } else {