更新Demo

2025-08-14 16:51:28 +00:00 · 2023-12-20 12:14:27 +08:00
parent f64bb37706
commit 17a5d7425c
847 changed files with 106095 additions and 0 deletions
--- a/UnityProject/Packages/UniTask/Runtime/Linq/UnityExtensions/EveryUpdate.cs
+++ b/UnityProject/Packages/UniTask/Runtime/Linq/UnityExtensions/EveryUpdate.cs
@@ -0,0 +1,77 @@
+using System.Threading;
+
+namespace Cysharp.Threading.Tasks.Linq
+{
+    public static partial class UniTaskAsyncEnumerable
+    {
+        public static IUniTaskAsyncEnumerable<AsyncUnit> EveryUpdate(PlayerLoopTiming updateTiming = PlayerLoopTiming.Update)
+        {
+            return new EveryUpdate(updateTiming);
+        }
+    }
+
+    internal class EveryUpdate : IUniTaskAsyncEnumerable<AsyncUnit>
+    {
+        readonly PlayerLoopTiming updateTiming;
+
+        public EveryUpdate(PlayerLoopTiming updateTiming)
+        {
+            this.updateTiming = updateTiming;
+        }
+
+        public IUniTaskAsyncEnumerator<AsyncUnit> GetAsyncEnumerator(CancellationToken cancellationToken = default)
+        {
+            return new _EveryUpdate(updateTiming, cancellationToken);
+        }
+
+        class _EveryUpdate : MoveNextSource, IUniTaskAsyncEnumerator<AsyncUnit>, IPlayerLoopItem
+        {
+            readonly PlayerLoopTiming updateTiming;
+            CancellationToken cancellationToken;
+
+            bool disposed;
+
+            public _EveryUpdate(PlayerLoopTiming updateTiming, CancellationToken cancellationToken)
+            {
+                this.updateTiming = updateTiming;
+                this.cancellationToken = cancellationToken;
+
+                TaskTracker.TrackActiveTask(this, 2);
+                PlayerLoopHelper.AddAction(updateTiming, this);
+            }
+
+            public AsyncUnit Current => default;
+
+            public UniTask<bool> MoveNextAsync()
+            {
+                // return false instead of throw
+                if (disposed || cancellationToken.IsCancellationRequested) return CompletedTasks.False;
+
+                completionSource.Reset();
+                return new UniTask<bool>(this, completionSource.Version);
+            }
+
+            public UniTask DisposeAsync()
+            {
+                if (!disposed)
+                {
+                    disposed = true;
+                    TaskTracker.RemoveTracking(this);
+                }
+                return default;
+            }
+
+            public bool MoveNext()
+            {
+                if (disposed || cancellationToken.IsCancellationRequested)
+                {
+                    completionSource.TrySetResult(false);
+                    return false;
+                }
+
+                completionSource.TrySetResult(true);
+                return true;
+            }
+        }
+    }
+}