NS_IMETHODIMP
SmsIPCService::DeleteMessage(int32_t *aMessageIds, uint32_t aSize,
nsIMobileMessageCallback* aRequest)
{
DeleteMessageRequest data;
data.messageIds().AppendElements(aMessageIds, aSize);
return SendRequest(data, aRequest);
}
void SQSQueue::Delete(const Message& message)
{
if(IsInitialized())
{
AWS_LOGSTREAM_TRACE(CLASS_TAG, "Deleting message " << message.GetReceiptHandle() << ". From queue " << m_queueUrl);
DeleteMessageRequest deleteMessageRequest;
deleteMessageRequest.SetQueueUrl(m_queueUrl);
deleteMessageRequest.SetReceiptHandle(message.GetReceiptHandle());
std::shared_ptr<AsyncCallerContext> deleteMessageContext = Aws::MakeShared<QueueMessageContext>(CLASS_TAG, message);
m_client->DeleteMessageAsync(deleteMessageRequest, std::bind(&SQSQueue::OnMessageDeletedOutcomeReceived, this, std::placeholders::_1,
std::placeholders::_2, std::placeholders::_3, std::placeholders::_4), deleteMessageContext);
}
else
{
AWS_LOG_ERROR(CLASS_TAG, "Queue is not initialized, not deleting. Call EnsureQueueIsInitialized before calling this method.");
}
}
bool
SmsRequestParent::DoRequest(const DeleteMessageRequest& aRequest)
{
nsCOMPtr<nsISmsDatabaseService> smsDBService =
do_GetService(SMS_DATABASE_SERVICE_CONTRACTID);
NS_ENSURE_TRUE(smsDBService, true);
mSmsRequest = SmsRequest::Create(this);
nsCOMPtr<nsISmsRequest> forwarder = new SmsRequestForwarder(mSmsRequest);
nsresult rv = smsDBService->DeleteMessage(aRequest.messageId(), forwarder);
NS_ENSURE_SUCCESS(rv, false);
return true;
}
bool
SmsRequestParent::DoRequest(const DeleteMessageRequest& aRequest)
{
nsresult rv = NS_ERROR_FAILURE;
nsCOMPtr<nsIMobileMessageDatabaseService> dbService =
do_GetService(MOBILE_MESSAGE_DATABASE_SERVICE_CONTRACTID);
if (dbService) {
rv = dbService->DeleteMessage(aRequest.messageId(), this);
}
if (NS_FAILED(rv)) {
return NS_SUCCEEDED(NotifyDeleteMessageFailed(nsIMobileMessageCallback::INTERNAL_ERROR));
}
return true;
}
bool
SmsRequestParent::DoRequest(const DeleteMessageRequest& aRequest)
{
nsresult rv = NS_ERROR_FAILURE;
nsCOMPtr<nsIMobileMessageDatabaseService> dbService =
do_GetService(MOBILE_MESSAGE_DATABASE_SERVICE_CONTRACTID);
if (dbService) {
const InfallibleTArray<int32_t>& messageIds = aRequest.messageIds();
rv = dbService->DeleteMessage(const_cast<int32_t *>(messageIds.Elements()),
messageIds.Length(), this);
}
if (NS_FAILED(rv)) {
return NS_SUCCEEDED(NotifyDeleteMessageFailed(nsIMobileMessageCallback::INTERNAL_ERROR));
}
return true;
}
TEST_F(QueueOperationTest, TestSendReceiveDelete)
{
Aws::String queueName = BuildResourceName(BASE_SEND_RECEIVE_QUEUE_NAME);
Aws::String queueUrl = CreateDefaultQueue(queueName);
ASSERT_TRUE(queueUrl.find(queueName) != Aws::String::npos);
SendMessageRequest sendMessageRequest;
sendMessageRequest.SetMessageBody("TestMessageBody");
MessageAttributeValue stringAttributeValue;
stringAttributeValue.SetStringValue("TestString");
stringAttributeValue.SetDataType("String");
sendMessageRequest.AddMessageAttributes("TestStringAttribute", stringAttributeValue);
MessageAttributeValue binaryAttributeValue;
Aws::Utils::ByteBuffer byteBuffer(10);
for(unsigned i = 0; i < 10; ++i)
{
byteBuffer[i] = (unsigned char)i;
}
binaryAttributeValue.SetBinaryValue(byteBuffer);
binaryAttributeValue.SetDataType("Binary");
sendMessageRequest.AddMessageAttributes("TestBinaryAttribute", binaryAttributeValue);
sendMessageRequest.SetQueueUrl(queueUrl);
SendMessageOutcome sendMessageOutcome = sqsClient->SendMessage(sendMessageRequest);
ASSERT_TRUE(sendMessageOutcome.IsSuccess());
EXPECT_TRUE(sendMessageOutcome.GetResult().GetMessageId().length() > 0);
ReceiveMessageRequest receiveMessageRequest;
receiveMessageRequest.SetMaxNumberOfMessages(1);
receiveMessageRequest.SetQueueUrl(queueUrl);
receiveMessageRequest.AddMessageAttributeNames("All");
ReceiveMessageOutcome receiveMessageOutcome = sqsClient->ReceiveMessage(receiveMessageRequest);
ASSERT_TRUE(receiveMessageOutcome.IsSuccess());
ReceiveMessageResult receiveMessageResult = receiveMessageOutcome.GetResult();
ASSERT_EQ(1uL, receiveMessageResult.GetMessages().size());
EXPECT_EQ("TestMessageBody", receiveMessageResult.GetMessages()[0].GetBody());
Aws::Map<Aws::String, MessageAttributeValue> messageAttributes = receiveMessageResult.GetMessages()[0].GetMessageAttributes();
ASSERT_TRUE(messageAttributes.find("TestStringAttribute") != messageAttributes.end());
EXPECT_EQ(stringAttributeValue.GetStringValue(), messageAttributes["TestStringAttribute"].GetStringValue());
ASSERT_TRUE(messageAttributes.find("TestBinaryAttribute") != messageAttributes.end());
EXPECT_EQ(byteBuffer, messageAttributes["TestBinaryAttribute"].GetBinaryValue());
DeleteMessageRequest deleteMessageRequest;
deleteMessageRequest.SetQueueUrl(queueUrl);
deleteMessageRequest.SetReceiptHandle(receiveMessageResult.GetMessages()[0].GetReceiptHandle());
DeleteMessageOutcome deleteMessageOutcome = sqsClient->DeleteMessage(deleteMessageRequest);
ASSERT_TRUE(deleteMessageOutcome.IsSuccess());
receiveMessageOutcome = sqsClient->ReceiveMessage(receiveMessageRequest);
EXPECT_EQ(0uL, receiveMessageOutcome.GetResult().GetMessages().size());
DeleteQueueRequest deleteQueueRequest;
deleteQueueRequest.WithQueueUrl(queueUrl);
DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest);
ASSERT_TRUE(deleteQueueOutcome.IsSuccess());
}