.NET CosmosDB SDK批量支持：多层异步任务能否触发批量？

不会启用批量支持，原因和调整方法如下：

为什么批量支持不生效 #

Cosmos DB .NET SDK的批量功能依赖于先收集多个未被立即等待的SDK操作任务（比如UpsertItemAsync），再统一用Task.WhenAll等待。这样SDK才能把符合条件的操作（同分区键、同操作类型等）打包成批量请求，优化网络调用。

在你的代码里，每个UpsertItemAsync调用都会在InsertOrUpdate方法里被立即await。这意味着每个操作都是单独启动并完成的——根本没有机会让SDK收集多个未等待的任务来打包。顶层的Task.WhenAll只是等待Save(a)返回的包装任务，但此时每个底层的SDK操作都已经单独执行完了。

如何修改代码启用批量支持 #

要用上批量功能，你需要直接收集SDK操作的任务，而不是在嵌套方法里立即等待它们。调整方式如下：

方案一：移除不必要的嵌套await，直接返回SDK任务 #

private async Task StartHere()
{
    var articles = new List<Article>(); // 10k条数据
    // 直接收集SDK任务
    var bulkTasks = articles.Select(a => InsertOrUpdate(a, CancellationToken.None));
    
    await Task.WhenAll(bulkTasks); // 此时会启用批量支持
}

private Task<string> InsertOrUpdate(Article entity, CancellationToken cancellationToken)
{
    var partitionKey = new PartitionKey(entity.Id);

    // 直接返回SDK任务，不立即await
    return _container.UpsertItemAsync(entity, partitionKey, new ItemRequestOptions
    {
        EnableContentResponseOnWrite = false,
        IfMatchEtag = entity.ETag
    }, cancellationToken: cancellationToken)
    .ContinueWith(t => entity.Id, cancellationToken);
}

// 如果需要保留Save/Upsert方法，让它们直接返回任务即可
private Task<Article> Save(Article article)
{
    return Upsert(article, CancellationToken.None)
        .ContinueWith(t => article);
}

private Task Upsert(Article article, CancellationToken cancellationToken)
{
    return InsertOrUpdate(article, cancellationToken);
}

方案二：保留async方法，但延迟await到顶层 #

private async Task StartHere()
{
    var articles = new List<Article>(); // 10k条数据
    var bulkTasks = new List<Task>();
    
    foreach (var article in articles)
    {
        // 获取SDK任务但不立即等待
        var sdkTask = GetUpsertTask(article);
        bulkTasks.Add(sdkTask);
    }
    
    await Task.WhenAll(bulkTasks);
}

private Task GetUpsertTask(Article article)
{
    var partitionKey = new PartitionKey(article.Id);
    return _container.UpsertItemAsync(article, partitionKey, new ItemRequestOptions
    {
        EnableContentResponseOnWrite = false,
        IfMatchEtag = article.ETag
    }, CancellationToken.None);
}

核心要点 #

批量支持生效的关键是：不要在收集完所有要批量的任务前，单独await每个SDK操作。任何在嵌套方法里提前await SDK任务的行为，都会破坏批量打包的能力。

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