RavenPersistence.cs

namespace ServiceControl.Persistence.RavenDB;

using CustomChecks;
using MessageFailures;
using MessageRedirects;
using Microsoft.Extensions.DependencyInjection;
using NServiceBus.Unicast.Subscriptions.MessageDrivenSubscriptions;
using Operations.BodyStorage;
using Operations.BodyStorage.RavenAttachments;
using Persistence.MessageRedirects;
using Persistence.Recoverability;
using Recoverability;
using ServiceControl.CustomChecks;
using ServiceControl.Infrastructure.RavenDB.Subscriptions;
using ServiceControl.Recoverability;
using Throughput;
using UnitOfWork;
using IPersistence = IPersistence;
using PersistenceSettings = PersistenceSettings;

class RavenPersistence(RavenPersisterSettings settings) : IPersistence
{
    public void AddPersistence(IServiceCollection services)
    {
        ConfigureLifecycle(services);

        if (settings.MaintenanceMode)
        {
            return;
        }

        services.AddSingleton<IServiceControlSubscriptionStorage, RavenSubscriptionStorage>();
        services.AddSingleton<ISubscriptionStorage>(p => p.GetRequiredService<IServiceControlSubscriptionStorage>());

        services.AddUnitOfWorkFactory<RavenIngestionUnitOfWorkFactory>();
        services.AddSingleton<ExpirationManager>();
        services.AddSingleton<MinimumRequiredStorageState>();
        services.AddSingleton<IBodyStorage, RavenAttachmentsBodyStorage>();

        services.AddSingleton<FailedMessageViewIndexNotifications>();
        services.AddSingleton<IFailedMessageViewIndexNotifications>(p => p.GetRequiredService<FailedMessageViewIndexNotifications>());
        services.AddHostedService(p => p.GetRequiredService<FailedMessageViewIndexNotifications>());

        services.AddSingleton<ExternalIntegrationRequestsDataStore>();
        services.AddSingleton<IExternalIntegrationRequestsDataStore>(p => p.GetRequiredService<ExternalIntegrationRequestsDataStore>());
        services.AddHostedService(p => p.GetRequiredService<ExternalIntegrationRequestsDataStore>());

        services.AddCustomCheck<CheckRavenDBIndexErrors>();
        services.AddCustomCheck<CheckRavenDBIndexLag>();
        services.AddCustomCheck<CheckFreeDiskSpace>();
        services.AddCustomCheck<CheckMinimumStorageRequiredForIngestion>();
        services.AddCustomCheck<CheckDirtyMemory>();

        services.AddSingleton<MemoryInformationRetriever>();
        services.AddSingleton<OperationsManager>();

        services.AddSingleton<IArchiveMessages, MessageArchiver>();
        services.AddSingleton<ICustomChecksDataStore, RavenCustomCheckDataStore>();
        services.AddSingleton<IErrorMessageDataStore, ErrorMessagesDataStore>();
        services.AddSingleton<IEventLogDataStore, EventLogDataStore>();
        services.AddSingleton<IFailedErrorImportDataStore, FailedErrorImportDataStore>();
        services.AddSingleton<IGroupsDataStore, GroupsDataStore>();
        services.AddSingleton<IMessageRedirectsDataStore, MessageRedirectsDataStore>();
        services.AddSingleton<IMonitoringDataStore, RavenMonitoringDataStore>();
        services.AddSingleton<IQueueAddressStore, QueueAddressStore>();
        services.AddSingleton<IRetryBatchesDataStore, RetryBatchesDataStore>();
        services.AddSingleton<IRetryDocumentDataStore, RetryDocumentDataStore>();
        services.AddSingleton<IRetryHistoryDataStore, RetryHistoryDataStore>();
        services.AddSingleton<IEndpointSettingsStore, EndpointSettingsStore>();
        services.AddSingleton<ITrialLicenseDataProvider, TrialLicenseDataProvider>();
    }

    public void AddInstaller(IServiceCollection services)
    {
        ConfigureLifecycle(services);
    }

    void ConfigureLifecycle(IServiceCollection services)
    {
        services.AddSingleton<PersistenceSettings>(settings);
        services.AddSingleton(settings);

        services.AddSingleton<IRavenSessionProvider, RavenSessionProvider>();
        services.AddHostedService<RavenPersistenceLifecycleHostedService>();

        services.AddThroughputRavenPersistence(settings.ThroughputDatabaseName);

        if (settings.UseEmbeddedServer)
        {
            services.AddSingleton<RavenEmbeddedPersistenceLifecycle>();
            services.AddSingleton<IRavenPersistenceLifecycle>(b => b.GetService<RavenEmbeddedPersistenceLifecycle>());
            services.AddSingleton<IRavenDocumentStoreProvider>(provider => provider.GetRequiredService<RavenEmbeddedPersistenceLifecycle>());
            return;
        }

        services.AddSingleton<RavenExternalPersistenceLifecycle>();
        services.AddSingleton<IRavenPersistenceLifecycle>(b => b.GetService<RavenExternalPersistenceLifecycle>());
        services.AddSingleton<IRavenDocumentStoreProvider>(provider => provider.GetRequiredService<RavenExternalPersistenceLifecycle>());
    }
}