Azure.Storage.DataMovement：如何共享凭证避免频繁调用GetToken？

问题背景 #

通过IAzureClientFactory<BlobServiceClient>创建Blob客户端，并用services.AddAzureClients(builder => builder.UseCredential(credential))配置全局共享凭证，但使用BlobsStorageResourceProvider生成StorageResource进行Blob迁移时，每个传输任务都会触发新的GetToken调用，频率过高导致大量请求超时，传输失败。

解决方案 #

1. 确保凭证实例为单例并启用令牌缓存 #

Azure SDK的TokenCredential（如DefaultAzureCredential）默认会缓存令牌，但必须保证凭证实例是全局单例，避免每次创建新实例导致缓存失效。

注册凭证和Azure客户端：

// 注册单例TokenCredential
services.AddSingleton<TokenCredential>(sp => new DefaultAzureCredential());

// 配置Azure客户端使用单例凭证
services.AddAzureClients(builder =>
{
    builder.UseCredential(sp => sp.GetRequiredService<TokenCredential>());
    // 根据需要注册BlobServiceClient
    builder.AddBlobServiceClient(new Uri("https://your-storage-account.blob.core.windows.net"));
});

2. 复用BlobsStorageResourceProvider实例，避免调用静态方法 #

原代码中直接调用BlobsStorageResourceProvider.FromClient静态方法，可能导致内部创建新的客户端上下文，绕过共享凭证的缓存。修改Wrapper类，直接使用注入的实例方法：

public class BlobsStorageResourceProviderWrapper(BlobsStorageResourceProvider blobsStorageResourceProvider) 
    : IBlobsStorageResourceProvider
{
    private readonly BlobsStorageResourceProvider _innerProvider = blobsStorageResourceProvider;

    public StorageResource FromClient(BlobContainerClient blobContainerClient)
        => _innerProvider.FromClient(blobContainerClient);

    public StorageResource FromClient(BlobContainerClient blobContainerClient, string blobPrefix)
        => _innerProvider.FromClient(blobContainerClient, new BlobStorageResourceContainerOptions { BlobPrefix = blobPrefix });

    public async Task<StorageResource> FromContainerAsync(Uri containerUri,
            BlobStorageResourceContainerOptions? options = default,
            CancellationToken cancellationToken = default)
        => await _innerProvider.FromContainerAsync(containerUri, options, cancellationToken);
}

注册Wrapper为单例：

services.AddSingleton<IBlobsStorageResourceProvider, BlobsStorageResourceProviderWrapper>();

3. 复用BlobContainerClient实例 #

生成StorageResource时，确保使用同一个BlobServiceClient创建的BlobContainerClient，避免每次传输都重新创建客户端：

// 从IAzureClientFactory获取全局共享的BlobServiceClient
var blobServiceClient = _clientFactory.CreateClient();
// 复用同一个BlobContainerClient
var sourceContainerClient = blobServiceClient.GetBlobContainerClient("source-container");
var targetContainerClient = blobServiceClient.GetBlobContainerClient("target-container");

// 生成StorageResource
var sourceResource = _resourceProvider.FromClient(sourceContainerClient);
var targetResource = _resourceProvider.FromClient(targetContainerClient);

// 执行迁移
await Transfer(sourceResource, targetResource, cancellationToken);

4. 验证令牌缓存效果 #

启用Azure SDK的调试日志，查看GetToken的调用频率，确认令牌是否按过期周期（默认1小时）刷新，而非每个Blob传输都调用：

// 在Program.cs中配置日志
builder.Logging.AddFilter("Azure.Core", LogLevel.Debug);

核心原理 #

问题根源在于多个客户端/凭证实例导致令牌缓存无法复用。通过确保凭证单例、客户端复用、BlobsStorageResourceProvider使用全局实例，让所有迁移操作共享同一个令牌缓存，减少GetToken调用频率，避免超时。

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