MQTTnet/Source/MQTTnet.Tests/ASP/Mockups/MemoryOwner.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System;
using System.Buffers;

namespace MQTTnet.Tests.ASP.Mockups;

public sealed class MemoryOwner : IMemoryOwner<byte>
{
    readonly byte[] _raw;

    public MemoryOwner(int size)
    {
        if (size <= 0)
        {
            size = 1024;
        }

        if (size > 4096)
        {
            size = 4096;
        }

        _raw = ArrayPool<byte>.Shared.Rent(size);
        Memory = _raw;
    }

    public Memory<byte> Memory { get; }

    public void Dispose()
    {
        ArrayPool<byte>.Shared.Return(_raw);
    }
}