elsa-core/src/modules/Elsa.Common/Multitenancy/Implementations/DefaultTenantService.cs

using Elsa.Extensions;
using Microsoft.Extensions.DependencyInjection;

namespace Elsa.Common.Multitenancy;

public class DefaultTenantService(IServiceScopeFactory scopeFactory, ITenantScopeFactory tenantScopeFactory, TenantEventsManager tenantEvents, ITenantAccessor tenantAccessor) : ITenantService, IAsyncDisposable
{
    private readonly AsyncServiceScope _serviceScope = scopeFactory.CreateAsyncScope();
    private readonly SemaphoreSlim _initializationLock = new(1, 1);
    private readonly SemaphoreSlim _refreshLock = new(1, 1);
    private IDictionary<string, Tenant>? _tenantsDictionary;
    private IDictionary<Tenant, TenantScope>? _tenantScopesDictionary;

    public async ValueTask DisposeAsync()
    {
        await _serviceScope.DisposeAsync();
        _initializationLock.Dispose();
    }

    public async Task<Tenant?> FindAsync(string id, CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return dictionary.TryGetValue(id.EmptyIfNull(), out var tenant) ? tenant : null;
    }

    public async Task<Tenant?> FindAsync(TenantFilter filter, CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return filter.Apply(dictionary.Values.AsQueryable()).FirstOrDefault();
    }

    public async Task<Tenant> GetAsync(string id, CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return dictionary[id.EmptyIfNull()];
    }

    public async Task<Tenant> GetAsync(TenantFilter filter, CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return filter.Apply(dictionary.Values.AsQueryable()).First();
    }

    public async Task<IEnumerable<Tenant>> ListAsync(CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return dictionary.Values;
    }

    public async Task<IEnumerable<Tenant>> ListAsync(TenantFilter filter, CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        return filter.Apply(dictionary.Values.AsQueryable());
    }

    public async Task ActivateTenantsAsync(CancellationToken cancellationToken = default)
    {
        await RefreshAsync(cancellationToken);
    }

    public async Task DeactivateTenantsAsync(CancellationToken cancellationToken = default)
    {
        var dictionary = await GetTenantsDictionaryAsync(cancellationToken);
        var tenants = dictionary.Values.ToArray();

        foreach (var tenant in tenants)
            await UnregisterTenantAsync(tenant, false, cancellationToken);
    }

    public async Task RefreshAsync(CancellationToken cancellationToken = default)
    {
        await _refreshLock.WaitAsync(cancellationToken);

        try
        {
            await using var scope = scopeFactory.CreateAsyncScope();
            var tenantsProvider = scope.ServiceProvider.GetRequiredService<ITenantsProvider>();
            var currentTenants = await GetTenantsDictionaryAsync(cancellationToken);
            var currentTenantIds = currentTenants.Keys;
            var newTenants = (await tenantsProvider.ListAsync(cancellationToken)).ToDictionary(x => x.Id.EmptyIfNull());
            var newTenantIds = newTenants.Keys;
            var removedTenantIds = currentTenantIds.Except(newTenantIds).ToArray();
            var addedTenantIds = newTenantIds.Except(currentTenantIds).ToArray();

            foreach (var removedTenantId in removedTenantIds)
            {
                var removedTenant = currentTenants[removedTenantId];
                await UnregisterTenantAsync(removedTenant, true, cancellationToken);
            }

            foreach (var addedTenantId in addedTenantIds)
            {
                var addedTenant = newTenants[addedTenantId];
                await RegisterTenantAsync(addedTenant, cancellationToken);
            }
        }
        finally
        {
            _refreshLock.Release();
        }
    }

    private async Task<IDictionary<string, Tenant>> GetTenantsDictionaryAsync(CancellationToken cancellationToken)
    {
        if (_tenantsDictionary == null)
        {
            await _initializationLock.WaitAsync(cancellationToken); // Lock to ensure single-threaded initialization
            try
            {
                if (_tenantsDictionary == null) // Double-check locking
                {
                    _tenantsDictionary = new Dictionary<string, Tenant>();
                    _tenantScopesDictionary = new Dictionary<Tenant, TenantScope>();
                    var tenantsProvider = _serviceScope.ServiceProvider.GetRequiredService<ITenantsProvider>();
                    var tenants = await tenantsProvider.ListAsync(cancellationToken);

                    foreach (var tenant in tenants)
                        await RegisterTenantAsync(tenant, cancellationToken);
                }
            }
            finally
            {
                _initializationLock.Release();
            }
        }

        return _tenantsDictionary;
    }

    private async Task RegisterTenantAsync(Tenant tenant, CancellationToken cancellationToken = default)
    {
        var scope = tenantScopeFactory.CreateScope(tenant);
        _tenantsDictionary![tenant.Id.EmptyIfNull()] = tenant;
        _tenantScopesDictionary![tenant] = scope;

        using (tenantAccessor.PushContext(tenant))
            await tenantEvents.TenantActivatedAsync(new(tenant, scope, cancellationToken));
    }

    private async Task UnregisterTenantAsync(Tenant tenant, bool isDeleted, CancellationToken cancellationToken = default)
    {
        if (_tenantScopesDictionary!.Remove(tenant, out var scope))
        {
            _tenantsDictionary!.Remove(tenant.Id.EmptyIfNull(), out _);

            using (tenantAccessor.PushContext(tenant))
            {
                await tenantEvents.TenantDeactivatedAsync(new(tenant, scope, cancellationToken));

                if (isDeleted)
                    await tenantEvents.TenantDeletedAsync(new(tenant, scope, cancellationToken));
            }
        }
    }
}