C# async / await 任务超时处理
在之前的帖子中,介绍了 async / await 的用法,所以新的问题又来了,如果调用异步方法后,异步方法一直不给返回结果怎么办?这时线程还会一直卡住无法动弹,如果你用的是多线程(Task.Run 或者 Thread)可能感受不到,如果在 Winform 的 UI 线程下执行你就知道了,界面卡住完全无法操作,要解决这个问题,这就知道怎么取消任务了
目录
一、需求
在之前的帖子中,介绍了 async / await 的用法,所以新的问题又来了,如果调用异步方法后,异步方法一直不给返回结果怎么办?这时线程还会一直卡住无法动弹,如果你用的是多线程(Task.Run 或者 Thread)可能感受不到,如果在 Winform 的 UI 线程下执行你就知道了,界面卡住完全无法操作,要解决这个问题,这就知道怎么取消任务了,添加一个任务后,要固定时间内返回结果,如果超时就取消任务,并且在多个任务同时执行的时候,依然按顺序来执行,这就是本文章要实现的效果。
二、Task取消任务
在 C# 以前的2.0 等版本中,线程是可以强制终止的,在后面的版本中,就不允许强制去结束线程的,后面微软就提供了一个 CancellationTokenSource 相关的接口开源取消任务。于是我搜索了大量的帖子,一看全是各种抄袭,相互抄来抄去的,完全是浪费时间。
网上的代码(不推荐):
using System;
using System.Threading;
using System.Threading.Tasks;
namespace 超时测试
{
internal class Program
{
static CancellationTokenSource source = new CancellationTokenSource();
static void Main(string[] args)
{
Task.Run(() => Test());
Thread.Sleep(1000);
Console.WriteLine("取消任务");
source.Cancel();
Console.ReadKey();
}
static void Test()
{
try
{
int index = 0;
for (int i = 0; i < 10; i++)
{
Thread.Sleep(200);
source.Token.ThrowIfCancellationRequested();
Console.WriteLine("执行次数:{0}", index++);
}
}
catch (OperationCanceledException)
{
}
}
}
}
运行:
上面代码中,使用 CancellationTokenSource 感觉就是用错了地方,几乎没起到对应的作用。
换一个写法:
using System;
using System.Threading;
using System.Threading.Tasks;
namespace 超时测试
{
internal class Program
{
static CancellationTokenSource source = new CancellationTokenSource();
static void Main(string[] args)
{
Task.Run(() => Test());
Thread.Sleep(1000);
Console.WriteLine("取消任务");
source.Cancel();
Console.ReadKey();
}
static void Test()
{
try
{
Thread.Sleep(3000);
Console.WriteLine("等待结束");
source.Token.ThrowIfCancellationRequested();
}
catch (OperationCanceledException)
{
}
}
}
}
任务中等待三秒,我在等待一秒后取消任务,看看结果:
在执行一秒后取消了任务,但结果 “等待结束” 却依然执行了。
如果你有一些 C# 基础,就很容易看出代码有多少问题了,问题如下:
1.CancellationTokenSource 的用法不对
ThrowIfCancellationRequested() 方法在执行后会检查 CancellationToken 是否已经有取消请求。如果有取消请求,这个方法会抛出一个 OperationCanceledException 异常,用来停止当前的操作。
但是当前延时等待,并没有使用任何的 Token ,并且在执行 source.Token.ThrowIfCancellationRequested() 这句代码的时候,等待结束这句已经执行了,所以此时取消没有任何意义。
2.catch 接收部分没有任何的代码
即使 source.Token.ThrowIfCancellationRequested() 这里抛出了异常,也是什么都看不到。
3.Thread.Sleep 不支持传入参数 source.Token
代码如下:
根据他们的例子,使用 Thread.Sleep 进行线程睡眠,为了实现超时取消的效果,在任务中必须加入 for 或者 while 循环,并且在下一轮循环中,执行 ThrowIfCancellationRequested() 方法来抛出一个异常,让代码跳转到 try catch 中继续执行,达到取消后续代码执行的目的。
根据他们的例子,可以模拟一下,换个写法,在 while 循环中加入一个判断,如果等于 true,直接跳出循环,这也可以中断任务,代码如下:
namespace 取消任务4
{
internal class Program
{
static void Main(string[] args)
{
bool isOut = false;
var task1 = Task.Run(() =>
{
for (int i = 0; i < 100; i++)
{
if (isOut) return;
Console.WriteLine("执行中" + i);
Thread.Sleep(500);
}
});
Thread.Sleep(2000);
Console.WriteLine("取消任务");
isOut= true;
Console.ReadKey();
}
}
}
运行:
同样的,上面的代码也没有什么意义,不推荐。
三、Task取消任务的回调
根据上节的写法,取消任务也是可以加入回调的,代码如下:
namespace 取消任务2
{
internal class Program
{
static CancellationTokenSource source = new CancellationTokenSource();
static void Main(string[] args)
{
var task1 = Task.Run(() =>
{
for (int i = 0; i < 100; i++)
{
source.Token.ThrowIfCancellationRequested();
Console.WriteLine("执行中" + i);
Thread.Sleep(500);
}
}, source.Token);
//在指定的毫秒数后取消task执行
source.CancelAfter(2 * 1000);
//取消任务后的回调
source.Token.Register(() =>
{
//不延迟会获取不到正确的状态
Thread.Sleep(50);
Console.WriteLine("task1状态:" + task1.Status);
Console.WriteLine("IsFaulted状态:" + task1.IsFaulted);//由于未处理的异常,任务已完成。
Console.WriteLine("IsCompleted状态:" + task1.IsCompleted);//获取一个值,该值指示任务是否已完成。
});
Console.ReadKey();
}
}
}
运行:
在这里依然是不推荐上面的写法,主要目的还是为了展示 CancellationTokenSource 的一些用法,告诉你 CancellationTokenSource 有这么一个东西,但是,当前的案例绝对不是一个好的案例。
四、Task超时处理的实现
在 C# 的开发中,我强烈推荐多使用 “异步” 去取代 Thread,使用异步实现超时处理非常的简单。
代码:
using System;
using System.Threading;
using System.Threading.Tasks;
namespace 超时测试
{
internal class Program
{
static CancellationTokenSource Cancellation = new CancellationTokenSource();
static void Main(string[] args)
{
//设置超时的时间
Cancellation.CancelAfter(TimeSpan.FromSeconds(1));
Task.Run(async () =>
{
try
{
Console.WriteLine("方法执行开始");
await Task.Delay(3000, Cancellation.Token);
Console.WriteLine("方法执行结束");
}
catch (OperationCanceledException)
{
Console.WriteLine("取消操作");
}
});
Console.ReadKey();
}
}
}
运行:
在上面我们设置的超时时间是1秒,等待为3秒,超时后,就自动抛出异常了,所以 Console.WriteLine("方法执行结束") 这句代码并没有执行,这就是我们想要的效果了。
接下来把超时时间改为3秒,等待时间改为1秒,看看效果:
using System;
using System.Threading;
using System.Threading.Tasks;
namespace 超时测试
{
internal class Program
{
static CancellationTokenSource Cancellation = new CancellationTokenSource();
static void Main(string[] args)
{
//设置超时的时间
Cancellation.CancelAfter(TimeSpan.FromSeconds(3));
Task.Run(async () =>
{
try
{
Console.WriteLine("方法执行开始");
await Task.Delay(1000, Cancellation.Token);
Console.WriteLine("方法执行结束");
}
catch (OperationCanceledException)
{
Console.WriteLine("取消操作");
}
});
Console.ReadKey();
}
}
}
运行:
那么超时取消代码执行的效果就实现了。
五、其他的写法
下面代码是各种超时取消的写法,我全部放一个类里面了,根据自己的需求看看那些是可以用的,具体我也没怎么仔细的测试,可能会有 bug,不要复制了就直接拿去用。
代码:
namespace 异步编程1
{
public static class Lib1
{
public static async Task<TResult?> TimeoutAfter<TResult>(this Task<TResult> task, int timeout)
{
using (var cancelToken = new CancellationTokenSource())
{
Task completedTask = await Task.WhenAny(task, Task.Delay(timeout, cancelToken.Token));
if (completedTask == task)
{
cancelToken.Cancel();
return await task;
}
else
{
// 超时处理
Console.WriteLine("超时了");
return default;
}
}
}
public static async Task<bool> OnTimeout<T>(T t, Action<T> action, int waitms) where T : Task
{
if (!(await Task.WhenAny(t, Task.Delay(waitms)) == t))
{
action(t);
return true;
}
else
{
return false;
}
}
public static async Task<bool> TryWithTimeoutAfter<TResult>(this Task<TResult> task, TimeSpan timeout, Action<TResult> successor)
{
using var timeoutCancellationTokenSource = new CancellationTokenSource();
var completedTask = await Task.WhenAny(task, Task.Delay(timeout, timeoutCancellationTokenSource.Token)).ConfigureAwait(continueOnCapturedContext: false);
if (completedTask == task)
{
timeoutCancellationTokenSource.Cancel();
// propagate exception rather than AggregateException, if calling task.Result.
var result = await task.ConfigureAwait(continueOnCapturedContext: false);
successor(result);
return true;
}
else
return false;
}
public static async Task<bool> BeforeTimeout(Task task, int millisecondsTimeout)
{
if (task.IsCompleted) return true;
if (millisecondsTimeout == 0) return false;
if (millisecondsTimeout == Timeout.Infinite)
{
await Task.WhenAll(task);
return true;
}
var tcs = new TaskCompletionSource<object>();
using (var timer = new Timer(state => ((TaskCompletionSource<object>)state).TrySetCanceled(), tcs, millisecondsTimeout, Timeout.Infinite))
{
return await Task.WhenAny(task, tcs.Task) == task;
}
}
}
}
结束
如果这个帖子对你有所帮助,欢迎 关注 、点赞 、留言
end
更多推荐
所有评论(0)