WinForms遗留代码中GetAwaiter()与GetResult()的优化方案探讨

问题背景 #

我遇到一段WinForms遗留代码，核心逻辑如下：

private static bool _connected = false;
private static bool _closing = false;

static void Main(string[] args)
{
    // 其他操作
    ThreadPool.QueueUserWorkItem(ThreadProc, null);
    // 其他操作

    while (!_closing)
    {
        if (_connected)
        {
            break;
        }
        // 其他操作
        Application.DoEvents();
        Thread.Sleep(100);
        // 其他操作
    }

    if (_connected)
    {
        // 最终执行
        Application.Run(new MyForm());
    }
}

private static void ThreadProc(object obj)
{
    while (!_closing)
    {
        // 其他操作
        if (!Method2())
        {
            // 错误处理
            _closing = true;
            return;
        }
        else
        {
            /// 成功
            break;
        }
    }

    _connected = true;
}

private static async Task<SomeData> Method1()
{
    // 等待启动画面加载完成
    while (Application.OpenForms.Count == 0) Thread.Sleep(100);

    // 在Windows GUI线程执行
    SomeData result = await (Application.OpenForms[0].Invoke(new Func<Task<SomeData>>(() => { SomethingAsync())));
    return result;
}

private static bool Method2()
{
    TaskAwaiter<SomeData> awaiter = Method1().GetAwaiter();
    while (!awaiter.IsCompleted)
    {
        // 避免阻塞Windows事件循环
        Application.DoEvents();
        Thread.Sleep(100);
    }

    var result = awaiter.GetResult();
    if (result == null) { return false; } // 错误处理

    // 处理返回结果
    return true;
}

我了解到使用GetAwaiter().GetResult()存在弊端，正在寻找更优方案。原代码这样实现是为了在Method1()执行时不阻塞应用事件循环，通过轮询awaiter的IsCompleted状态获取结果。

疑问 #

• 这种GetAwaiter()和GetResult()的用法是否合理？
• 是否有更好的实现方案？

这段遗留代码目前可以正常运行，但我不确定它是否是合理的解决方案，还是存在潜在风险。

补充说明 #

• Method2()由ThreadPool.QueueUserWorkItem()创建的线程调用，该线程在Main()方法中启动
• Application.Run()要等到该线程终止后才会调用，且运行在进入Main()的同一线程
• 代码目的：应用启动时创建线程执行用户登录认证，认证通过后使用JWT连接服务器进程；Main()线程循环等待连接线程成功或出错，同时调用Application.DoEvents()

回答 #

一、原用法的合理性与潜在风险 #

这种GetAwaiter().GetResult()的用法能实现功能但存在明显缺陷，不算合理方案：

• 死锁隐患：当前代码因Method2在ThreadPool线程执行暂时无死锁，但后续若调整为GUI线程调用Method2，GetResult()会阻塞GUI线程，而SomethingAsync可能依赖GUI线程完成后续操作，直接触发死锁。
• 轮询低效：while(!awaiter.IsCompleted)加Thread.Sleep(100)的轮询方式，既浪费CPU资源，又会导致界面响应延迟（100ms间隔会让操作不流畅）。
• 异常处理不规范：GetResult()会直接抛出未捕获的异步异常，若没有完善的错误处理逻辑，可能导致程序崩溃。
• Application.DoEvents()副作用：频繁调用会打乱消息队列，引发重复触发事件、控件状态异常等问题，增加调试难度。

二、更优的实现方案 #

利用异步/await原生特性，结合TaskCompletionSource替代轮询与阻塞，同时避免DoEvents()滥用：

1. 支持异步Main的.NET Framework 4.5+版本方案

private static TaskCompletionSource<bool> _authCompletionSource = new TaskCompletionSource<bool>();

static async Task Main(string[] args)
{
    // 启动启动画面（确保OpenForms有内容）
    var splashForm = new SplashForm();
    splashForm.Show();

    // 异步执行认证逻辑，不阻塞主线程
    _ = Task.Run(async () =>
    {
        try
        {
            var result = await Method1Async();
            // 处理返回结果
            _authCompletionSource.SetResult(result != null);
        }
        catch (Exception ex)
        {
            // 错误处理
            _authCompletionSource.SetException(ex);
        }
    });

    // 异步等待认证完成，自动保持界面响应
    bool authSuccess = await _authCompletionSource.Task;

    splashForm.Close();

    if (authSuccess)
    {
        Application.Run(new MyForm());
    }
}

private static async Task<SomeData> Method1Async()
{
    // 异步等待启动画面加载完成
    while (Application.OpenForms.Count == 0)
        await Task.Delay(100);

    // 在GUI线程执行异步操作并等待结果
    var invokeTask = (Task<SomeData>)Application.OpenForms[0].Invoke(new Func<Task<SomeData>>(SomethingAsync));
    return await invokeTask;
}

2. 关键优化点

• 异步Main方法：.NET Framework 4.5+支持，无需手动循环等待，await会自然挂起主线程，同时WinForms会自动处理消息循环，保证界面响应。
• TaskCompletionSource：替代全局变量_connected和_closing，安全传递认证结果，避免线程安全问题。
• Task.Run替代ThreadPool.QueueUserWorkItem：更贴合现代异步编程范式，便于处理异步逻辑。
• Task.Delay替代Thread.Sleep：异步等待不阻塞线程，资源利用率更高。
• Invoke+await：在异步方法中调用GUI线程操作，避免阻塞工作线程，同时优雅处理结果。

3. 兼容.NET Framework 4.0及以下版本方案

如果项目不支持异步Main，可改用以下方式：

private static TaskCompletionSource<bool> _authCompletionSource = new TaskCompletionSource<bool>();

static void Main(string[] args)
{
    var splashForm = new SplashForm();
    splashForm.Show();

    _ = Task.Run(async () =>
    {
        try
        {
            var result = await Method1Async();
            _authCompletionSource.SetResult(result != null);
        }
        catch
        {
            _authCompletionSource.SetResult(false);
        }
    });

    // 循环等待认证完成，缩短Sleep间隔提升响应性
    while (!_authCompletionSource.Task.IsCompleted)
    {
        Application.DoEvents();
        Thread.Sleep(10);
    }

    bool authSuccess = _authCompletionSource.Task.Result;
    splashForm.Close();

    if (authSuccess)
    {
        Application.Run(new MyForm());
    }
}

// Method1Async同上面的实现

该方案保留循环等待，但用TaskCompletionSource替代全局变量，减少Sleep间隔提升界面流畅度，比原代码更可靠。

内容的提问来源于stack exchange，提问作者rhoonah