| | | 1 | | // Licensed to the .NET Foundation under one or more agreements. |
| | | 2 | | // The .NET Foundation licenses this file to you under the MIT license. |
| | | 3 | | |
| | | 4 | | using System.IO; |
| | | 5 | | using System.Threading; |
| | | 6 | | using System.Threading.Tasks; |
| | | 7 | | |
| | | 8 | | namespace System |
| | | 9 | | { |
| | | 10 | | // Static-abstract adapter that lets a single generic implementation serve both |
| | | 11 | | // synchronous and asynchronous code paths. The sync entry point passes |
| | | 12 | | // SyncReadWriteAdapter and asserts the returned ValueTask is already completed; |
| | | 13 | | // the async entry point passes AsyncReadWriteAdapter. |
| | | 14 | | // |
| | | 15 | | // Originally used by SslStream and SmtpClient for networking I/O deduplication. |
| | | 16 | | // Extended for System.Formats.Tar (CopyToAsync, ReadExactlyAsync, DisposeAsync) |
| | | 17 | | // to unify TarReader/TarWriter/GnuSparseStream sync/async pairs. |
| | | 18 | | internal interface IReadWriteAdapter |
| | | 19 | | { |
| | | 20 | | static abstract ValueTask<int> ReadAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationToken |
| | | 21 | | static abstract ValueTask<int> ReadAtLeastAsync(Stream stream, Memory<byte> buffer, int minimumBytes, bool throw |
| | | 22 | | static abstract ValueTask ReadExactlyAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationTok |
| | | 23 | | static abstract ValueTask WriteAsync(Stream stream, ReadOnlyMemory<byte> buffer, CancellationToken cancellationT |
| | | 24 | | static abstract ValueTask CopyToAsync(Stream source, Stream destination, CancellationToken cancellationToken); |
| | | 25 | | static abstract ValueTask DisposeAsync(Stream stream); |
| | | 26 | | static abstract Task FlushAsync(Stream stream, CancellationToken cancellationToken); |
| | | 27 | | static abstract Task WaitAsync(TaskCompletionSource<bool> waiter); |
| | | 28 | | static abstract Task WaitAsync(Task task); |
| | | 29 | | static abstract ValueTask<T> WaitAsync<T>(ValueTask<T> task); |
| | | 30 | | } |
| | | 31 | | |
| | | 32 | | internal readonly struct AsyncReadWriteAdapter : IReadWriteAdapter |
| | | 33 | | { |
| | | 34 | | public static ValueTask<int> ReadAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationToken) |
| | | 35 | | stream.ReadAsync(buffer, cancellationToken); |
| | | 36 | | |
| | | 37 | | public static ValueTask<int> ReadAtLeastAsync(Stream stream, Memory<byte> buffer, int minimumBytes, bool throwOn |
| | | 38 | | stream.ReadAtLeastAsync(buffer, minimumBytes, throwOnEndOfStream, cancellationToken); |
| | | 39 | | |
| | | 40 | | public static ValueTask ReadExactlyAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationToken |
| | | 41 | | stream.ReadExactlyAsync(buffer, cancellationToken); |
| | | 42 | | |
| | | 43 | | public static ValueTask WriteAsync(Stream stream, ReadOnlyMemory<byte> buffer, CancellationToken cancellationTok |
| | | 44 | | stream.WriteAsync(buffer, cancellationToken); |
| | | 45 | | |
| | | 46 | | public static ValueTask CopyToAsync(Stream source, Stream destination, CancellationToken cancellationToken) => |
| | | 47 | | new ValueTask(source.CopyToAsync(destination, cancellationToken)); |
| | | 48 | | |
| | | 49 | | public static ValueTask DisposeAsync(Stream stream) => stream.DisposeAsync(); |
| | | 50 | | |
| | | 51 | | public static Task FlushAsync(Stream stream, CancellationToken cancellationToken) => stream.FlushAsync(cancellat |
| | | 52 | | |
| | | 53 | | public static Task WaitAsync(TaskCompletionSource<bool> waiter) => waiter.Task; |
| | | 54 | | public static Task WaitAsync(Task task) => task; |
| | | 55 | | public static ValueTask<T> WaitAsync<T>(ValueTask<T> task) => task; |
| | | 56 | | } |
| | | 57 | | |
| | | 58 | | internal readonly struct SyncReadWriteAdapter : IReadWriteAdapter |
| | | 59 | | { |
| | | 60 | | public static ValueTask<int> ReadAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationToken) |
| | 0 | 61 | | new ValueTask<int>(stream.Read(buffer.Span)); |
| | | 62 | | |
| | | 63 | | public static ValueTask<int> ReadAtLeastAsync(Stream stream, Memory<byte> buffer, int minimumBytes, bool throwOn |
| | 0 | 64 | | new ValueTask<int>(stream.ReadAtLeast(buffer.Span, minimumBytes, throwOnEndOfStream)); |
| | | 65 | | |
| | | 66 | | public static ValueTask ReadExactlyAsync(Stream stream, Memory<byte> buffer, CancellationToken cancellationToken |
| | 0 | 67 | | { |
| | 0 | 68 | | stream.ReadExactly(buffer.Span); |
| | 0 | 69 | | return default; |
| | 0 | 70 | | } |
| | | 71 | | |
| | | 72 | | public static ValueTask WriteAsync(Stream stream, ReadOnlyMemory<byte> buffer, CancellationToken cancellationTok |
| | 0 | 73 | | { |
| | 0 | 74 | | stream.Write(buffer.Span); |
| | 0 | 75 | | return default; |
| | 0 | 76 | | } |
| | | 77 | | |
| | | 78 | | public static ValueTask CopyToAsync(Stream source, Stream destination, CancellationToken cancellationToken) |
| | 0 | 79 | | { |
| | 0 | 80 | | source.CopyTo(destination); |
| | 0 | 81 | | return default; |
| | 0 | 82 | | } |
| | | 83 | | |
| | | 84 | | public static ValueTask DisposeAsync(Stream stream) |
| | 0 | 85 | | { |
| | 0 | 86 | | stream.Dispose(); |
| | 0 | 87 | | return default; |
| | 0 | 88 | | } |
| | | 89 | | |
| | | 90 | | public static Task FlushAsync(Stream stream, CancellationToken cancellationToken) |
| | 0 | 91 | | { |
| | 0 | 92 | | stream.Flush(); |
| | 0 | 93 | | return Task.CompletedTask; |
| | 0 | 94 | | } |
| | | 95 | | |
| | | 96 | | public static Task WaitAsync(TaskCompletionSource<bool> waiter) |
| | 0 | 97 | | { |
| | 0 | 98 | | waiter.Task.GetAwaiter().GetResult(); |
| | 0 | 99 | | return Task.CompletedTask; |
| | 0 | 100 | | } |
| | | 101 | | |
| | | 102 | | public static Task WaitAsync(Task task) |
| | 0 | 103 | | { |
| | 0 | 104 | | task.GetAwaiter().GetResult(); |
| | 0 | 105 | | return Task.CompletedTask; |
| | 0 | 106 | | } |
| | | 107 | | |
| | | 108 | | public static ValueTask<T> WaitAsync<T>(ValueTask<T> task) |
| | 0 | 109 | | { |
| | 0 | 110 | | return ValueTask.FromResult(task.AsTask().GetAwaiter().GetResult()); |
| | 0 | 111 | | } |
| | | 112 | | } |
| | | 113 | | } |
| | | 114 | | |