Add ArrayPool<T>.RentOwner() extension returning IMemoryOwner<T>

PowerKit.Tests/ArrayPoolExtensionsTests.cs

@@ -0,0 +1,39 @@
+using System.Buffers;
+using FluentAssertions;
+using PowerKit.Extensions;
+using Xunit;
+
+namespace PowerKit.Tests;
+
+public class ArrayPoolExtensionsTests
+{
+    [Fact]
+    public void RentOwner_Test()
+    {
+        // Arrange
+        var pool = ArrayPool<byte>.Shared;
+
+        // Act
+        using var owner = pool.RentOwner(16);
+
+        // Assert
+        owner.Memory.Length.Should().Be(16);
+    }
+
+    [Fact]
+    public void RentOwner_Dispose_Test()
+    {
+        // Arrange
+        var pool = ArrayPool<byte>.Create(16, 1);
+
+        // Act & assert
+        using (var owner = pool.RentOwner(16))
+        {
+            owner.Memory.Length.Should().Be(16);
+        }
+
+        // Should be able to rent again after the previous owner was disposed
+        using var owner2 = pool.RentOwner(16);
+        owner2.Memory.Length.Should().Be(16);
+    }
+}

PowerKit/Extensions/ArrayPoolExtensions.cs

@@ -0,0 +1,37 @@
+using System;
+using System.Buffers;
+
+namespace PowerKit.Extensions;
+
+internal static class ArrayPoolExtensions
+{
+    extension<T>(ArrayPool<T> pool)
+    {
+        /// <summary>
+        /// Rents a buffer of at least <paramref name="minimumLength" /> elements from the pool
+        /// and wraps it in an <see cref="IMemoryOwner{T}" /> that returns the buffer to the pool
+        /// when disposed.
+        /// </summary>
+        public IMemoryOwner<T> RentOwner(int minimumLength = 1) =>
+            new ArrayPoolMemoryOwner<T>(pool, pool.Rent(minimumLength), minimumLength);
+    }
+}
+
+file sealed class ArrayPoolMemoryOwner<T>(ArrayPool<T> pool, T[] buffer, int minimumLength)
+    : IMemoryOwner<T>
+{
+    private bool _disposed;
+
+    public Memory<T> Memory { get; } = buffer.AsMemory(0, minimumLength);
+
+    public void Dispose()
+    {
+        if (_disposed)
+        {
+            return;
+        }
+
+        _disposed = true;
+        pool.Return(buffer);
+    }
+}