NIFI-13790: Report SEND provenance event in PublishKafka

mattyb149 · Oct 4, 2024 · 6a4af82 · 6a4af82
1 parent af31d98
commit 6a4af82
Showing 1 changed file with 3 additions and 0 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
@@ -485,6 +485,9 @@ private void routeResults(final ProcessSession session, final List<FlowFileResul
 
             final Relationship relationship = flowFileResult.getExceptions().isEmpty() ? REL_SUCCESS : REL_FAILURE;
             session.transfer(flowFile, relationship);
+            final int numTopics = flowFileResult.getSentPerTopic().keySet().size();
+            final String topicList = String.join(",", flowFileResult.getSentPerTopic().keySet().toArray(new String[numTopics]));
+            session.getProvenanceReporter().send(flowFile, "kafka://" + topicList);
         }
     }