DatabaseContextFactory.cs

Go to the documentation of this file.

using Microsoft.Extensions.DependencyInjection;
using System;
using System.Threading.Tasks;
using Tgstation.Server.Host.Models;

namespace Tgstation.Server.Host.Core
{
    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 Task UseContext(Func<IDatabaseContext, Task> operation)
        {
            if (operation == null)
                throw new ArgumentNullException(nameof(operation));

            using (var scope = scopeFactory.CreateScope())
                await operation(scope.ServiceProvider.GetRequiredService<IDatabaseContext>()).ConfigureAwait(false);
        }
    }
}