NIFI-1124, NIFI-1062 addressed PR comments Signed-off-by: Bryan Bende <bbe...@apache.org>
Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/70cc61bb Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/70cc61bb Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/70cc61bb Branch: refs/heads/master Commit: 70cc61bb53d0fc46468f706465f34790d714fff2 Parents: 62e3cfc Author: Oleg Zhurakousky <o...@suitcase.io> Authored: Tue Nov 10 19:57:01 2015 -0500 Committer: Bryan Bende <bbe...@apache.org> Committed: Wed Nov 11 10:12:30 2015 -0500 ---------------------------------------------------------------------- .../src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java | 2 +- .../test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/70cc61bb/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java b/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java index 4b929bd..f7f4d03 100644 --- a/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java +++ b/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/main/java/org/apache/nifi/processors/hadoop/PutHDFS.java @@ -345,7 +345,7 @@ public class PutHDFS extends AbstractHadoopProcessor { } } getLogger().error("Failed to write to HDFS due to {}", t); - session.transfer(flowFile, REL_FAILURE); + session.transfer(session.penalize(flowFile), REL_FAILURE); context.yield(); } } http://git-wip-us.apache.org/repos/asf/nifi/blob/70cc61bb/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java b/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java index 2eff5c3..f5fe544 100644 --- a/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java +++ b/nifi-nar-bundles/nifi-hadoop-bundle/nifi-hdfs-processors/src/test/java/org/apache/nifi/processors/hadoop/PutHDFSTest.java @@ -208,6 +208,7 @@ public class PutHDFSTest { List<MockFlowFile> failedFlowFiles = runner .getFlowFilesForRelationship(new Relationship.Builder().name("failure").build()); assertFalse(failedFlowFiles.isEmpty()); + assertTrue(failedFlowFiles.get(0).isPenalized()); fs.setPermission(p, new FsPermission(FsAction.EXECUTE, FsAction.EXECUTE, FsAction.EXECUTE)); fs.delete(p, true);
