add AsValueTask extension method to cast from ValueTask<T> to ValueTask

Fresa · Jan 21, 2022 · 10df6ef · 10df6ef
1 parent 8697fef
commit 10df6ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/Kafka.TestFramework/NetworkStream.cs b/src/Kafka.TestFramework/NetworkStream.cs
@@ -38,10 +38,8 @@ public override void Write(byte[] buffer, int offset, int count)
             throw new System.NotImplementedException();
         }
 
-        public override ValueTask WriteAsync(ReadOnlyMemory<byte> buffer, CancellationToken cancellationToken = new CancellationToken())
-        {
-            return new ValueTask(_networkClient.SendAsync(buffer, cancellationToken).AsTask());
-        }
+        public override ValueTask WriteAsync(ReadOnlyMemory<byte> buffer, CancellationToken cancellationToken = new CancellationToken()) => 
+            _networkClient.SendAsync(buffer, cancellationToken).AsValueTask();
 
         public override bool CanRead => false;
         public override bool CanSeek => false;

diff --git a/src/Kafka.TestFramework/ValueTaskExtensions.cs b/src/Kafka.TestFramework/ValueTaskExtensions.cs
@@ -13,5 +13,8 @@ internal static Task WhenAllAsync(
                         valueTask
                             => !valueTask.IsCompletedSuccessfully)
                     .Select(valueTask => valueTask.AsTask()));
+
+        internal static ValueTask AsValueTask<T>(this ValueTask<T> valueTask) => 
+            valueTask.IsCompletedSuccessfully ? default : new ValueTask(valueTask.AsTask());
     }
 }