public override async Task WriteAsync(byte[] buffer, int offset, int length, CancellationToken cancellationToken)
{
CheckNotDisposed();
ValidateBufferArgs(buffer, offset, length);
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
// enough space left in buffer?
var free = WriteBuffer.Capacity - WriteBuffer.Length;
if (length > free)
{
ArraySegment <byte> bufSegment;
WriteBuffer.TryGetBuffer(out bufSegment);
await InnerTransport.WriteAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
WriteBuffer.SetLength(0);
}
// do the data even fit into the buffer?
// Note we test for < instead of <= to avoid nonsense buffering
if (length < WriteBuffer.Capacity)
{
await WriteBuffer.WriteAsync(buffer, offset, length, cancellationToken);
return;
}
// write thru
await InnerTransport.WriteAsync(buffer, offset, length, cancellationToken);
}
public override async Task FlushAsync(CancellationToken cancellationToken)
{
CheckNotDisposed();
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
ArraySegment <byte> bufSegment;
WriteBuffer.TryGetBuffer(out bufSegment);
int dataLen = bufSegment.Count - HeaderSize;
if (dataLen < 0)
{
throw new InvalidOperationException(); // logic error actually
}
// Inject message header into the reserved buffer space
//BinaryPrimitives.WriteInt32BigEndian(bufSegment.Array, dataLen);
Binary.BigEndian.Set(dataLen, bufSegment.Array);
// Send the entire message at once
await InnerTransport.WriteAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
InitWriteBuffer();
await InnerTransport.FlushAsync(cancellationToken);
}
public override async Task FlushAsync(CancellationToken cancellationToken)
{
CheckNotDisposed();
if (!IsOpen)
{
throw new TTransportException(TTransportException.ExceptionType.NotOpen);
}
if (WriteBuffer.Length > 0)
{
WriteBuffer.TryGetBuffer(out ArraySegment <byte> bufSegment);
await InnerTransport.WriteAsync(bufSegment.Array, 0, bufSegment.Count, cancellationToken);
WriteBuffer.SetLength(0);
}
await InnerTransport.FlushAsync(cancellationToken);
}
