如何将Azure MSI AccessToken传入EF6 DbContext执行Azure SQL迁移

刚好之前踩过EF6迁移结合Azure MSI的坑，给你两个实用的解决方案，应该能解决你的问题：

方案1：自定义DbContext构造函数 + 重写迁移配置 #

这个方案的核心是给DbContext加一个带令牌参数的构造函数，然后在迁移配置里主动获取MSI令牌并传入DbContext，让迁移工具用这个实例来执行操作。

第一步：修改你的DbContext #

给DbContext添加支持令牌注入的构造函数，同时封装连接字符串+令牌的处理逻辑：

public class YourDbContext : DbContext
{
    // 用于迁移的构造函数：接收连接字符串名称和MSI令牌
    public YourDbContext(string connectionStringName, string accessToken)
        : base(BuildConnectionWithToken(connectionStringName, accessToken))
    {
    }

    // 保留CRUD用的构造函数（如果你的业务代码需要）
    public YourDbContext(string accessToken)
        : base(BuildConnectionWithToken("YourDefaultConnString", accessToken))
    {
    }

    // 封装连接字符串+令牌的逻辑
    private static DbConnection BuildConnectionWithToken(string connStrName, string accessToken)
    {
        // 从配置文件拿基础连接字符串（注意：不要包含密码，只留服务器、数据库名等）
        var baseConnStr = ConfigurationManager.ConnectionStrings[connStrName].ConnectionString;
        var sqlConn = new SqlConnection(baseConnStr);
        // 注入MSI令牌
        sqlConn.AccessToken = accessToken;
        return sqlConn;
    }

    // 你的DbSet定义
    public DbSet<YourEntity> YourEntities { get; set; }
}

第二步：重写迁移配置类 #

找到迁移文件夹下的Configuration.cs，重写CreateDbContext方法，主动获取MSI令牌并创建带令牌的DbContext实例：

internal sealed class Configuration : DbMigrationsConfiguration<YourDbContext>
{
    public Configuration()
    {
        AutomaticMigrationsEnabled = false;
        // 根据你的需求设置迁移表的架构等
        MigrationsDirectory = @"Migrations";
    }

    // 关键：重写CreateDbContext，返回带MSI令牌的实例
    protected override YourDbContext CreateDbContext()
    {
        // 调用你已有的MSI令牌获取逻辑
        var accessToken = GetMsiAccessToken();
        // 传入连接字符串名称和令牌，创建DbContext
        return new YourDbContext("YourDefaultConnString", accessToken);
    }

    // 这里就是你原来获取MSI令牌的逻辑，直接复用即可
    private string GetMsiAccessToken()
    {
        var tokenProvider = new AzureServiceTokenProvider();
        // 注意scope要对应Azure SQL的资源标识符
        return tokenProvider.GetAccessTokenAsync("https://database.windows.net/").Result;
    }

    protected override void Seed(YourDbContext context)
    {
        // 你的Seed数据逻辑，不需要改
    }
}

第三步：执行迁移命令 #

在Package Manager Console或者终端里执行迁移时，指定你的配置类（如果迁移工具没自动识别的话）：

Update-Database -ConfigurationTypeName YourProjectNamespace.Configuration

方案2：用DbConfiguration全局拦截连接创建 #

这个方案更省心，不需要修改DbContext构造函数，而是通过EF6的DbConfiguration来全局拦截连接的创建过程，自动注入MSI令牌——不管是CRUD还是迁移，都会自动生效。

第一步：创建自定义DbConfiguration #

public class MsiDbConfig : DbConfiguration
{
    public MsiDbConfig()
    {
        // 注册SQL Server的服务提供者
        SetProviderServices("System.Data.SqlClient", SqlProviderServices.Instance);
        // 替换默认的连接工厂为我们自己的，用来注入令牌
        SetConnectionFactory(new MsiSqlConnectionFactory());
    }
}

// 自定义连接工厂，负责创建带MSI令牌的SqlConnection
public class MsiSqlConnectionFactory : IDbConnectionFactory
{
    public DbConnection CreateConnection(string nameOrConnectionString)
    {
        SqlConnection conn;
        // 判断传入的是连接字符串名称还是完整连接字符串
        if (ConfigurationManager.ConnectionStrings[nameOrConnectionString] != null)
        {
            conn = new SqlConnection(ConfigurationManager.ConnectionStrings[nameOrConnectionString].ConnectionString);
        }
        else
        {
            conn = new SqlConnection(nameOrConnectionString);
        }

        // 注入MSI令牌
        var tokenProvider = new AzureServiceTokenProvider();
        var accessToken = tokenProvider.GetAccessTokenAsync("https://database.windows.net/").Result;
        conn.AccessToken = accessToken;

        return conn;
    }
}

第二步：给DbContext标注配置 #

在你的DbContext类上加上DbConfigurationType特性，指定我们的自定义配置：

[DbConfigurationType(typeof(MsiDbConfig))]
public class YourDbContext : DbContext
{
    // 用默认的构造函数，直接传入连接字符串名称即可
    public YourDbContext() : base("YourDefaultConnString")
    {
    }

    // 你的DbSet定义...
}

这样之后，不管是业务代码里的CRUD，还是执行Update-Database迁移命令，EF都会自动使用带MSI令牌的连接，完全不需要额外处理迁移时的令牌注入问题。

一些注意事项 #

• 本地开发时，要确保你的开发环境能获取MSI令牌：比如用Visual Studio登录Azure账号，或者用Azure CLI执行az login登录。
• 你的连接字符串不要包含密码或者User ID/Password这类认证信息，只保留服务器地址、数据库名、加密设置等基础内容，比如：Server=tcp:your-sql-server.database.windows.net,1433;Database=your-db;Encrypt=True;Connection Timeout=30;
• 如果方案1里用了异步获取令牌，注意.Result可能带来的死锁问题，本地开发可以接受，但生产环境如果有同步获取令牌的方式会更稳妥。

内容的提问来源于stack exchange，提问作者Asif Iqbal