From 09837002eaca46951780fd7312026dc2109e4ec3 Mon Sep 17 00:00:00 2001
From: Jon Manning <jon@secretlab.com.au>
Date: Fri, 6 Sep 2024 22:56:13 +1000
Subject: [PATCH] Add wrapper for UniTask.WaitUntilCanceled

---
 Runtime/AsyncHelpers.cs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Runtime/AsyncHelpers.cs b/Runtime/AsyncHelpers.cs
index d26a60c4..94104a99 100644
--- a/Runtime/AsyncHelpers.cs
+++ b/Runtime/AsyncHelpers.cs
@@ -34,10 +34,14 @@ public static partial class YarnAsync
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         public async static YarnTask WaitUntilCanceled(System.Threading.CancellationToken token)
         {
+#if USE_UNITASK
+            await YarnTask.WaitUntilCanceled(token);
+#else
             while (token.IsCancellationRequested == false)
             {
                 await YarnTask.Yield();
             }
+#endif
         }
 
 #if !USE_UNITASK