Skip to content

Commit 4728090

Browse files
committed
feat: optimize message request creation in SqsTemplate for batching support
1 parent cff92cb commit 4728090

File tree

1 file changed

+5
-2
lines changed
  • spring-cloud-aws-sqs/src/main/java/io/awspring/cloud/sqs/operations

1 file changed

+5
-2
lines changed

spring-cloud-aws-sqs/src/main/java/io/awspring/cloud/sqs/operations/SqsTemplate.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -605,8 +605,7 @@ private CompletableFuture<ReceiveMessageRequest> createReceiveMessageRequest(Str
605605
private ReceiveMessageRequest doCreateReceiveMessageRequest(Duration pollTimeout, Integer maxNumberOfMessages,
606606
QueueAttributes attributes, Map<String, Object> additionalHeaders) {
607607
ReceiveMessageRequest.Builder builder = ReceiveMessageRequest.builder().queueUrl(attributes.getQueueUrl())
608-
.maxNumberOfMessages(maxNumberOfMessages).messageAttributeNames(this.messageAttributeNames)
609-
.attributeNamesWithStrings(this.messageSystemAttributeNames)
608+
.maxNumberOfMessages(maxNumberOfMessages)
610609
.waitTimeSeconds(toInt(pollTimeout.toSeconds()));
611610
if (additionalHeaders.containsKey(SqsHeaders.SQS_VISIBILITY_TIMEOUT_HEADER)) {
612611
builder.visibilityTimeout(
@@ -618,6 +617,10 @@ private ReceiveMessageRequest doCreateReceiveMessageRequest(Duration pollTimeout
618617
getValueAs(additionalHeaders, SqsHeaders.SQS_RECEIVE_REQUEST_ATTEMPT_ID_HEADER, UUID.class)
619618
.toString());
620619
}
620+
if (!(this.sqsAsyncClient instanceof BatchingSqsClientAdapter)) {
621+
builder.messageAttributeNames(this.messageAttributeNames)
622+
.attributeNamesWithStrings(this.messageSystemAttributeNames);
623+
}
621624
return builder.build();
622625
}
623626

0 commit comments

Comments
 (0)