NIFI-1684 fixed NPE in PutKafka when retrieving key attribute bytes

Signed-off-by: joewitt <[email protected]>


Project: http://git-wip-us.apache.org/repos/asf/nifi/repo
Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/8d960f52
Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/8d960f52
Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/8d960f52

Branch: refs/heads/master
Commit: 8d960f52436a516776d6fe775be6c6f47645078b
Parents: e35c40b
Author: Oleg Zhurakousky <[email protected]>
Authored: Thu Mar 24 14:35:40 2016 -0400
Committer: joewitt <[email protected]>
Committed: Tue Mar 29 09:56:04 2016 -0400

----------------------------------------------------------------------
 .../src/main/java/org/apache/nifi/processors/kafka/PutKafka.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/nifi/blob/8d960f52/nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/main/java/org/apache/nifi/processors/kafka/PutKafka.java
----------------------------------------------------------------------
diff --git 
a/nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/main/java/org/apache/nifi/processors/kafka/PutKafka.java
 
b/nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/main/java/org/apache/nifi/processors/kafka/PutKafka.java
index 4510038..513f4f3 100644
--- 
a/nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/main/java/org/apache/nifi/processors/kafka/PutKafka.java
+++ 
b/nifi-nar-bundles/nifi-kafka-bundle/nifi-kafka-processors/src/main/java/org/apache/nifi/processors/kafka/PutKafka.java
@@ -408,7 +408,7 @@ public class PutKafka extends AbstractProcessor {
         String failedSegmentsString = 
flowFile.getAttribute(ATTR_FAILED_SEGMENTS);
         if (flowFile.getAttribute(ATTR_PROC_ID) != null && 
flowFile.getAttribute(ATTR_PROC_ID).equals(this.getIdentifier()) && 
failedSegmentsString != null) {
             topicName = flowFile.getAttribute(ATTR_TOPIC);
-            key = flowFile.getAttribute(ATTR_KEY).getBytes();
+            key = flowFile.getAttribute(ATTR_KEY) == null ? null : 
flowFile.getAttribute(ATTR_KEY).getBytes();
             delimiterPattern = flowFile.getAttribute(ATTR_DELIMITER);
         } else {
             failedSegmentsString = null;

Reply via email to