EF Core DB First，以及如何避免在模型生成时覆盖构造函数

Question

我将不再对 Azure DB 使用 SQL 身份验证，而是改用 Active Directory 托管身份验证，如 article 中所述。

基本上，我正在做两件主要的事情来让它发挥作用。

1- 在 DBContext 构造函数中注入令牌：

public MyDBContext(DbContextOptions<MyDBContext> options)
    : base(options)
{
    var conn = (SqlConnection)Database.GetDbConnection();
    conn.AccessToken = (new Microsoft.Azure.Services.AppAuthentication.AzureServiceTokenProvider()).GetAccessTokenAsync("https://database.windows.net/").Result;

}

2- 在我的 Web 应用启动文件中，我正在注入 DBContext

    string SqlConnection = localConfig["SqlConnectionString"];
    services.AddDbContext<MyDBContext>(options => options.UseSqlServer(SqlConnection, sqlServerOptions => { sqlServerOptions.CommandTimeout(1000); }));

我现在的问题是，每次我需要使用Scaffold-DbContext 命令刷新模型时，我的MyDbContext 都会被覆盖，并且我会丢失对构造函数所做的更改。

有哪些解决方案可以避免这个问题？或者，有没有更好的方法将令牌注入其他地方有效？

编辑： 请注意，我使用的是 EF 2.x

Answer 1

我使用拦截器注入访问令牌：

public class ManagedIdentityConnectionInterceptor : DbConnectionInterceptor
{
    private readonly bool _useManagedIdentity;
    private readonly AzureServiceTokenProvider _tokenProvider;

    public ManagedIdentityConnectionInterceptor(bool useManagedIdentity)
    {
        _useManagedIdentity = useManagedIdentity;
        _tokenProvider = new AzureServiceTokenProvider();
    }

    public override async Task<InterceptionResult> ConnectionOpeningAsync(
        DbConnection connection,
        ConnectionEventData eventData,
        InterceptionResult result,
        CancellationToken cancellationToken = default)
    {
        if (_useManagedIdentity)
        {
            // In Azure, get an access token for the connection
            var sqlConnection = (SqlConnection)connection;
            var accessToken = await GetAccessTokenAsync(cancellationToken);
            sqlConnection.AccessToken = accessToken;
        }

        return result;
    }

    private async Task<string> GetAccessTokenAsync(CancellationToken cancellationToken)
    {
        // Get access token for Azure SQL DB
        var resource = "https://database.windows.net/";
        return await _tokenProvider.GetAccessTokenAsync(resource, cancellationToken: cancellationToken);
    }
}

可以这样添加：

// Detect environment somehow (locally you might not want to use tokens)
var useManagedIdentity = true;
var managedIdentityInterceptor = new ManagedIdentityConnectionInterceptor(useManagedIdentity);
services.AddDbContext<Context>(options => options.UseSqlServer(connectionString).AddInterceptors(managedIdentityInterceptor));

这种方式不需要在构造函数中进行任何更改。 在打开与 SQL DB 的连接之前，拦截器将获得一个令牌。 我们还避免在构造函数中执行同步异步。 请注意，这需要 EF Core 3.x。

Answer 2

您可以将UseSqlServer 重载与DbConnection 参数一起使用并传递配置的SqlConnection 对象：

var sqlConnectionString = localConfig["SqlConnectionString"];
services.AddDbContext<MyDBContext>(
    options => options.UseSqlServer(new SqlConnection(sqlConnectionString)
    {
        AccessToken = (new Microsoft.Azure.Services.AppAuthentication.AzureServiceTokenProvider()).GetAccessTokenAsync("https://database.windows.net/").Result
    },
    sqlServerOptions => { sqlServerOptions.CommandTimeout(1000); }));