75 lines
2.1 KiB
C#
75 lines
2.1 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Buffers;
|
|
using System.IO;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace System.IO.Pipelines
|
|
{
|
|
// Write only stream implementation for efficiently writing bytes from the request body
|
|
internal class PipeWriterStream : Stream
|
|
{
|
|
private long _length;
|
|
private readonly PipeWriter _pipeWriter;
|
|
|
|
public PipeWriterStream(PipeWriter pipeWriter)
|
|
{
|
|
_pipeWriter = pipeWriter;
|
|
}
|
|
|
|
public override bool CanRead => false;
|
|
|
|
public override bool CanSeek => false;
|
|
|
|
public override bool CanWrite => true;
|
|
|
|
public override long Length => _length;
|
|
|
|
public override long Position { get => throw new NotSupportedException(); set => throw new NotSupportedException(); }
|
|
|
|
public override void Flush()
|
|
{
|
|
throw new NotSupportedException();
|
|
}
|
|
|
|
public override int Read(byte[] buffer, int offset, int count)
|
|
{
|
|
throw new NotSupportedException();
|
|
}
|
|
|
|
public override long Seek(long offset, SeekOrigin origin)
|
|
{
|
|
throw new NotSupportedException();
|
|
}
|
|
|
|
public override void SetLength(long value)
|
|
{
|
|
throw new NotSupportedException();
|
|
}
|
|
|
|
public override void Write(byte[] buffer, int offset, int count)
|
|
{
|
|
_pipeWriter.Write(new ReadOnlySpan<byte>(buffer, offset, count));
|
|
_length += count;
|
|
}
|
|
|
|
public override Task WriteAsync(byte[] buffer, int offset, int count, CancellationToken cancellationToken)
|
|
{
|
|
Write(buffer, offset, count);
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
#if NETCOREAPP2_1
|
|
public override ValueTask WriteAsync(ReadOnlyMemory<byte> source, CancellationToken cancellationToken = default)
|
|
{
|
|
_pipeWriter.Write(source.Span);
|
|
_length += source.Length;
|
|
return new ValueTask(Task.CompletedTask);
|
|
}
|
|
#endif
|
|
}
|
|
}
|