更新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/ElementAt.cs
+++ b/UnityProject/Packages/UniTask/Runtime/Linq/ElementAt.cs
@@ -0,0 +1,58 @@
+using Cysharp.Threading.Tasks.Internal;
+using System;
+using System.Threading;
+
+namespace Cysharp.Threading.Tasks.Linq
+{
+    public static partial class UniTaskAsyncEnumerable
+    {
+        public static UniTask<TSource> ElementAtAsync<TSource>(this IUniTaskAsyncEnumerable<TSource> source, int index, CancellationToken cancellationToken = default)
+        {
+            Error.ThrowArgumentNullException(source, nameof(source));
+
+            return ElementAt.ElementAtAsync(source, index, cancellationToken, false);
+        }
+
+        public static UniTask<TSource> ElementAtOrDefaultAsync<TSource>(this IUniTaskAsyncEnumerable<TSource> source, int index, CancellationToken cancellationToken = default)
+        {
+            Error.ThrowArgumentNullException(source, nameof(source));
+
+            return ElementAt.ElementAtAsync(source, index, cancellationToken, true);
+        }
+    }
+
+    internal static class ElementAt
+    {
+        public static async UniTask<TSource> ElementAtAsync<TSource>(IUniTaskAsyncEnumerable<TSource> source, int index, CancellationToken cancellationToken, bool defaultIfEmpty)
+        {
+            var e = source.GetAsyncEnumerator(cancellationToken);
+            try
+            {
+                int i = 0;
+                while (await e.MoveNextAsync())
+                {
+                    if (i++ == index)
+                    {
+                        return e.Current;
+                    }
+                }
+
+                if (defaultIfEmpty)
+                {
+                    return default;
+                }
+                else
+                {
+                    throw Error.ArgumentOutOfRange(nameof(index));
+                }
+            }
+            finally
+            {
+                if (e != null)
+                {
+                    await e.DisposeAsync();
+                }
+            }
+        }
+    }
+}