DatabaseContextFactory.cs

Go to the documentation of this file.

using System;
using System.Threading.Tasks;
 
using Microsoft.Extensions.DependencyInjection;
 
namespace Tgstation.Server.Host.Database
{
    sealed class DatabaseContextFactory : IDatabaseContextFactory
    {
        readonly IServiceScopeFactory scopeFactory;
 
        public DatabaseContextFactory(IServiceScopeFactory scopeFactory)
        {
            this.scopeFactory = scopeFactory ?? throw new ArgumentNullException(nameof(scopeFactory));
 
            using var scope = scopeFactory.CreateScope();
            scope.ServiceProvider.GetRequiredService<IDatabaseContext>();
        }
 
        public async ValueTask UseContext(Func<IDatabaseContext, ValueTask> operation)
        {
            ArgumentNullException.ThrowIfNull(operation);
 
            await using var scope = scopeFactory.CreateAsyncScope();
            await operation(scope.ServiceProvider.GetRequiredService<IDatabaseContext>());
        }
 
        public async ValueTask UseContextTaskReturn(Func<IDatabaseContext, Task> operation)
        {
            ArgumentNullException.ThrowIfNull(operation);
 
            await using var scope = scopeFactory.CreateAsyncScope();
            await operation(scope.ServiceProvider.GetRequiredService<IDatabaseContext>());
        }
    }
}