Azure Functions中如何自动刷新数据库连接的访问令牌？

问题根源分析 #

你遇到的两个问题本质上都是令牌生命周期与连接管理不匹配导致的：

1. 第一种写法在DbContext构造函数中设置AccessToken时，EF可能已提前打开数据库连接，因此无法修改已打开连接的令牌属性。
2. 第二种写法提前创建单例连接并仅设置一次令牌，当令牌过期后，连接不会自动重新获取新令牌，最终触发登录失败报错。

正确解决方案：利用SqlConnection的AccessTokenProvider自动刷新令牌 #

Microsoft.Data.SqlClient提供的AccessTokenProvider属性，支持指定一个委托逻辑，在每次需要令牌时（包括连接初始化、令牌过期重连场景）自动获取最新有效令牌。结合依赖注入管理DefaultAzureCredential的生命周期，即可实现令牌自动刷新。

步骤1：配置服务与DbContext #

在Program.cs中注册DefaultAzureCredential为单例（避免重复创建实例，同时利用其内置令牌缓存），并在DbContext配置中指定动态令牌获取逻辑：

var host = new HostBuilder()
    .ConfigureFunctionsWorkerDefaults()
    .ConfigureServices(s =>
    {
        // 注册DefaultAzureCredential为单例，复用实例并利用其令牌缓存能力
        s.AddSingleton<DefaultAzureCredential>();
        
        // 配置DbContext，动态获取有效令牌
        s.AddDbContext<MyDbContext>((serviceProvider, options) =>
        {
            var credential = serviceProvider.GetRequiredService<DefaultAzureCredential>();
            var connectionString = "你的数据库连接字符串";

            options.UseSqlServer(connectionString, sqlOptions =>
            {
                // 可选：启用重试机制，应对临时连接异常
                sqlOptions.EnableRetryOnFailure();
                
                // 配置连接的令牌获取逻辑
                sqlOptions.ConfigureConnection(connection =>
                {
                    var sqlConn = (SqlConnection)connection;
                    sqlConn.AccessTokenProvider = async () =>
                    {
                        // 每次需要令牌时，从Azure AD获取最新有效令牌
                        var tokenResult = await credential.GetTokenAsync(
                            new TokenRequestContext(new[] { "https://database.windows.net/.default" })
                        );
                        return tokenResult.Token;
                    };
                });
            });
        });
    })
    .Build();

host.Run();

步骤2：简化MyDbContext实现 #

无需在DbContext构造函数中处理令牌逻辑，保持基础定义即可：

public class MyDbContext : DbContext
{
    public MyDbContext(DbContextOptions<MyDbContext> options) : base(options)
    {
    }

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

关键说明 #

• 令牌自动刷新：AccessTokenProvider委托会在每次连接需要身份验证时触发，包括令牌过期后的重连场景，确保始终使用有效令牌。
• DefaultAzureCredential单例：单例注册可复用实例，同时它会自动缓存已获取的令牌，避免频繁向Azure AD发起请求，提升性能。
• 依赖版本要求：确保使用最新版的Microsoft.Data.SqlClient和Microsoft.EntityFrameworkCore.SqlServer，AccessTokenProvider为较新特性，旧版本可能不支持。

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