Unity实现一个简单易用的对象池

最新推荐文章于 2025-06-08 18:37:45 发布

示申○言舌

最新推荐文章于 2025-06-08 18:37:45 发布

阅读量252

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/sdhexu/article/details/115399503

Unity 专栏收录该内容

40 篇文章

订阅专栏

本文介绍了Unity中实现对象池的原理和源码，通过预先创建并管理对象，减少频繁实例化带来的性能消耗。对象池包含正在使用和未使用对象的列表，通过Pop和Push操作实现对象的复用，适用于处理如子弹等短生命周期的游戏对象。示例展示了如何创建子弹对象池，并在子弹超出寿命后自动归还到池中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Unity实现简单易用的对象池

原理

当场景中需要频繁实例化一种对象时，由于实例化的过程会比较耗时，所以可以事先创建一堆对象，然后在需要时激活它。
池子中，我们有两个列表，一个是正在被使用的对象，另一个是还没有被使用的对象，需要一个物体时，从正在使用的列表中取出一个物体，然后将它标记为“已被使用”，其实就是放到“已被使用的”列表中。当物体被使用完，归还物体到池中时，进行相反的操作。
##源码

public static class ObjectPool
{
    private class ObjectBuffer<T> where T : MonoBehaviour
    {
        public T prefab = null;
        public int initSize = 0;
        public Action<T> onCreate = null;
        public Action<T> onPop = null;
        public Action<T> onPush = null;

        public List<T> used = new List<T>();
        public List<T> unused = new List<T>();

        public void Grow()
        {
            T obj;
            for (int i = 0; i < initSize; ++i)
            {
                obj = GameObject.Instantiate<T>(prefab);
                onCreate?.Invoke(obj);
                unused.Add(obj);
            }
        }

        public T Pop()
        {
            if (unused.Count <= 0)
                Grow();

            T obj = unused[0];
            unused.RemoveAt(0);
            used.Add(obj);
            onPop?.Invoke(obj);
            return obj;
        }

        public void Push( T obj )
        {
            if (used.Contains(obj))
            {
                used.Remove(obj);
                unused.Add(obj);
                onPush?.Invoke(obj);
            }
        }
    }

    private static Dictionary<Type, object> m_buffers = new Dictionary<Type, object>();
    public static void CreatePool<T>( T prefab, Action<T> onCreate = null, Action<T> onPop = null, Action<T> onPush = null, int count = 64 ) where T : MonoBehaviour
    {
        ObjectBuffer<T> buffer = new ObjectBuffer<T>()
        {
            prefab = prefab,
            initSize = count,
            onCreate = onCreate,
            onPop = onPop,
            onPush = onPush
        };
        buffer.Grow();
        m_buffers.Add(typeof(T), buffer);
    }

    public static T PopObject<T>() where T : MonoBehaviour
    {
        if(m_buffers.ContainsKey(typeof(T)))
        {
            if (m_buffers[typeof(T)] is ObjectBuffer<T> buffer)
                return buffer.Pop();
        }
        return null;
    }

    public static void PushObject<T>( T obj) where T : MonoBehaviour
    {
        if(m_buffers.ContainsKey(typeof(T)))
        {
            if (m_buffers[typeof(T)] is ObjectBuffer<T> buffer)
                buffer.Push(obj);
        }
    }
}

例子：
子弹类

public class Bullet : MonoBehaviour
{
    private float life = 0;

    void Update()
    {
        life += Time.deltaTime;
        if( life > 5 )
        {
            ObjectPool.PushObject<Bullet>(this);
            life = 0;
        }
        transform.Translate(transform.forward * 0.5f);
    }
}

测试类，按下鼠标发射子弹

public class Test : MonoBehaviour
{
    [SerializeField]
    private Bullet m_prefab = null;

    private void Awake()
    {
        ObjectPool.CreatePool<Bullet>(m_prefab, OnCreateOrPush,
            x => { x.gameObject.SetActive(true); }, OnCreateOrPush);
    }

    private void OnCreateOrPush( Bullet obj)
    {
        obj.gameObject.SetActive(false);
        obj.transform.position = transform.position;
        obj.transform.rotation = transform.rotation;
    }

    private void Update()
    {
        if(Input.GetMouseButton(0))
        {
            ObjectPool.PopObject<Bullet>();
        }
    }
}