Code cleanup

dotnet · Jan 23, 2025 · 5e8f102 · 5e8f102
1 parent 2113e53
commit 5e8f102
Showing 1 changed file with 5 additions and 19 deletions.
diff --git a/src/DotNext.IO/IO/PoolingBufferedStream.cs b/src/DotNext.IO/IO/PoolingBufferedStream.cs
@@ -213,27 +213,13 @@ public void Write()
     /// <returns>The task representing asynchronous execution of the operation.</returns>
     /// <exception cref="OperationCanceledException">The operation has been canceled.</exception>
     /// <exception cref="ObjectDisposedException">The stream is disposed.</exception>
-    public ValueTask WriteAsync(CancellationToken token = default)
+    public async ValueTask WriteAsync(CancellationToken token = default)
     {
-        ValueTask task;
-        if (stream is null)
-        {
-            task = new(DisposedTask);
-        }
-        else if (stream.CanWrite)
-        {
-            task = WriteCoreAsync(token);
-        }
-        else
-        {
-            task = ValueTask.FromException(new NotSupportedException());
-        }
-
-        return task;
-    }
+        AssertState();
+        ThrowIfDisposed();
+        if (!stream.CanWrite)
+            throw new NotSupportedException();
 
-    private async ValueTask WriteCoreAsync(CancellationToken token)
-    {
         await WriteCoreAsync(out var isWritten, token).ConfigureAwait(false);
         if (isWritten)
             Reset();