Task.WhenAll 不等待任务完成[关闭]

Question

我正在编写从服务器获取不同数据部分的代码，将它们存储在 dict 中，以便它们按正确的顺序排列，然后在获取所有数据后将它们组合起来。但由于某种原因，它不起作用。它在获取所有内容之前进入CombineDataParts()。我不明白我做错了什么，因为所有方法都在等待。请帮忙！

public async Task<Data> GetTableData(List<string> partNames)
{
    List<Task> TaskList = new List<Task>();
    SortedDictionary<string, Data> dataList = new SortedDictionary<string, Data>();
    
    foreach (var name in partNames)
    {
        TaskList.Add(SaveDatainDict(name, DataList));
    }
    
    await Task.WhenAll(TaskList);
    return CombineDataParts(tableDataList.Select(t => t.Value).ToList());
}


private async Task SaveDatainDict(string part, SortedDictionary<string, Data> dataList)
{
    dataList[part] = await GetDataPart(part);
}


private async Task<Data> GetDataPart(string key)
{   
    using (var response = await client.GetData(key))
    {
        return ProcessData(response.Stream);
    }   
}

Answer 1

为了让未来的读者更清楚，将我的评论分解为完整的答案...

正如@TheodorZoulais 所暗示的，这可能是SortedDictionary 不是线程安全的问题。这意味着当任务本身完成时，SortedDictionary 不能保证立即在线程间一致地反映状态，这在这里引起了头疼。这可能可以通过锁定正确的位置来解决……但有一种更简单的方法。

由于您只是使用SortedDictionary 来组合Tasks 的结果，我们可以通过稍微不同的使用Task.WhenAll 来完全避开问题。 Task.WhenAll 做了两件非常方便的事情：

1. 为我们返回一个任务到await
2. 如果我们传递给Task.WhenAll 的Tasks 有结果，它会将这些结果合并到一个可枚举的结果中(link to the relevant documentation)

第一点是你现在的使用方式，这很好。但是如果我们也使用第二点，我们可以完全取消 SortedDictionary。

因此，只需对您的代码进行一些调整即可让我们实现线程安全...

public async Task<Data> GetTableData(List<string> partNames)
{
    // This time around, we will keep a list of Task<Data> instead of just Task
    List<Task<Data>> taskList = new List<Task<Data>>();

    foreach (var name in partNames)
    {
        // Directly save the tasks from GetDataPart
        taskList.Add(GetDataPart(name));
    }

    // Save the results combined by Task.WhenAll
    IEnumerable<Data> dataParts = await Task.WhenAll(taskList);

    // Looks like CombineDataParts was expecting a list, so convert and we're done!
    return CombineDataParts(dataParts.ToList());
}

private async Task<Data> GetDataPart(string key)
{   
    using (var response = await client.GetData(key))
    {
        return ProcessData(response.Stream);
    }   
}

这里有一些额外的说明：

• SortedDictionary 确保对键进行排序，而此代码没有。此实现将按照名称在partNames 中的顺序返回值（引用：Tasks.WhenAll<TResult>(Task<TResult>[]) documentation 上的注释）。如果您想在此方法中强制执行特定排序（正如 SortedDictionary 在原始代码中所做的那样），您应该考虑使用类似 List.Sort 的方式对 partNames 进行排序
• 根据您的偏好，您可以在顶部使用另一个 LINQ 语句而不是 foreach 循环来节省几行：IEnumerable<Task<Data>> taskList = partNames.Select(name => GetDataPart(name));