Add await support for OptionAsync

louthy · Oct 1, 2023 · 3283992 · 3283992
1 parent bce0a16
commit 3283992
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/LanguageExt.Core/Monads/Alternative Value Monads/Option/OptionAsync/OptionAsync.cs b/LanguageExt.Core/Monads/Alternative Value Monads/Option/OptionAsync/OptionAsync.cs
@@ -139,6 +139,14 @@ public OptionAsync(IEnumerable<A?> option)
                 : null;
         }
 
+        /// <summary>
+        /// Custom awaiter that turns an OptionAsync into an Option
+        /// </summary>
+        public TaskAwaiter<Option<A>> GetAwaiter()
+            => Data.Map(d =>
+                d.IsSome ? Option<A>.Some(d.Value!) : Option<A>.None)
+            .GetAwaiter();
+
         /// <summary>
         /// Implicit conversion operator from A to Option<A>
         /// </summary>

diff --git a/LanguageExt.Tests/OptionAsyncTests.cs b/LanguageExt.Tests/OptionAsyncTests.cs
@@ -300,6 +300,13 @@ public async Task MOptionAsync_BiFoldAsync_OverloadAsyncAsync_IsNone_DelegateNon
             Assert.True(result);
         }
 
+        [Fact]
+        public async Task AwaitSomeCaseTest()
+            => Assert.Equal(Some(10), await OptionAsync<int>.Some(10));
+
+        [Fact]
+        public async Task AwaitNoneCaseTest()
+            => Assert.Equal(None, await OptionAsync<int>.None);
         // Not valuable any more 
         //[Fact]
         //public async void SequenceFlip()