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);
        }
    }
}