如果同步方法执行时间过长,我正在寻找一种有效的方法来引发超时异常。我看过一些样品,但没有什么能完全符合我的要求。
我需要做的是
- 检查同步方法是否超出其 SLA
- 如果它确实抛出超时异常
如果同步方法执行时间过长,我不必终止它。(多次故障将使断路器跳闸并防止级联故障)
到目前为止,我的解决方案如下所示。请注意,我确实将 CancellationToken 传递给同步方法,希望它能够在超时时兑现取消请求。我的解决方案还返回一个任务,然后可以根据我的调用代码的需要等待等。
我担心的是这段代码会为每个正在监控的方法创建两个任务。我认为 TPL 会很好地处理这一点,但我想确认一下。
这有意义吗?有一个更好的方法吗?
private Task TimeoutSyncMethod( Action<CancellationToken> syncAction, TimeSpan timeout )
{
var cts = new CancellationTokenSource();
var outer = Task.Run( () =>
{
try
{
//Start the synchronous method - passing it a cancellation token
var inner = Task.Run( () => syncAction( cts.Token ), cts.Token );
if( !inner.Wait( timeout ) )
{
//Try give the sync method a chance to abort grecefully
cts.Cancel();
//There was a timeout regardless of what the sync method does - so throw
throw new TimeoutException( "Timeout waiting for method after " + timeout );
}
}
finally
{
cts.Dispose();
}
}, cts.Token );
return outer;
}
编辑:
使用@Timothy's answer我现在正在使用它。虽然代码并没有显着减少,但它更清晰。谢谢!
private Task TimeoutSyncMethod( Action<CancellationToken> syncAction, TimeSpan timeout )
{
var cts = new CancellationTokenSource();
var inner = Task.Run( () => syncAction( cts.Token ), cts.Token );
var delay = Task.Delay( timeout, cts.Token );
var timeoutTask = Task.WhenAny( inner, delay ).ContinueWith( t =>
{
try
{
if( !inner.IsCompleted )
{
cts.Cancel();
throw new TimeoutException( "Timeout waiting for method after " + timeout );
}
}
finally
{
cts.Dispose();
}
}, cts.Token );
return timeoutTask;
}