com.alicizax.unity.framework/Runtime/ABase/Base/MemoryPool/MemoryPoolExtension.cs

using System;

namespace AlicizaX
{
    /// <summary>
    /// 内存池对象基类。
    /// </summary>
    public abstract class MemoryObject : IMemory
    {
        /// <summary>
        /// 清理内存对象回收入池。
        /// </summary>
        public virtual void Clear()
        {
        }

        /// <summary>
        /// 从内存池中初始化。
        /// </summary>
        public abstract void InitFromPool();

        /// <summary>
        /// 回收到内存池。
        /// </summary>
        public abstract void RecycleToPool();
    }

    public static partial class MemoryPool
    {
        /// <summary>
        /// 从内存池获取内存对象。
        /// </summary>
        public static T Alloc<T>() where T : MemoryObject, new()
        {
            T memory = MemoryPool<T>.Acquire();
            memory.InitFromPool();
            return memory;
        }

        /// <summary>
        /// 将内存对象归还内存池。
        /// </summary>
        public static void Dealloc(MemoryObject memory)
        {
            if (memory == null)
            {
                throw new ArgumentNullException(nameof(memory));
            }

            memory.RecycleToPool();
            MemoryPoolRegistry.Release(memory);
        }
    }
}