| | | 1 | | using System.Collections.Generic; |
| | | 2 | | using DCL.Interface; |
| | | 3 | | using UnityEngine; |
| | | 4 | | |
| | | 5 | | namespace DCL.Social.Chat |
| | | 6 | | { |
| | | 7 | | [CreateAssetMenu(fileName = "PoolChatEntryFactory", menuName = "DCL/Social/PoolChatEntryFactory")] |
| | | 8 | | public class PoolChatEntryFactory : ScriptableObject, IChatEntryFactory |
| | | 9 | | { |
| | | 10 | | private const string POOL_NAME_PREFIX = "ChatMessages_"; |
| | | 11 | | private const int PRE_INSTANTIATED_ENTRIES = 30; |
| | | 12 | | |
| | | 13 | | [SerializeField] private DefaultChatEntry defaultMessagePrefab; |
| | | 14 | | [SerializeField] private DefaultChatEntry systemMessagePrefab; |
| | | 15 | | [SerializeField] private DefaultChatEntry privateReceivedMessagePrefab; |
| | | 16 | | [SerializeField] private DefaultChatEntry privateSentMessagePrefab; |
| | | 17 | | [SerializeField] private DefaultChatEntry publicReceivedMessagePrefab; |
| | | 18 | | [SerializeField] private DefaultChatEntry publicSentMessagePrefab; |
| | | 19 | | |
| | 3 | 20 | | private readonly Dictionary<ChatEntry, PoolableObject> pooledObjects = |
| | | 21 | | new Dictionary<ChatEntry, PoolableObject>(); |
| | | 22 | | |
| | | 23 | | public ChatEntry Create(ChatEntryModel model) |
| | | 24 | | { |
| | 5 | 25 | | if (model.messageType == ChatMessage.Type.SYSTEM) |
| | 0 | 26 | | return GetEntryFromPool("SYSTEM", systemMessagePrefab); |
| | | 27 | | |
| | 5 | 28 | | if (model.messageType == ChatMessage.Type.PUBLIC) |
| | | 29 | | { |
| | 3 | 30 | | if (model.subType == ChatEntryModel.SubType.RECEIVED) |
| | 1 | 31 | | return GetEntryFromPool("PUBLIC_RECEIVED", publicReceivedMessagePrefab); |
| | 2 | 32 | | if (model.subType == ChatEntryModel.SubType.SENT) |
| | 2 | 33 | | return GetEntryFromPool("PUBLIC_SENT", publicSentMessagePrefab); |
| | | 34 | | } |
| | 2 | 35 | | else if (model.messageType == ChatMessage.Type.PRIVATE) |
| | | 36 | | { |
| | 2 | 37 | | if (model.subType == ChatEntryModel.SubType.RECEIVED) |
| | 1 | 38 | | return GetEntryFromPool("PRIVATE_RECEIVED", privateReceivedMessagePrefab); |
| | 1 | 39 | | if (model.subType == ChatEntryModel.SubType.SENT) |
| | 1 | 40 | | return GetEntryFromPool("PRIVATE_SENT", privateSentMessagePrefab); |
| | | 41 | | } |
| | 0 | 42 | | return GetEntryFromPool("DEFAULT", defaultMessagePrefab); |
| | | 43 | | } |
| | | 44 | | |
| | | 45 | | public void Destroy(ChatEntry entry) |
| | | 46 | | { |
| | 0 | 47 | | if (!pooledObjects.TryGetValue(entry, out var pooledObj)) return; |
| | 0 | 48 | | pooledObj.Release(); |
| | 0 | 49 | | pooledObjects.Remove(entry); |
| | 0 | 50 | | } |
| | | 51 | | |
| | | 52 | | private DefaultChatEntry GetEntryFromPool(string poolName, DefaultChatEntry prefab) |
| | | 53 | | { |
| | 5 | 54 | | var pool = GetPool(poolName, prefab); |
| | 5 | 55 | | var pooledObj = pool.Get(); |
| | 5 | 56 | | var entry = pooledObj.gameObject.GetComponent<DefaultChatEntry>(); |
| | 5 | 57 | | pooledObjects[entry] = pooledObj; |
| | 5 | 58 | | return entry; |
| | | 59 | | } |
| | | 60 | | |
| | | 61 | | private Pool GetPool(string poolName, DefaultChatEntry prefab) |
| | | 62 | | { |
| | 5 | 63 | | var poolId = POOL_NAME_PREFIX + poolName + GetInstanceID(); |
| | 5 | 64 | | var entryPool = PoolManager.i.GetPool(poolId); |
| | 6 | 65 | | if (entryPool != null) return entryPool; |
| | | 66 | | |
| | 4 | 67 | | entryPool = PoolManager.i.AddPool( |
| | | 68 | | poolId, |
| | | 69 | | Instantiate(prefab).gameObject, |
| | | 70 | | maxPrewarmCount: PRE_INSTANTIATED_ENTRIES, |
| | | 71 | | isPersistent: true); |
| | 4 | 72 | | entryPool.ForcePrewarm(); |
| | | 73 | | |
| | 4 | 74 | | return entryPool; |
| | | 75 | | } |
| | | 76 | | } |
| | | 77 | | } |