ObjectPool.cs 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. using System;
  2. using System.Collections.Generic;
  3. namespace ETModel
  4. {
  5. public class ObjectPool
  6. {
  7. private readonly Dictionary<Type, Queue<Component>> dictionary = new Dictionary<Type, Queue<Component>>();
  8. public Component Fetch(Type type)
  9. {
  10. Queue<Component> queue;
  11. if (!this.dictionary.TryGetValue(type, out queue))
  12. {
  13. queue = new Queue<Component>();
  14. this.dictionary.Add(type, queue);
  15. }
  16. Component obj;
  17. if (queue.Count > 0)
  18. {
  19. obj = queue.Dequeue();
  20. }
  21. else
  22. {
  23. obj = (Component)Activator.CreateInstance(type);
  24. }
  25. obj.IsFromPool = true;
  26. return obj;
  27. }
  28. public T Fetch<T>() where T: Component
  29. {
  30. T t = (T) this.Fetch(typeof(T));
  31. return t;
  32. }
  33. public void Recycle(Component obj)
  34. {
  35. Type type = obj.GetType();
  36. Queue<Component> queue;
  37. if (!this.dictionary.TryGetValue(type, out queue))
  38. {
  39. queue = new Queue<Component>();
  40. this.dictionary.Add(type, queue);
  41. }
  42. queue.Enqueue(obj);
  43. }
  44. public void Clear()
  45. {
  46. foreach (var kv in this.dictionary)
  47. {
  48. foreach (Component component in kv.Value)
  49. {
  50. component.IsFromPool = false;
  51. component.Dispose();
  52. }
  53. }
  54. this.dictionary.Clear();
  55. }
  56. }
  57. }