public async Task <IActionResult> DequeueMessageAndDelete(string queueName, [FromQuery] int?waitMs)
{
Verify.IsNotEmpty(nameof(queueName), queueName);
RequestContext requestContext = HttpContext.GetRequestContext();
var context = requestContext.Context.WithTag(_tag);
TimeSpan?waitFor = (waitMs ?? 0) != 0 ? waitFor = TimeSpan.FromMilliseconds((int)waitMs) : null;
try
{
InternalMessageV1 result = await _message.DequeueMessageAndDelete(context, queueName, waitFor);
if (result == null)
{
return(new StandardActionResult(context, HttpStatusCode.NoContent));
}
return(new StandardActionResult(context)
.SetContent(result.ConvertTo()));
}
catch (QueueNotFound ex)
{
return(new StandardActionResult(context, HttpStatusCode.NotFound)
.SetContent($"Queue not found {ex.Message}"));
}
}
public async Task SimpleScheduleSuccessTest()
{
var context = _workContext.WithTag(_tag);
int agentId = await _utility.SetupAgentAndQueue(_queueName, _agentName);
InternalEnqueueMessageV1 message = _utility.CreateMessage(context, agentId, _queueName);
message.ScheduleDate = DateTime.UtcNow.AddSeconds(2);
await _message.EnqueueMessage(context, message);
await _utility.VerifyQueue(context, _queueName, 0, 1);
InternalMessageV1 readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
readMessage.Should().BeNull();
await Task.Delay(TimeSpan.FromSeconds(5));
readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
_utility.VerifyMessage(message, readMessage, agentId, false);
InternalHistoryV1 history = await _management.GetHistory(context, readMessage.MessageId);
_utility.VerifyHistoryMessage(readMessage, history, _queueName, _agentName);
}
public async Task QueueAbandonTest()
{
var context = _workContext.WithTag(_tag);
int agentId = await _utility.SetupAgentAndQueue(_queueName, _agentName);
InternalEnqueueMessageV1 message = _utility.CreateMessage(context, agentId, _queueName);
await _message.EnqueueMessage(context, message);
await _utility.VerifyQueue(context, _queueName, 1, 0);
InternalMessageV1 readMessage = await _message.DequeueMessageWithLock(context, _queueName, agentId);
_utility.VerifyMessage(message, readMessage, agentId, true);
await _message.SettleMessage(context, readMessage.MessageId, agentId, SettleType.Abandon);
await _utility.VerifyQueue(context, _queueName, 1, 0);
readMessage = await _message.DequeueMessageWithLock(context, _queueName, agentId);
_utility.VerifyMessage(message, readMessage, agentId, true);
await _message.SettleMessage(context, readMessage.MessageId, agentId, SettleType.Processed);
await _utility.VerifyQueue(context, _queueName, 0, 0);
InternalHistoryV1 history = await _management.GetHistory(context, readMessage.MessageId);
_utility.VerifyHistoryMessage(readMessage, history, _queueName, _agentName, 2);
}
private async Task ReadQueue(IWorkContext context, CancellationToken token)
{
while (!token.IsCancellationRequested)
{
InternalMessageV1 readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
if (readMessage != null)
{
Interlocked.Increment(ref _readCount);
}
}
}
public static MessageRow ConvertTo(this InternalMessageV1 self)
{
return(new MessageRow
{
MessageId = self.MessageId,
QueueId = self.QueueId,
ClientMessageId = self.ClientMessageId,
Cv = self.Cv,
Payload = self.Payload,
LockedDate = self.LockedDate,
LockedByAgentId = self.LockedByAgentId,
RetryCount = self.RetryCount,
CreatedByAgentId = self.CreatedByAgentId,
_createdDate = self._createdDate,
});
}
public void VerifyHistoryMessage(
InternalMessageV1 readMessage,
InternalHistoryV1 history,
string queueName,
string agentName,
int retryCount = 1,
SettleType settleType = SettleType.Processed
)
{
history.Should().NotBeNull();
history.HistoryId.Should().BeGreaterOrEqualTo(0);
history.MessageId.Should().Be(readMessage.MessageId);
history.ActivityType.Should().Be(settleType.ToString());
history.QueueName.Should().Be(queueName);
history.Cv.Should().Be(readMessage.Cv);
history.ClientMessageId.Should().Be(readMessage.ClientMessageId);
history.Payload.Should().Be(readMessage.Payload);
history.SettleByAgent.Should().Be(agentName);
history.ErrorMesage.Should().BeNullOrEmpty();
history.RetryCount.Should().Be(retryCount);
}
public void VerifyMessage(InternalEnqueueMessageV1 message, InternalMessageV1 readMessage, int agentId, bool testLock)
{
readMessage.Should().NotBeNull();
readMessage.QueueId.Should().BeGreaterOrEqualTo(0);
readMessage.MessageId.Should().BeGreaterOrEqualTo(0);
readMessage.CreatedByAgentId.Should().Be(agentId);
readMessage.ClientMessageId.Should().Be(message.ClientMessageId);
readMessage.Cv.Should().Be(message.Cv);
readMessage.Payload.Should().Be(message.Payload);
if (testLock)
{
readMessage.LockedByAgentId.Should().Be(agentId);
}
WorkRequest readWorkRequest = readMessage.Deserialize <WorkRequest>();
readWorkRequest.Should().NotBeNull();
readWorkRequest.ProcessName.Should().Be("this process");
readWorkRequest.Parameters.Should().NotBeNull();
readWorkRequest.Parameters.Count().Should().Be(3);
}
public async Task SimpleEnqueueSuccessTest()
{
var context = _workContext.WithTag(_tag);
int agentId = await _utility.SetupAgentAndQueue(_queueName, _agentName);
InternalEnqueueMessageV1 message = _utility.CreateMessage(context, agentId, _queueName);
await _message.EnqueueMessage(context, message);
await _utility.VerifyQueue(context, _queueName, 1, 0);
InternalMessageV1 readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
_utility.VerifyMessage(message, readMessage, agentId, false);
await _utility.VerifyQueue(context, _queueName, 0, 0);
InternalHistoryV1 history = await _management.GetHistory(context, readMessage.MessageId);
_utility.VerifyHistoryMessage(readMessage, history, _queueName, _agentName);
}
public async Task SimpleEnqueueWithLockTimeoutSuccessTest()
{
var context = _workContext.WithTag(_tag);
int agentId = await _utility.SetupAgentAndQueue(_queueName, _agentName, lockValidForSec : 5);
InternalEnqueueMessageV1 message = _utility.CreateMessage(context, agentId, _queueName);
await _message.EnqueueMessage(context, message);
await _utility.VerifyQueue(context, _queueName, 1, 0);
InternalMessageV1 readMessage = await _message.DequeueMessageWithLock(context, _queueName, agentId);
long saveMessageId = readMessage.MessageId;
_utility.VerifyMessage(message, readMessage, agentId, true);
InternalMessageV1 tempMessage = await _message.DequeueMessageWithLock(context, _queueName, agentId);
tempMessage.Should().BeNull();
await Task.Delay(TimeSpan.FromSeconds(10));
readMessage = await _message.DequeueMessageWithLock(context, _queueName, agentId);
_utility.VerifyMessage(message, readMessage, agentId, true);
readMessage.MessageId.Should().Be(saveMessageId);
await _message.SettleMessage(context, readMessage.MessageId, agentId, SettleType.Processed);
await _utility.VerifyQueue(context, _queueName, 0, 0);
InternalHistoryV1 history = await _management.GetHistory(context, readMessage.MessageId);
_utility.VerifyHistoryMessage(readMessage, history, _queueName, _agentName, retryCount: 2);
}
public async Task MultipleScheduleSuccessTest()
{
const int scheduleSize = 10;
var context = _workContext.WithTag(_tag);
int agentId = await _utility.SetupAgentAndQueue(_queueName, _agentName);
var list = new List <InternalEnqueueMessageV1>();
foreach (var item in Enumerable.Range(0, scheduleSize))
{
InternalEnqueueMessageV1 message = _utility.CreateMessage(context, agentId, _queueName);
message.ScheduleDate = DateTime.UtcNow.AddSeconds(2);
await _message.EnqueueMessage(context, message);
list.Add(message);
}
await _utility.VerifyQueue(context, _queueName, 0, scheduleSize);
InternalMessageV1 readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
readMessage.Should().BeNull();
await Task.Delay(TimeSpan.FromSeconds(5));
var readList = new List <InternalMessageV1>();
foreach (var item in Enumerable.Range(0, scheduleSize))
{
readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
readMessage.Should().NotBeNull();
readList.Add(readMessage);
}
readList.Count.Should().Be(scheduleSize);
readMessage = await _message.DequeueMessageAndDelete(context, _queueName);
readMessage.Should().BeNull();
var zip = list
.OrderBy(x => x.ClientMessageId)
.Zip(readList.OrderBy(x => x.ClientMessageId), (f, s) => new Tuple <InternalEnqueueMessageV1, InternalMessageV1>(f, s));
foreach (var item in zip)
{
_utility.VerifyMessage(item.Item1, item.Item2, agentId, false);
}
await _utility.VerifyQueue(context, _queueName, 0, 0);
foreach (var item in readList)
{
InternalHistoryV1 history = await _management.GetHistory(context, item.MessageId);
_utility.VerifyHistoryMessage(item, history, _queueName, _agentName);
}
}
