我们在2019-12-1-使用SemaphoreSlim实现异步等待 - huangtengxiao给大家介绍了信号量的异步等待使用方法。可惜的是.NET的ManualResetEvent和ManualResetEventSlim目前都没有提供异步的等待方法。所以我们自己实现一个
这里是我们创建的异步版本的ManualResetEvent。
实际上这里的本质就是使用了一个TaskCompletionSource。
TaskCompletionSource是一个task的封装源,代表了一个完成的task。可以通过SetResult方法标识task的完成情况。
public class ManualResetEventAsync
{
/// <summary>
/// 提供一个信号初始值,确定是否有信号
/// </summary>
/// <param name="initialState">true为有信号,所有等待可以直接通过</param>
public ManualResetEventAsync(bool initialState)
{
_source = new TaskCompletionSource<bool>();
if (initialState)
{
_source.SetResult(true);
}
}
/// <summary>
/// 异步等待一个信号,需要await
/// </summary>
/// <returns></returns>
public Task WaitOneAsync()
{
lock (_locker)
{
return _source.Task;
}
}
/// <summary>
/// 设置一个信号量,所有等待获得信号
/// </summary>
public void Set()
{
lock (_locker)
{
_source.SetResult(true);
}
}
/// <summary>
/// 设置一个信号量,所有wait等待
/// </summary>
public void Reset()
{
lock (_locker)
{
if (!_source.Task.IsCompleted)
{
return;
}
_source = new TaskCompletionSource<bool>();
}
}
private TaskCompletionSource<bool> _source;
private readonly object _locker = new object();
}
分析下上述代码。ManualResetEventAsync最重要的三个方法就是Set,Reset,和WaitOneAsync
WaitOneAsync很简单就是等待TaskCompletionSource.Task。如果TaskCompletionSource未标记完成则会等待,如果标记完成,则会继续执行
Set也很简单就是调用TaskCompletionSource.SetResult方法将其标记为完成,这样所有正在等待的task可以继续执行。
最复杂的Reset方法。由于每个TaskCompletionSource的实例只能标识一次已完成,所以我们在每次重置时,需要重新实例化一个对象,进行替换。
当然还有很重要的一点是这里涉及多线程的判断,赋值操作,所以所有的这些方法必须放置在同步区进行。
参考文档:
本文会经常更新,请阅读原文: https://xinyuehtx.github.io/post/%E5%AE%9E%E7%8E%B0%E4%B8%80%E7%A7%8D%E5%BC%82%E6%AD%A5%E7%89%88%E6%9C%AC%E7%9A%84ManualResetEvent.html ,以避免陈旧错误知识的误导,同时有更好的阅读体验。
本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。欢迎转载、使用、重新发布,但务必保留文章署名黄腾霄(包含链接: https://xinyuehtx.github.io ),不得用于商业目的,基于本文修改后的作品务必以相同的许可发布。如有任何疑问,请 与我联系 。