Add SQS sample: manual acknowledge (#787)

Author: mcruzdev

spring-cloud-aws-samples/spring-cloud-aws-sqs-sample/src/main/java/io/awspring/cloud/sqs/sample/SqsManualAckSample.java

@@ -0,0 +1,77 @@
+package io.awspring.cloud.sqs.sample;
+
+import io.awspring.cloud.sqs.annotation.SqsListener;
+import io.awspring.cloud.sqs.config.SqsMessageListenerContainerFactory;
+import io.awspring.cloud.sqs.listener.acknowledgement.Acknowledgement;
+import io.awspring.cloud.sqs.listener.acknowledgement.AcknowledgementResultCallback;
+import io.awspring.cloud.sqs.listener.acknowledgement.handler.AcknowledgementMode;
+import io.awspring.cloud.sqs.operations.SqsTemplate;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.boot.ApplicationRunner;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.Message;
+import software.amazon.awssdk.services.sqs.SqsAsyncClient;
+
+import java.time.Duration;
+import java.time.OffsetDateTime;
+import java.util.Collection;
+import java.util.UUID;
+
+@Configuration
+public class SqsManualAckSample {
+
+	public static final String NEW_USER_QUEUE = "new-user-queue";
+
+	private static final Logger LOGGER = LoggerFactory.getLogger(SqsManualAckSample.class);
+
+	@Bean
+	public ApplicationRunner sendMessageToQueue(SqsTemplate sqsTemplate) {
+		LOGGER.info("Sending message");
+		return args -> sqsTemplate.send(to -> to.queue(NEW_USER_QUEUE)
+			.payload(new User(UUID.randomUUID(), "John"))
+		);
+	}
+
+	@Bean
+	public SqsTemplate sqsTemplate(SqsAsyncClient sqsAsyncClient) {
+		return SqsTemplate.builder()
+			.sqsAsyncClient(sqsAsyncClient)
+			.build();
+	}
+
+	@SqsListener(NEW_USER_QUEUE)
+	public void listen(Message<User> message) {
+		LOGGER.info("Message received on listen method at {}", OffsetDateTime.now());
+		Acknowledgement.acknowledge(message);
+	}
+
+	@Bean
+	SqsMessageListenerContainerFactory<Object> defaultSqsListenerContainerFactory(SqsAsyncClient sqsAsyncClient) {
+		return SqsMessageListenerContainerFactory
+			.builder()
+			.configure(options -> options
+				.acknowledgementMode(AcknowledgementMode.MANUAL)
+				.acknowledgementInterval(Duration.ofSeconds(3)) // NOTE: With acknowledgementInterval 3 seconds, we can batch and ack async.
+				.acknowledgementThreshold(0)
+			)
+			.acknowledgementResultCallback(new AckResultCallback())
+			.sqsAsyncClient(sqsAsyncClient)
+			.build();
+	}
+
+	public record User(UUID id, String name) {
+	}
+
+	static class AckResultCallback implements AcknowledgementResultCallback<Object> {
+		@Override
+		public void onSuccess(Collection<Message<Object>> messages) {
+			LOGGER.info("Ack with success at {}", OffsetDateTime.now());		}
+
+		@Override
+		public void onFailure(Collection<Message<Object>> messages, Throwable t) {
+			LOGGER.error("Ack with fail", t);
+		}
+	}
+}