fix(sqs): SQS Consume should delete the message after use (#156)

kestra-io · May 22, 2023 · 32c42ef · 32c42ef
1 parent 8eda798
commit 32c42ef
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/java/io/kestra/plugin/aws/sqs/Consume.java b/src/main/java/io/kestra/plugin/aws/sqs/Consume.java
@@ -10,6 +10,7 @@
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.*;
 import lombok.experimental.SuperBuilder;
+import software.amazon.awssdk.services.sqs.model.DeleteMessageRequest;
 import software.amazon.awssdk.services.sqs.model.ReceiveMessageRequest;
 
 import java.io.BufferedOutputStream;
@@ -69,6 +70,10 @@ public Output run(RunContext runContext) throws Exception {
                     var msg = sqsClient.receiveMessage(receiveRequest);
                     msg.messages().forEach(throwConsumer(m -> {
                         FileSerde.write(outputFile, m.body());
+                        sqsClient.deleteMessage(DeleteMessageRequest.builder()
+                            .queueUrl(getQueueUrl())
+                            .receiptHandle(m.receiptHandle()).build()
+                        );
                         total.getAndIncrement();
                     }));