|
1 | 1 | package org.elasticmq.rest.sqs.client
|
2 | 2 |
|
3 | 3 | import com.amazonaws.services.sqs.AmazonSQS
|
4 |
| -import com.amazonaws.services.sqs.model.{ |
5 |
| - BatchResultErrorEntry, |
6 |
| - CancelMessageMoveTaskRequest, |
7 |
| - ChangeMessageVisibilityBatchRequest, |
8 |
| - ChangeMessageVisibilityBatchRequestEntry, |
9 |
| - CreateQueueRequest, |
10 |
| - DeleteMessageBatchRequest, |
11 |
| - DeleteMessageBatchRequestEntry, |
12 |
| - GetQueueAttributesRequest, |
13 |
| - GetQueueUrlRequest, |
14 |
| - ListDeadLetterSourceQueuesRequest, |
15 |
| - ListMessageMoveTasksRequest, |
16 |
| - MessageAttributeValue, |
17 |
| - MessageSystemAttributeValue, |
18 |
| - PurgeQueueRequest, |
19 |
| - QueueDoesNotExistException, |
20 |
| - ReceiveMessageRequest, |
21 |
| - ResourceNotFoundException, |
22 |
| - SendMessageBatchRequest, |
23 |
| - SendMessageBatchRequestEntry, |
24 |
| - SendMessageRequest, |
25 |
| - StartMessageMoveTaskRequest, |
26 |
| - UnsupportedOperationException |
27 |
| -} |
| 4 | +import com.amazonaws.services.sqs.model.{BatchResultErrorEntry, CancelMessageMoveTaskRequest, ChangeMessageVisibilityBatchRequest, ChangeMessageVisibilityBatchRequestEntry, CreateQueueRequest, DeleteMessageBatchRequest, DeleteMessageBatchRequestEntry, GetQueueAttributesRequest, GetQueueUrlRequest, ListDeadLetterSourceQueuesRequest, ListMessageMoveTasksRequest, MessageAttributeValue, MessageSystemAttributeValue, PurgeQueueRequest, QueueDoesNotExistException, ReceiveMessageRequest, ResourceNotFoundException, SendMessageBatchRequest, SendMessageBatchRequestEntry, SendMessageRequest, StartMessageMoveTaskRequest, UnsupportedOperationException} |
28 | 5 | import org.elasticmq._
|
29 | 6 |
|
30 | 7 | import java.nio.ByteBuffer
|
@@ -70,16 +47,21 @@ class AwsSdkV1SqsClient(client: AmazonSQS) extends SqsClient {
|
70 | 47 |
|
71 | 48 | override def sendMessage(
|
72 | 49 | queueUrl: QueueUrl,
|
73 |
| - messageBody: String, |
74 |
| - messageAttributes: Map[String, MessageAttribute] = Map.empty, |
75 |
| - awsTraceHeader: Option[String] = None, |
76 |
| - messageGroupId: Option[String] = None, |
77 |
| - messageDeduplicationId: Option[String] = None |
| 50 | + messageBody: MessageMoveTaskStatus, |
| 51 | + delaySeconds: Option[Int] = None, |
| 52 | + messageAttributes: Map[ |
| 53 | + MessageMoveTaskStatus, |
| 54 | + MessageAttribute |
| 55 | + ] = Map.empty, |
| 56 | + awsTraceHeader: Option[MessageMoveTaskStatus] = None, |
| 57 | + messageGroupId: Option[MessageMoveTaskStatus] = None, |
| 58 | + messageDeduplicationId: Option[MessageMoveTaskStatus] = None |
78 | 59 | ): Either[SqsClientError, Unit] = interceptErrors {
|
79 | 60 | client.sendMessage(
|
80 | 61 | new SendMessageRequest()
|
81 | 62 | .withQueueUrl(queueUrl)
|
82 | 63 | .withMessageBody(messageBody)
|
| 64 | + .withDelaySeconds(delaySeconds.map(Int.box).orNull) |
83 | 65 | .withMessageSystemAttributes(
|
84 | 66 | mapAwsTraceHeader(awsTraceHeader)
|
85 | 67 | )
|
|
0 commit comments