public void When_a_group_is_prepared_and_SC_is_started_the_group_is_marked_as_failed() { var retryManager = new RetryOperationManager(); RetryOperationManager.Operations = new Dictionary <string, RetryOperation>(); using (var documentStore = InMemoryStoreBuilder.GetInMemoryStore()) { CreateAFailedMessageAndMarkAsPartOfRetryBatch(documentStore, retryManager, "Test-group", false, 1); new RetryBatches_ByStatusAndSession().Execute(documentStore); new FailedMessageRetries_ByBatch().Execute(documentStore); documentStore.WaitForIndexing(); var documentManager = new CustomRetryDocumentManager(false, documentStore) { RetryOperationManager = retryManager }; var orphanage = new FailedMessageRetries.AdoptOrphanBatchesFromPreviousSession(documentManager, null, documentStore); orphanage.AdoptOrphanedBatches(); var status = retryManager.GetStatusForRetryOperation("Test-group", RetryType.FailureGroup); Assert.True(status.Failed); } }
public void When_a_group_is_prepared_with_three_batches_and_SC_is_restarted_while_the_first_group_is_being_forwarded_then_the_count_still_matches() { var retryManager = new RetryOperationManager(); RetryOperationManager.Operations = new Dictionary <string, RetryOperation>(); using (var documentStore = InMemoryStoreBuilder.GetInMemoryStore()) { CreateAFailedMessageAndMarkAsPartOfRetryBatch(documentStore, retryManager, "Test-group", true, 2001); new RetryBatches_ByStatus_ReduceInitialBatchSize().Execute(documentStore); var testBus = new TestBus(); var sender = new TestSender(); var bodyStorage = new RavenAttachmentsBodyStorage { DocumentStore = documentStore }; var settingsHolder = new NServiceBus.Settings.SettingsHolder(); settingsHolder.Set("EndpointName", "TestEndpoint"); var configure = new Configure(settingsHolder, new TestContainer(), new List <Action <NServiceBus.ObjectBuilder.IConfigureComponents> >(), new NServiceBus.Pipeline.PipelineSettings(new BusConfiguration())); var processor = new RetryProcessor(sender, testBus, new TestReturnToSenderDequeuer(bodyStorage, sender, documentStore, testBus, configure), retryManager); documentStore.WaitForIndexing(); using (var session = documentStore.OpenSession()) { processor.ProcessBatches(session, CancellationToken.None); // mark ready session.SaveChanges(); // Simulate SC restart retryManager = new RetryOperationManager(); RetryOperationManager.Operations = new Dictionary <string, RetryOperation>(); var documentManager = new CustomRetryDocumentManager(false, documentStore) { RetryOperationManager = retryManager }; documentManager.RebuildRetryOperationState(session); processor = new RetryProcessor(sender, testBus, new TestReturnToSenderDequeuer(bodyStorage, sender, documentStore, testBus, configure), retryManager); processor.ProcessBatches(session, CancellationToken.None); session.SaveChanges(); } var status = retryManager.GetStatusForRetryOperation("Test-group", RetryType.FailureGroup); Assert.AreEqual(2001, status.TotalNumberOfMessages); } }
public void When_a_group_is_processed_it_is_set_to_the_Preparing_state() { var retryManager = new RetryOperationManager(); RetryOperationManager.Operations = new Dictionary <string, RetryOperation>(); using (var documentStore = InMemoryStoreBuilder.GetInMemoryStore()) { CreateAFailedMessageAndMarkAsPartOfRetryBatch(documentStore, retryManager, "Test-group", true, 1); var status = retryManager.GetStatusForRetryOperation("Test-group", RetryType.FailureGroup); Assert.AreEqual(RetryState.Preparing, status.RetryState); } }
public void When_a_group_has_one_batch_out_of_two_forwarded_the_status_is_Forwarding() { var retryManager = new RetryOperationManager(); RetryOperationManager.Operations = new Dictionary <string, RetryOperation>(); using (var documentStore = InMemoryStoreBuilder.GetInMemoryStore()) { CreateAFailedMessageAndMarkAsPartOfRetryBatch(documentStore, retryManager, "Test-group", true, 1001); var bodyStorage = new RavenAttachmentsBodyStorage { DocumentStore = documentStore }; var testBus = new TestBus(); var sender = new TestSender(); var settingsHolder = new NServiceBus.Settings.SettingsHolder(); settingsHolder.Set("EndpointName", "TestEndpoint"); var configure = new Configure(settingsHolder, new TestContainer(), new List <Action <NServiceBus.ObjectBuilder.IConfigureComponents> >(), new NServiceBus.Pipeline.PipelineSettings(new BusConfiguration())); var processor = new RetryProcessor(sender, testBus, new TestReturnToSenderDequeuer(bodyStorage, sender, documentStore, testBus, configure), retryManager); documentStore.WaitForIndexing(); using (var session = documentStore.OpenSession()) { processor.ProcessBatches(session, CancellationToken.None); // mark ready session.SaveChanges(); processor.ProcessBatches(session, CancellationToken.None); session.SaveChanges(); } var status = retryManager.GetStatusForRetryOperation("Test-group", RetryType.FailureGroup); Assert.AreEqual(RetryState.Forwarding, status.RetryState); } }
public GroupFetcher(RetryOperationManager retryOperationManager) { this.retryOperationManager = retryOperationManager; }
void CreateAFailedMessageAndMarkAsPartOfRetryBatch(IDocumentStore documentStore, RetryOperationManager retryManager, string groupId, bool progressToStaged, int numberOfMessages) { var messages = Enumerable.Range(0, numberOfMessages).Select(i => { var id = Guid.NewGuid().ToString(); return(new FailedMessage { Id = FailedMessage.MakeDocumentId(id), UniqueMessageId = id, FailureGroups = new List <FailedMessage.FailureGroup> { new FailedMessage.FailureGroup { Id = groupId, Title = groupId, Type = groupId } }, Status = FailedMessageStatus.Unresolved, ProcessingAttempts = new List <FailedMessage.ProcessingAttempt> { new FailedMessage.ProcessingAttempt { AttemptedAt = DateTime.UtcNow, MessageMetadata = new Dictionary <string, object>(), FailureDetails = new FailureDetails(), Headers = new Dictionary <string, string>() } } }); }); using (var session = documentStore.OpenSession()) { foreach (var message in messages) { session.Store(message); } session.SaveChanges(); } new FailedMessages_ByGroup().Execute(documentStore); documentStore.WaitForIndexing(); var documentManager = new CustomRetryDocumentManager(progressToStaged, documentStore); var gateway = new RetriesGateway(documentStore, documentManager); documentManager.RetryOperationManager = retryManager; gateway.RetryOperationManager = retryManager; gateway.StartRetryForIndex <FailureGroupMessageView, FailedMessages_ByGroup>("Test-group", RetryType.FailureGroup, DateTime.UtcNow, x => x.FailureGroupId == "Test-group", "Test-Context"); documentStore.WaitForIndexing(); gateway.ProcessNextBulkRetry(); }