UniTask中避免忘记TrySetResult导致程序卡死的解决方案

在使用Cysharp的UniTask库进行异步编程时，开发者可能会遇到一个常见问题：当使用UniTaskCompletionSource创建自定义异步任务时，如果忘记调用TrySetResult()方法，会导致程序永久卡在await状态。本文将深入分析这个问题，并提供几种实用的解决方案。

问题分析

UniTaskCompletionSource是UniTask中用于手动控制任务完成状态的核心组件。典型的使用场景是：

public static UniTask GetUTCSUniTask(Action<UniTaskCompletionSource> callback)
{
    var src = GetUTCS();
    callback?.Invoke(src);
    return src.Task;
}

这种模式的问题在于，如果callback中忘记调用src.TrySetResult()，那么等待这个UniTask的代码将永远无法继续执行，导致程序"卡死"。

解决方案

1. 使用超时机制

最直接的解决方案是添加超时控制，确保即使忘记设置结果，任务也不会无限期等待：

public static UniTask GetUTCSUniTask(Action<UniTaskCompletionSource> callback)
{
    var src = GetUTCS();
    if (callback != null)
    {
        callback?.Invoke(src);
        UniTask.Delay(2000).ContinueWith(() =>
        {
            if (!src.Task.Status.IsCompleted())
            {
                src.TrySetException(new TimeoutException("操作超时"));
            }
        }).Forget();
    }
    else
    {
        src.TrySetResult();
    }
    return src.Task;
}

这种方法通过2秒后检查任务状态，如果仍未完成则抛出超时异常。

2. 使用CancellationToken

更优雅的方式是结合CancellationToken实现超时控制：

public static UniTask GetUTCSUniTask(Action<UniTaskCompletionSource> callback)
{
    var src = GetUTCS();
    var cts = new CancellationTokenSource();
    cts.CancelAfterSlim(TimeSpan.FromSeconds(2));
    callback?.Invoke(src);
    return src.Task.AttachExternalCancellation(cts.Token);
}

当超过指定时间后，任务会自动取消，避免无限等待。

3. 使用异步回调模式

重构API设计，强制回调必须是异步的，确保开发者必须处理完成状态：

public static async UniTask GetUTCSUniTask(Func<UniTaskCompletionSource, UniTask> callback)
{
    var src = GetUTCS();
    if (callback != null)
    {
        await callback.Invoke(src);
    }
    else
    {
        src.TrySetResult();
    }
    await src.Task;
}

这种设计更符合异步编程的最佳实践，减少了忘记设置结果的可能性。

最佳实践建议

1. 始终设置超时：为所有手动创建的UniTask添加合理的超时机制
2. 使用UniTask的调试工具：在开发阶段启用UniTask的任务追踪功能，帮助发现未完成的任务
3. 代码审查：建立团队规范，对UniTaskCompletionSource的使用进行重点检查
4. 文档注释：在API文档中明确说明需要调用TrySetResult的责任

通过以上方法，可以显著减少因忘记设置任务结果而导致的程序卡死问题，提高异步代码的健壮性和可靠性。