为Parallel.ForEach中的项目设置超时

Question

我正在使用Parallel.ForEach处理并发词典集合。 ConcurrentDictionary包含“密钥”和“字节”字段。

concurrentDictionary-是此处的并发字典集合。

Parallel.ForEach(concurentDictionary, (item) =>
{
    var mailObject = dbContext.MailObjects.Where(x=> x.MailObjectId == item.key).Select(y=> y);
    mailObject.MailBody = ConvertToPdf(item.Value.MailBody);
} 
dbContext.SaveChanges();

我想设置超时时间，如果任何特定项目花费的最大时间设置为60秒。 如果耗时超过60秒，我想退出而不保存该特定项目。 其余项目应通过dbContext.SaveChanges()方法保存。 如何在Parallel.ForEach中实现？

Answer 1

这样的事情可能近似您想要的。

CancellationTokenSource cts = new CancellationTokenSource();

ParallelOptions parallelOptions = new ParallelOptions();
parallelOptions.CancellationToken = cts.Token;
parallelOptions.MaxDegreeOfParallelism = System.Environment.ProcessorCount;

Timer timer = new Timer(callback => { cts.Cancel(); }, null, 60*1000, Timeout.Infinite);

try
{
    Parallel.ForEach(concurentDictionary, parallelOptions, (item) =>
    {
        var mailObject = dbContext.MailObjects.Where(x => x.MailObjectId == item.key).Select(y => y);
        mailObject.MailBody = ConvertToPdf(item.Value.MailBody);
        parallelOptions.CancellationToken.ThrowIfCancellationRequested();
    });
}
catch (OperationCanceledException e)
{
    // Log the cancellation event...
}
finally
{
    dbContext.SaveChanges();
    cts.Dispose();
}

这是该解决方案的完整修订版，其中为每次迭代分配了计时器。 如果不超过取消令牌，它会利用循环状态来停止迭代（如果一个（或多个）超过时间限制）。 这不依赖于异常退出循环。 但是，在编写本文时，任何异常也会触发所有循环停止。 请注意，这不会立即停止所有循环，并会完成现有循环的执行，直到它们也完成或超时。

{
    if (Parallel.ForEach(concurentDictionary, (item, loopState) =>
        {
            Timer timer = new Timer(callback => { loopState.Stop(); }, null, 60 * 1000, Timeout.Infinite);

            try
            {
                if (loopState.ShouldExitCurrentIteration || loopState.IsExceptional)
                    loopState.Stop();

                var mailObject = dbContext.MailObjects.Where(x => x.MailObjectId == item.key).Select(y => y);
                mailObject.MailBody = ConvertToPdf(item.Value.MailBody);

                if (loopState.ShouldExitCurrentIteration || loopState.IsExceptional)
                    loopState.Stop();
            }
            catch (Exception)
            {
                loopState.Stop();
                throw;
            }
            finally
            {
                timer.Dispose();
            }
        }
    ).IsCompleted)
    {
        // All events complete
    }

}
catch (AggregateException)
{
    // Execution will not get to this point until all of the iterations have completed (or one 
    // has failed, and all that were running when that failure occurred complete).
}