StitchATon2/Infra/Buffers/ArrayOwner.cs

using System.Buffers;

namespace StitchATon2.Infra.Buffers;

public class ArrayOwner<T> : IBuffer<T> where T : unmanaged 
{
    private readonly ArrayPool<T> _owner;
    private readonly T[] _buffer;

    public ArrayOwner(ArrayPool<T> owner, int length)
    {
        _owner = owner;
        _buffer = owner.Rent(length);
        Length = length;
    }
    
    ~ArrayOwner() => Dispose();
    
    public void Dispose()
    {
        _owner.Return(_buffer);
        GC.SuppressFinalize(this);
    }

    public ref T this[int index] => ref _buffer[index];

    public Span<T> Span => _buffer.AsSpan(0, Length);
    public Memory<T> Memory => _buffer.AsMemory(0, Length);

    public T[] Array => _buffer;

    public int Length { get; }
}