如何防止EF Core模拟用户身份时出现连接重置导致会话终止问题

问题根本原因 #

这个报错是ADO.NET连接池机制与直接修改连接安全上下文的操作冲突导致的：
EF Core默认复用连接池中的连接，你在某条连接上执行EXECUTE AS LOGIN切换身份后，连接使用完毕放回池时，ADO.NET会自动执行sp_reset_connection重置连接状态，但此时连接的安全上下文已经是模拟用户，SQL Server不允许用模拟身份执行连接重置操作，因此会直接销毁该连接，后续复用这条被销毁的连接时就会抛出kill state错误。

可行解决方案 #

方案1：使用EF Core连接拦截器（最推荐） #

将身份模拟和还原逻辑与连接的生命周期绑定，确保连接放回池之前一定还原初始身份：
首先自定义连接拦截器：

public class UserImpersonationInterceptor : DbConnectionInterceptor
{
    private readonly string _targetLogin;

    public UserImpersonationInterceptor(string targetLogin)
    {
        _targetLogin = targetLogin;
    }

    // 同步打开连接后执行模拟
    public override InterceptionResult ConnectionOpened(DbConnection connection, ConnectionEndEventData eventData)
    {
        using var cmd = connection.CreateCommand();
        cmd.CommandText = "EXECUTE AS LOGIN = @TargetLogin;";
        cmd.Parameters.Add(new SqlParameter("@TargetLogin", _targetLogin));
        cmd.ExecuteNonQuery();
        return base.ConnectionOpened(connection, eventData);
    }

    // 异步打开连接后执行模拟
    public override async Task<InterceptionResult> ConnectionOpenedAsync(DbConnection connection, ConnectionEndEventData eventData, CancellationToken cancellationToken = default)
    {
        using var cmd = connection.CreateCommand();
        cmd.CommandText = "EXECUTE AS LOGIN = @TargetLogin;";
        cmd.Parameters.Add(new SqlParameter("@TargetLogin", _targetLogin));
        await cmd.ExecuteNonQueryAsync(cancellationToken);
        return await base.ConnectionOpenedAsync(connection, eventData, cancellationToken);
    }

    // 同步关闭连接前还原身份
    public override InterceptionResult ConnectionClosing(DbConnection connection, ConnectionEventData eventData)
    {
        using var cmd = connection.CreateCommand();
        cmd.CommandText = "REVERT;";
        cmd.ExecuteNonQuery();
        return base.ConnectionClosing(connection, eventData);
    }

    // 异步关闭连接前还原身份
    public override async Task<InterceptionResult> ConnectionClosingAsync(DbConnection connection, ConnectionEventData eventData, CancellationToken cancellationToken = default)
    {
        using var cmd = connection.CreateCommand();
        cmd.CommandText = "REVERT;";
        await cmd.ExecuteNonQueryAsync(cancellationToken);
        return await base.ConnectionClosingAsync(connection, eventData, cancellationToken);
    }
}

创建DbContext时动态注入对应当前用户的拦截器即可：

var dbOptions = new DbContextOptionsBuilder<YourDbContext>()
    .UseSqlServer("你的数据库连接字符串")
    .AddInterceptors(new UserImpersonationInterceptor("当前需要模拟的登录名"))
    .Options;
await using var dbContext = new YourDbContext(dbOptions);
// 后续所有数据库操作都会自动以模拟身份执行

方案2：手动封装操作作用域 #

如果你不想使用拦截器，也可以封装操作委托，确保所有数据库操作执行完毕后主动还原身份：

public static async Task<T> RunAsUserAsync<T>(string targetUser, DbContext dbContext, Func<Task<T>> dbOperation)
{
    var connection = dbContext.Database.GetDbConnection();
    if (connection.State != ConnectionState.Open)
        await connection.OpenAsync();

    // 校验当前身份
    using var checkCmd = connection.CreateCommand();
    checkCmd.CommandText = "SELECT SUSER_NAME()";
    var currentUser = (string)await checkCmd.ExecuteScalarAsync();
    if (currentUser.Equals(targetUser, StringComparison.OrdinalIgnoreCase))
    {
        return await dbOperation();
    }

    // 执行模拟
    bool needRevert = false;
    using var execCmd = connection.CreateCommand();
    execCmd.CommandText = "EXECUTE AS LOGIN = @TargetUser;";
    execCmd.Parameters.Add(new SqlParameter("@TargetUser", targetUser));
    await execCmd.ExecuteNonQueryAsync();
    needRevert = true;

    try
    {
        // 执行传入的数据库操作
        return await dbOperation();
    }
    finally
    {
        // 无论操作是否成功都还原身份
        if (needRevert)
        {
            using var revertCmd = connection.CreateCommand();
            revertCmd.CommandText = "REVERT;";
            await revertCmd.ExecuteNonQueryAsync();
        }
    }
}

使用示例：

var result = await RunAsUserAsync("要模拟的用户名", dbContext, async () =>
{
    return await dbContext.YourTable.ToListAsync();
});

方案3：禁用连接池（不推荐） #

如果并发量极低，也可以直接在连接字符串中添加Pooling=false禁用连接池，每次操作都会新建独立连接，不会有复用冲突问题，但数据库连接开销会大幅提升，仅适合测试或极低负载场景。

注意事项 #

• 运行应用的服务账号仅需要授予IMPERSONATE ANY LOGIN权限，遵循最小权限原则，不要分配过高权限。
• 如果需要模拟的是数据库级别的用户而非服务器级别的登录，将语句中的EXECUTE AS LOGIN改为EXECUTE AS USER即可。

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