如何忽略WorkloadIdentityCredentialBuilder错误，无需依赖激活配置文件？

我们使用Azure Identity Java库构建链式凭证访问Azure资源，代码如下：

public TokenCredential buildCredential() {
    return new ChainedTokenCredentialBuilder()
        .addFirst(new WorkloadIdentityCredentialBuilder().build())
        .addLast(new AzureCliCredentialBuilder().build())
        .build();
}

本地调试时触发以下异常导致程序终止：

IllegalArgumentException: Must provide non-null values for Client ID, Tenant ID, Service Token File Path properties in WorkloadIdentityCredentialBuilder

现有解决方案依赖激活配置文件判断环境，但我们希望不依赖配置文件，沿用链式凭证逻辑，实现异常时自动切换到AzureCliCredential，而非直接终止程序。

解决方案 #

核心思路是先检查Workload Identity依赖的环境变量是否齐全，仅当所有必填变量存在时才构建WorkloadIdentityCredential，否则直接使用AzureCliCredential作为兜底。修改后的代码如下：

public TokenCredential buildCredential() {
    ChainedTokenCredentialBuilder builder = new ChainedTokenCredentialBuilder();

    // 检查Workload Identity所需的环境变量是否全部存在
    String clientId = System.getenv("AZURE_CLIENT_ID");
    String tenantId = System.getenv("AZURE_TENANT_ID");
    String tokenFilePath = System.getenv("AZURE_FEDERATED_TOKEN_FILE");
    
    if (clientId != null && !clientId.isEmpty()
        && tenantId != null && !tenantId.isEmpty()
        && tokenFilePath != null && !tokenFilePath.isEmpty()) {
        builder.addFirst(new WorkloadIdentityCredentialBuilder().build());
    }

    // 始终添加AzureCliCredential作为 fallback
    builder.addLast(new AzureCliCredentialBuilder().build());

    return builder.build();
}

原理说明 #

• 生产环境（如K8s）中，Workload Identity会自动注入AZURE_CLIENT_ID、AZURE_TENANT_ID、AZURE_FEDERATED_TOKEN_FILE三个环境变量，此时链式凭证会优先使用WorkloadIdentityCredential完成认证
• 本地调试时这些环境变量不存在，WorkloadIdentityCredential不会被加入链式列表，直接使用AzureCliCredential完成认证，避免参数缺失异常
• 完全保留链式凭证的设计逻辑，无需依赖配置文件判断环境，实现环境自适应

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