当前位置：首页 > news >正文

将 async/await 异步代码转换为安全的不会死锁的同步代码

news 来源：原创 2024/5/9 22:48:54

在 async/await 异步模型（即 TAP Task-based Asynchronous Pattern）出现以前，有大量的同步代码存在于代码库中，以至于这些代码全部迁移到 async/await 可能有些困难。这里就免不了将一部分异步代码修改为同步代码。然而传统的迁移方式存在或多或少的问题。本文将总结这些传统方法的坑，并推出一款异步转同步的新方法，解决传统方法的这些坑。

背景问题和传统方法

为什么有些方法不容易迁移到 async/await？
- 参见微软的博客 async/await 最佳实践 Async/Await - Best Practices in Asynchronous Programming。如果某个方法从同步方法修改为异步方法（例如从 var content = file.Read() 修改为 var content = await file.ReadAsync()），那么调用此方法的整个调用链全部都要改成 async/await 才能让返回值在调用链中成功传递。
传统的异步转同步的方法有哪些？有什么坑？
- 参见我的好朋友林德熙的博客 win10 uwp 异步转同步。文章里使用 Task.Wait() 或者 Task.Result 来获取异步方法的返回值。
- 这种方法会阻塞调用线程。如果调用线程是 UI 线程，那么 UI 将会无响应；更严重地，如果 UI 线程使用 DispatcherSynchronizationContext（参见我的另一篇文章 DispatcherSynchronizationContext - walterlv）进行线程上下文的同步，那么极有可能会造成死锁（参见我的另一篇文章使用 Task.Wait()？立刻死锁（deadlock））。

安全的方法

传统方法的坑在于 UI 线程无响应和死锁问题。既要解决无响应问题，又要阻塞调用方，可选的方法就是 Windows 消息循环了。在使用消息循环时还要避免使用 async/await 的同步上下文（SynchronizationContext），这样才能避免 UI 线程的死锁问题。

所以，我考虑使用 PushFrame 来阻塞当前线程并创建一个新的消息循环。使用 Task.ContinueWith 来恢复阻塞，而不使用 Task 中默认同步所采用的同步上下文。

代码如下：

/// <summary>
/// 通过 PushFrame（进入一个新的消息循环）的方式来同步等待一个必须使用 await 才能等待的异步操作。
/// 由于使用了消息循环，所以并不会阻塞 UI 线程。<para/>
/// 此方法适用于将一个 async/await 模式的异步代码转换为同步代码。<para/>
/// </summary>
/// <remarks>
/// 此方法适用于任何线程，包括 UI 线程、非 UI 线程、STA 线程、MTA 线程。
/// </remarks>
/// <typeparam name="TResult">
/// 异步方法返回值的类型。
/// 我们认为只有包含返回值的方法才会出现无法从异步转为同步的问题，所以必须要求异步方法返回一个值。
/// </typeparam>
/// <param name="task">异步的带有返回值的任务。</param>
/// <returns>异步方法在同步返回过程中的返回值。</returns>
public static TResult AwaitByPushFrame<TResult>(Task<TResult> task)
{
    if (task == null) throw new ArgumentNullException(nameof(task));
    Contract.EndContractBlock();

    var frame = new DispatcherFrame();
    task.ContinueWith(t =>
    {
        frame.Continue = false;
    });
    Dispatcher.PushFrame(frame);
    return task.Result;
}

▲ 这就是全部代码了，仅适用于 Windows 平台（如果使用 .NET Core，需要额外的 Windows 兼容 NuGet 包）

新方法的适用范围和优劣

事实上，虽然我们使用了消息循环，但其实也适用于控制台程序，适用于各种各样奇奇怪怪的线程 —— 无论是 UI 线程还是非 UI 线程，无论是 STA 还是 MTA。

例如，我们现在在一个 MTA 线程模型的控制台程序中试用一下：

namespace Walterlv.Demo
{
    class Program
    {
        static void Main(string[] args)
        {
            Console.Title = "walterlv's demo";
            var foo = Foo();
            var result = AwaitByPushFrame(foo);
            Console.WriteLine($"输入的字符串为：{result}");
            Console.ReadKey();
        }

        private static async Task<string> Foo()
        {
            Console.WriteLine("请稍后……");
            await Task.Delay(1000);
            Console.Write("请输入：");
            var line = Console.ReadLine();
            Console.WriteLine("正在处理……");
            await Task.Run(() =>
            {
                // 模拟耗时的操作。
                Thread.Sleep(1000);
            });
            return line;
        }
    }
}