btcpayserver/BTCPayServer.Data/ApplicationDbContextExtensions.cs

#nullable enable
using System;
using System.Data.Common;
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;

namespace BTCPayServer.Data;

public static partial class ApplicationDbContextExtensions
{
    public static DbContext GetDbContext<T>(this DbSet<T> dbSet) where T : class
    {
        var infrastructure = dbSet as IInfrastructure<IServiceProvider>;
        var serviceProvider = infrastructure.Instance;
        var currentDbContext = (ICurrentDbContext)serviceProvider.GetService(typeof(ICurrentDbContext))!;
        return currentDbContext.Context;
    }

    public static DbConnection GetDbConnection<T>(this DbSet<T> dbSet) where T : class
        => dbSet.GetDbContext().Database.GetDbConnection();
}