RawAvailableMessageFrameCodec.cs

이 파일의 문서화 페이지로 가기

using System.Buffers;
 
namespace Dreamine.Communication.Core.Framing;

public sealed class RawAvailableMessageFrameCodec : IMessageFrameCodec
{
    private readonly int _bufferSize;

    public RawAvailableMessageFrameCodec()
        : this(8192)
    {
    }

    public RawAvailableMessageFrameCodec(int bufferSize)
    {
        if (bufferSize <= 0)
        {
            throw new ArgumentOutOfRangeException(nameof(bufferSize));
        }
 
        _bufferSize = bufferSize;
    }

    public async Task WriteFrameAsync(
        Stream stream,
        byte[] payload,
        CancellationToken cancellationToken = default)
    {
        ArgumentNullException.ThrowIfNull(stream);
        ArgumentNullException.ThrowIfNull(payload);
 
        await stream.WriteAsync(payload, cancellationToken).ConfigureAwait(false);
        await stream.FlushAsync(cancellationToken).ConfigureAwait(false);
    }

    public async Task<byte[]?> ReadFrameAsync(
        Stream stream,
        CancellationToken cancellationToken = default)
    {
        ArgumentNullException.ThrowIfNull(stream);
 
        var buffer = ArrayPool<byte>.Shared.Rent(_bufferSize);
 
        try
        {
            var read = await stream.ReadAsync(
                    buffer.AsMemory(0, _bufferSize),
                    cancellationToken)
                .ConfigureAwait(false);
 
            if (read == 0)
            {
                return null;
            }
 
            var payload = new byte[read];
            Buffer.BlockCopy(buffer, 0, payload, 0, read);
 
            return payload;
        }
        finally
        {
            ArrayPool<byte>.Shared.Return(buffer);
        }
    }
}