From a2783d3c8adb3b7c2be9154322cd7a518588e114 Mon Sep 17 00:00:00 2001
From: neuecc <ils@neue.cc>
Date: Mon, 15 Jun 2020 12:29:46 +0900
Subject: [PATCH] yes meta

---
 .../Plugins/UniTask/Runtime/Linq/Subscribe.cs | 50 +++++++++++++++++++
 .../UniTask/Runtime/Linq/Subscribe.cs.meta    | 11 ++++
 2 files changed, 61 insertions(+)
 create mode 100644 src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs.meta

diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs b/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs
index 31bb86b..9643b06 100644
--- a/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs
+++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs
@@ -29,6 +29,16 @@ namespace Cysharp.Threading.Tasks.Linq
             return cts;
         }
 
+        public static IDisposable Subscribe<TSource>(this IUniTaskAsyncEnumerable<TSource> source, Func<TSource, CancellationToken, UniTaskVoid> action)
+        {
+            Error.ThrowArgumentNullException(source, nameof(source));
+            Error.ThrowArgumentNullException(action, nameof(action));
+
+            var cts = new CancellationTokenDisposable();
+            Subscribes.SubscribeCore(source, action, Subscribes.NopError, Subscribes.NopCompleted, cts.Token).Forget();
+            return cts;
+        }
+
         public static void Subscribe<TSource>(this IUniTaskAsyncEnumerable<TSource> source, Action<TSource> action, CancellationToken cancellationToken)
         {
             Error.ThrowArgumentNullException(source, nameof(source));
@@ -45,6 +55,14 @@ namespace Cysharp.Threading.Tasks.Linq
             Subscribes.SubscribeCore(source, action, Subscribes.NopError, Subscribes.NopCompleted, cancellationToken).Forget();
         }
 
+        public static void Subscribe<TSource>(this IUniTaskAsyncEnumerable<TSource> source, Func<TSource, CancellationToken, UniTaskVoid> action, CancellationToken cancellationToken)
+        {
+            Error.ThrowArgumentNullException(source, nameof(source));
+            Error.ThrowArgumentNullException(action, nameof(action));
+
+            Subscribes.SubscribeCore(source, action, Subscribes.NopError, Subscribes.NopCompleted, cancellationToken).Forget();
+        }
+
         // OnNext, OnError
 
         public static IDisposable Subscribe<TSource>(this IUniTaskAsyncEnumerable<TSource> source, Action<TSource> onNext, Action<Exception> onError)
@@ -234,6 +252,38 @@ namespace Cysharp.Threading.Tasks.Linq
             }
         }
 
+        public static async UniTaskVoid SubscribeCore<TSource>(IUniTaskAsyncEnumerable<TSource> source, Func<TSource, CancellationToken, UniTaskVoid> onNext, Action<Exception> onError, Action onCompleted, CancellationToken cancellationToken)
+        {
+            var e = source.GetAsyncEnumerator(cancellationToken);
+            try
+            {
+                while (await e.MoveNextAsync())
+                {
+                    onNext(e.Current, cancellationToken).Forget();
+                }
+                onCompleted();
+            }
+            catch (Exception ex)
+            {
+                if (onError == NopError)
+                {
+                    UniTaskScheduler.PublishUnobservedTaskException(ex);
+                    return;
+                }
+
+                if (ex is OperationCanceledException) return;
+
+                onError(ex);
+            }
+            finally
+            {
+                if (e != null)
+                {
+                    await e.DisposeAsync();
+                }
+            }
+        }
+
         public static async UniTaskVoid SubscribeCore<TSource>(IUniTaskAsyncEnumerable<TSource> source, IObserver<TSource> observer, CancellationToken cancellationToken)
         {
             var e = source.GetAsyncEnumerator(cancellationToken);
diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs.meta b/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs.meta
new file mode 100644
index 0000000..ea83567
--- /dev/null
+++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/Linq/Subscribe.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 263479eb04c189741931fc0e2f615c2d
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: