open-project-1/UOP1_Project/Assets/Scripts/Pool/PoolSO.cs

using UOP1.Factory;
using System.Collections.Generic;
using UnityEngine;

namespace UOP1.Pool
{
	/// <summary>
	/// A generic pool that generates members of type T on-demand via a factory.
	/// </summary>
	/// <typeparam name="T">Specifies the type of elements to pool.</typeparam>
	public abstract class PoolSO<T> : ScriptableObject, IPool<T>
	{
		protected readonly Stack<T> Available = new Stack<T>();
		public abstract IFactory<T> Factory { get; set; }
		protected bool HasBeenPrewarmed { get; set; }

		protected virtual T Create()
		{
			return Factory.Create();
		}

		public virtual void Prewarm(int num)
		{
			if (HasBeenPrewarmed)
			{
				Debug.LogWarning($"Pool {name} has already been prewarmed.");
				return;
			}
			for (int i = 0; i < num; i++)
			{
				Available.Push(Create());
			}
			HasBeenPrewarmed = true;
		}

		public virtual T Request()
		{
			return Available.Count > 0 ? Available.Pop() : Create();
		}

		public virtual IEnumerable<T> Request(int num = 1)
		{
			List<T> members = new List<T>(num);
			for (int i = 0; i < num; i++)
			{
				members.Add(Request());
			}
			return members;
		}

		public virtual void Return(T member)
		{
			Available.Push(member);
		}

		public virtual void Return(IEnumerable<T> members)
		{
			foreach (T member in members)
			{
				Return(member);
			}
		}

		public virtual void OnDisable()
		{
			Available.Clear();
		}
	}
}