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); } }