diff --git a/src/main/java/org/springframework/integration/aws/outbound/AbstractAwsMessageHandler.java b/src/main/java/org/springframework/integration/aws/outbound/AbstractAwsMessageHandler.java index 882adc5..f71c80e 100644 --- a/src/main/java/org/springframework/integration/aws/outbound/AbstractAwsMessageHandler.java +++ b/src/main/java/org/springframework/integration/aws/outbound/AbstractAwsMessageHandler.java @@ -61,6 +61,8 @@ public abstract class AbstractAwsMessageHandler extends AbstractMessageProduc private HeaderMapper headerMapper; + private boolean headerMapperSet; + public void setSendTimeout(long sendTimeout) { setSendTimeoutExpression(new ValueExpression<>(sendTimeout)); } @@ -84,6 +86,11 @@ public abstract class AbstractAwsMessageHandler extends AbstractMessageProduc */ public void setHeaderMapper(HeaderMapper headerMapper) { this.headerMapper = headerMapper; + this.headerMapperSet = true; + } + + protected boolean isHeaderMapperSet() { + return this.headerMapperSet; } /** diff --git a/src/main/java/org/springframework/integration/aws/outbound/SnsMessageHandler.java b/src/main/java/org/springframework/integration/aws/outbound/SnsMessageHandler.java index 7b09866..7880652 100644 --- a/src/main/java/org/springframework/integration/aws/outbound/SnsMessageHandler.java +++ b/src/main/java/org/springframework/integration/aws/outbound/SnsMessageHandler.java @@ -187,7 +187,9 @@ public class SnsMessageHandler extends AbstractAwsMessageHandler