Task.WhenAny с отменой незавершенных задач и тайм-аут - PullRequest
Новая
       50

Task.WhenAny с отменой незавершенных задач и тайм-аут

8 голосов
/ 19 мая 2019

В моем приложении я создаю несколько одновременных веб-запросов, и я удовлетворен, когда любой из них завершается, поэтому я использую метод Task.WhenAny: 

var urls = new string[] {
    "https://stackoverflow.com",
    "https://superuser.com",
    "https://www.reddit.com/r/chess",
};
var tasks = urls.Select(async url =>
{
    using (var webClient = new WebClient())
    {
        return (Url: url, Data: await webClient.DownloadStringTaskAsync(url));
    }
}).ToArray();
var firstTask = await Task.WhenAny(tasks);
Console.WriteLine($"First Completed Url: {firstTask.Result.Url}");
Console.WriteLine($"Data: {firstTask.Result.Data.Length:#,0} chars");

Первый завершенный URL: https://superuser.com
Данные: 121,954 символа

Что мне не нравится в этой реализации, так это то, что незавершенные задачи продолжают загружать данные, которые мне больше не нужны, итрата пропускной способности я бы предпочел сохранить для моей следующей партии запросов.Поэтому я думаю об отмене других задач, но я не уверен, как это сделать.Я нашел, как использовать CancellationToken для отмены определенного веб-запроса: 

public static async Task<(string Url, string Data)> DownloadUrl(
    string url, CancellationToken cancellationToken)
{
    try
    {
        using (var webClient = new WebClient())
        {
            cancellationToken.Register(webClient.CancelAsync);
            return (url, await webClient.DownloadStringTaskAsync(url));
        }
    }
    catch (WebException ex) when (ex.Status == WebExceptionStatus.RequestCanceled)
    {
        cancellationToken.ThrowIfCancellationRequested();
        throw;
    }
}

Теперь мне нужна реализация Task.WhenAny, которая будет принимать массив URL-адресов и будет использовать мою функцию DownloadUrlизвлекать данные самого быстро отвечающего сайта и будет обрабатывать логику отмены более медленных задач.Было бы неплохо, если бы он имел аргумент timeout , чтобы обеспечить защиту от бесконечных задач.Поэтому мне нужно что-то вроде этого: 

public static Task<Task<TResult>> WhenAnyEx<TSource, TResult>(
    this IEnumerable<TSource> source,
    Func<TSource, CancellationToken, Task<TResult>> taskFactory,
    int timeout)
{
    // What to do here?
}

Есть идеи?

Ответы [ 2 ]

5 голосов
/ 19 мая 2019

Просто передайте всем своим задачам один и тот же токен отмены, что-то вроде этого: 

CancellationTokenSource cts = new CancellationTokenSource();
CancellationToken ct = cts.Token;
// here you specify how long you want to wait for task to finish before cancelling
int timeout = 5000;
cts.CancelAfter(timeout);
// pass ct to all your tasks and start them
await Task.WhenAny(/* your tasks here */);
// cancel all tasks
cts.Cancel();

Кроме того, вам необходимо прочитать эту ветку, чтобы знать, как правильно использовать CancellationToken: Когда я использую CancelAfter (), задание все еще выполняется

0 голосов
/ 19 мая 2019

Обновление : лучшее решение на основе Ответ Стивена Клири и MSDN и ответ Свика : 

CancellationTokenSource source = new CancellationTokenSource();
source.CancelAfter(TimeSpan.FromSeconds(1));

var tasks = urls.Select(url => Task.Run( async () => 
{
    using (var webClient = new WebClient())
    {
        token.Register(webClient.CancelAsync);
        var result = (Url: url, Data: await webClient.DownloadStringTaskAsync(url));
        token.ThrowIfCancellationRequested();
        return result.Url;
    }
}, token)).ToArray();

string url;
try
{
    // (A canceled task will raise an exception when awaited).
    var firstTask = await Task.WhenAny(tasks);
    url = (await firstTask).Url;
}   
catch (AggregateException ae) {
   foreach (Exception e in ae.InnerExceptions) {
      if (e is TaskCanceledException)
         Console.WriteLine("Timeout: {0}", 
                           ((TaskCanceledException) e).Message);
      else
         Console.WriteLine("Exception: " + e.GetType().Name);
   }
}

неоптимальное решение

Время ожидания можно решить, добавив задачу, которая просто ждет и завершается через заданное время.Затем вы проверяете, какое задание выполнено первым, если оно является ожидающим, а затем истекло время ожидания. 

Task timeout = Task.Delay(10000);
var firstTask = await Task.WhenAny(tasks.Concat(new Task[] {timeout}));
if(firstTask == timeout) { ... } //timed out
source.Cancel();
...