updated attributes using hashmap rather than individually
Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/1211996d Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/1211996d Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/1211996d Branch: refs/heads/master Commit: 1211996d35d20934dc4fb6c533d32acc90642f96 Parents: 6138b03 Author: mans2singh <[email protected]> Authored: Tue Mar 8 21:12:49 2016 -0800 Committer: mans2singh <[email protected]> Committed: Tue Mar 8 21:12:49 2016 -0800 ---------------------------------------------------------------------- .../nifi/processors/aws/lambda/PutLambda.java | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/1211996d/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/main/java/org/apache/nifi/processors/aws/lambda/PutLambda.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/main/java/org/apache/nifi/processors/aws/lambda/PutLambda.java b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/main/java/org/apache/nifi/processors/aws/lambda/PutLambda.java index dc26f2d..6daf19b 100644 --- a/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/main/java/org/apache/nifi/processors/aws/lambda/PutLambda.java +++ b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/main/java/org/apache/nifi/processors/aws/lambda/PutLambda.java @@ -21,7 +21,9 @@ import java.nio.ByteBuffer; import java.nio.charset.Charset; import java.util.Arrays; import java.util.Collections; +import java.util.HashMap; import java.util.List; +import java.util.Map; import java.util.concurrent.TimeUnit; import org.apache.commons.lang3.StringUtils; @@ -225,14 +227,16 @@ public class PutLambda extends AbstractAWSLambdaProcessor { */ private FlowFile populateExceptionAttributes(final ProcessSession session, FlowFile flowFile, final AmazonServiceException exception) { - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_MESSAGE, exception.getErrorMessage()); - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_ERROR_CODE, exception.getErrorCode()); - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_REQUEST_ID, exception.getRequestId()); - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_STATUS_CODE, Integer.toString(exception.getStatusCode())); + Map<String,String> attributes = new HashMap<>(); + attributes.put(AWS_LAMBDA_EXCEPTION_MESSAGE, exception.getErrorMessage()); + attributes.put(AWS_LAMBDA_EXCEPTION_ERROR_CODE, exception.getErrorCode()); + attributes.put(AWS_LAMBDA_EXCEPTION_REQUEST_ID, exception.getRequestId()); + attributes.put(AWS_LAMBDA_EXCEPTION_STATUS_CODE, Integer.toString(exception.getStatusCode())); if ( exception.getCause() != null ) - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_CAUSE, exception.getCause().getMessage()); - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_ERROR_TYPE, exception.getErrorType().toString()); - flowFile = session.putAttribute(flowFile, AWS_LAMBDA_EXCEPTION_MESSAGE, exception.getErrorMessage()); + attributes.put(AWS_LAMBDA_EXCEPTION_CAUSE, exception.getCause().getMessage()); + attributes.put(AWS_LAMBDA_EXCEPTION_ERROR_TYPE, exception.getErrorType().toString()); + attributes.put(AWS_LAMBDA_EXCEPTION_MESSAGE, exception.getErrorMessage()); + flowFile = session.putAllAttributes(flowFile, attributes); return flowFile; }
