Unity对象池的实现

烟雨

网上介绍对象池的文章有很多，但是总感觉代码不太清晰，并不适合新手学习最近在一个工程里看到一段对象池的代码感觉不错，故分享一下

1. using UnityEngine;
   
2. using System.Collections;
   
3. using System.Collections.Generic;
   
4. using System;
   
5. 
   
6. public class PoolManager : UnitySingleton<PoolManager> {
   
7. 
   
8.         public static Dictionary<Type, Stack<IPoolable>> ObjectPoolDic = new Dictionary<Type, Stack<IPoolable>>();
   
9.         public static Dictionary<Type, int> ObjectPoolSizeDic = new Dictionary<Type,int>();
   
10.        
    
11.         void Start () {
    
12. 
    
13.         }
    
14.        
    
15.         public void RegistPoolableType(Type type, int poolSize)
    
16.         {
    
17.                 if (!ObjectPoolDic.ContainsKey(type))
    
18.                 {
    
19.                         ObjectPoolDic[type] = new Stack<IPoolable>();
    
20.                         ObjectPoolSizeDic[type] = poolSize;
    
21.                 }
    
22.         }
    
23.        
    
24.         public bool HasPoolObject(Type type)
    
25.         {
    
26.                 return ObjectPoolDic.ContainsKey(type) && ObjectPoolDic[type].Count > 0;
    
27.         }
    
28.        
    
29.         public bool IsPoolFull(Type type)
    
30.         {
    
31.                 if (!ObjectPoolDic.ContainsKey(type))
    
32.                         return true;
    
33.                 else if (ObjectPoolDic[type].Count >= ObjectPoolSizeDic[type])
    
34.                         return true;
    
35.                 return false;
    
36.         }
    
37.        
    
38.         public IPoolable TakePoolObject(Type type)
    
39.         {
    
40.                 if (ObjectPoolDic.ContainsKey(type) && ObjectPoolDic[type].Count > 0)
    
41.                 {
    
42.                         return ObjectPoolDic[type].Pop();
    
43.                 }
    
44.                 else
    
45.                 {
    
46.                         return null;
    
47.                 }
    
48.         }
    
49.        
    
50.         public bool PutPoolObject(Type type, IPoolable obj)
    
51.         {
    
52.                 if (!ObjectPoolDic.ContainsKey(type) || ObjectPoolDic[type].Count >= ObjectPoolSizeDic[type])
    
53.                 {
    
54.                         GameObject.Destroy((obj as MonoBehaviour).gameObject);
    
55.                         return false;
    
56.                 }
    
57.                 else
    
58.                 {
    
59.                         (obj as MonoBehaviour).gameObject.SetActive(false);
    
60.                         //(obj as MonoBehaviour).transform.parent = GameManager.Instance.PoolRoot;
    
61.                         ObjectPoolDic[type].Push(obj);
    
62.                         return true;
    
63.                 }
    
64.         }
    
65. }

复制代码

首先继承自一个单例类，就可以用PoolManager.Instance来获取这个类的单例了
定义了两个字典，一个用来对应存储对应的对象类型的栈，一个用来记录实例化的最大个数来控制内存
用的时候Pop，用完了Push
可存储在对象池的对象必须实现IPoolable接口

1. using UnityEngine;
   
2. using System.Collections;
   
3. 
   
4. public interface IPoolable {
   
5. 
   
6.         void Destroy();
   
7. }

复制代码

destroy里可以这样写

1. if (!PoolManager.Instance.IsPoolFull(GetType()))
   
2.                 {
   
3.                         PoolManager.Instance.PutPoolObject(GetType(), this);
   
4.                 }
   
5.                 else
   
6.                 {
   
7.                         GameObject.Destroy(this.gameObject);
   
8.                 }

复制代码