Azure Data Studio 连接问题：Entra ID 认证失败解决方案

问题背景

近期有用户报告在使用 Azure Data Studio（ADS）连接 Azure Analytics 数据仓库时遇到了认证问题。具体表现为：在 ADS 1.45 版本中可以正常使用 Azure AD（现称 Entra ID）认证连接，但在升级到 1.47 或更高版本后，用户尝试使用"Entra"选项连接时会出现错误。

错误现象

用户在连接时收到以下错误信息：

Error: invalid_grant occurred when acquiring token.
135010 - [2024-07-02 17:52:26Z]: AADSTS135010: UserPrincipal doesn't have the key ID configured. 
Trace ID: 8658fae7-f500-44b1-984f-b35d331bad00 
Correlation ID: 8fd53e14-9074-491d-b792-5436a1167be5 
Timestamp: 2024-07-02 17:52:26Z

值得注意的是，使用 SQL Server 登录方式仍可正常连接数据库，这表明问题特定于 Entra ID 认证流程。

问题分析

根据错误信息和用户反馈，我们可以得出以下关键点：

1. 这是一个与 Azure Active Directory（现 Entra ID）令牌获取相关的认证问题
2. 错误代码 AADSTS135010 表明用户主体缺少配置的密钥 ID
3. 问题出现在 ADS 版本升级后，表明可能与认证流程的变更有关
4. 相同用户在 SSMS（SQL Server Management Studio）中可以正常使用 Entra ID 认证

解决方案

经过用户实践和微软技术支持团队的指导，我们整理出以下几种解决方案：

临时解决方案

1. 切换用户账户：尝试在 Azure Data Studio 中添加一个不同的用户账户（即使该账户没有数据库访问权限），然后切换回原账户
2. 清除认证缓存：
   • 在浏览器中登出所有 Azure AD 账户
   • 清除浏览器缓存和 cookies
   • 重新尝试连接

使用 Azure Data Studio 内置命令

对于更彻底的解决方案，可以使用 ADS 内置的命令来管理认证状态：

1. 打开命令面板：
   • Windows: 按下 Ctrl+Shift+P
   • macOS: 按下 Command+Shift+P
2. 在命令面板中输入"azure"，将显示相关命令
3. 选择并执行以下两个命令：
   • "Azure: 清除所有账户"
   • "Azure: 登录"

这将重置认证状态并重新建立连接。

技术原理

这个问题的根本原因可能与 ADS 升级后对 Entra ID 认证流程的变更有关。新版本可能引入了更严格的密钥验证机制，或者改变了令牌获取的方式。当系统无法找到用户主体配置的密钥 ID 时，就会抛出 AADSTS135010 错误。

最佳实践建议

1. 版本升级注意事项：在升级 ADS 前，建议先测试 Entra ID 认证功能
2. 多账户管理：如果使用多个 Azure AD 账户，确保在切换账户时完全登出前一个账户
3. 定期清除认证缓存：这可以预防许多认证相关的问题
4. 备用连接方式：对于关键业务，建议同时配置 SQL Server 认证作为备用方案

总结

Azure Data Studio 升级后出现的 Entra ID 认证问题虽然令人困扰，但通过上述方法可以有效解决。微软团队也在持续优化认证流程，未来版本可能会提供更稳定的连接体验。对于遇到类似问题的用户，建议先尝试最简单的账户切换方法，如果无效再使用命令面板中的高级功能重置认证状态。