前言
假设有如下代码,功能是首先从缓存获取数据,如果没有命中缓存,则直接从数据库获取:
var data = await GetFromCache();
if (data is null)
{data = await GetFromDB();
}
对于获取缓存数据,我们需要限制一下GetFromCache
方法的执行时间,比如 3 秒,超过这个时间,可以认为拿不到缓存数据,应该直接从数据库取值。
那么,能否在不修改GetFromCache
方法的前提下,让 Task 超时时自动结束执行?
思路
虽然不能把普通Task
变成一个可超时的 Task
,但是我们可以定时检查任务状态。如果到达超时时间,原始 Task
仍未执行完成,则跳过该 Task
,继续执行后续代码。
实现
为 Task
创建扩展方法:
public static async Task<T> Timeout<T>(this Task<T> task, int milliseconds)
{var now = DateTime.Now.AddMilliseconds(milliseconds);while (DateTime.Now < now){if (task.IsCompleted){return await task;}await Task.Delay(100);}return default(T);
}
使用也很简单:
var data = await GetFromCache().Timeout(1000);if (data is null)
{data = await GetFromDB();
}
需要注意的是,我们仅仅是不再 await
原始 Task
执行完成,并不代表真正地终止了它的运行,原始 Task
还是会继续执行,只是我们不再关心它的返回结果而已。
结论
今天,我们通过实现 Timeout
扩展方法,为异步任务添加了超时功能。
添加微信号【MyIO666】,邀你加入技术交流群