ReferencePool.cs 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
  1. //------------------------------------------------------------
  2. // Game Framework
  3. // Copyright © 2013-2021 loyalsoft. All rights reserved.
  4. // Homepage: http://www.game7000.com/
  5. // Feedback: http://www.game7000.com/
  6. //------------------------------------------------------------
  7. using System;
  8. using System.Collections.Generic;
  9. namespace GameFramework
  10. {
  11. /// <summary>
  12. /// 引用池。
  13. /// </summary>
  14. public static partial class ReferencePool
  15. {
  16. private static readonly Dictionary<Type, ReferenceCollection> s_ReferenceCollections = new Dictionary<Type, ReferenceCollection>();
  17. private static bool m_EnableStrictCheck = false;
  18. /// <summary>
  19. /// 获取或设置是否开启强制检查。
  20. /// </summary>
  21. public static bool EnableStrictCheck
  22. {
  23. get
  24. {
  25. return m_EnableStrictCheck;
  26. }
  27. set
  28. {
  29. m_EnableStrictCheck = value;
  30. }
  31. }
  32. /// <summary>
  33. /// 获取引用池的数量。
  34. /// </summary>
  35. public static int Count
  36. {
  37. get
  38. {
  39. return s_ReferenceCollections.Count;
  40. }
  41. }
  42. /// <summary>
  43. /// 获取所有引用池的信息。
  44. /// </summary>
  45. /// <returns>所有引用池的信息。</returns>
  46. public static ReferencePoolInfo[] GetAllReferencePoolInfos()
  47. {
  48. int index = 0;
  49. ReferencePoolInfo[] results = null;
  50. lock (s_ReferenceCollections)
  51. {
  52. results = new ReferencePoolInfo[s_ReferenceCollections.Count];
  53. foreach (KeyValuePair<Type, ReferenceCollection> referenceCollection in s_ReferenceCollections)
  54. {
  55. results[index++] = new ReferencePoolInfo(referenceCollection.Key, referenceCollection.Value.UnusedReferenceCount, referenceCollection.Value.UsingReferenceCount, referenceCollection.Value.AcquireReferenceCount, referenceCollection.Value.ReleaseReferenceCount, referenceCollection.Value.AddReferenceCount, referenceCollection.Value.RemoveReferenceCount);
  56. }
  57. }
  58. return results;
  59. }
  60. /// <summary>
  61. /// 清除所有引用池。
  62. /// </summary>
  63. public static void ClearAll()
  64. {
  65. lock (s_ReferenceCollections)
  66. {
  67. foreach (KeyValuePair<Type, ReferenceCollection> referenceCollection in s_ReferenceCollections)
  68. {
  69. referenceCollection.Value.RemoveAll();
  70. }
  71. s_ReferenceCollections.Clear();
  72. }
  73. }
  74. /// <summary>
  75. /// 从引用池获取引用。
  76. /// </summary>
  77. /// <typeparam name="T">引用类型。</typeparam>
  78. /// <returns>引用。</returns>
  79. public static T Acquire<T>() where T : class, IReference, new()
  80. {
  81. return GetReferenceCollection(typeof(T)).Acquire<T>();
  82. }
  83. /// <summary>
  84. /// 从引用池获取引用。
  85. /// </summary>
  86. /// <param name="referenceType">引用类型。</param>
  87. /// <returns>引用。</returns>
  88. public static IReference Acquire(Type referenceType)
  89. {
  90. InternalCheckReferenceType(referenceType);
  91. return GetReferenceCollection(referenceType).Acquire();
  92. }
  93. /// <summary>
  94. /// 将引用归还引用池。
  95. /// </summary>
  96. /// <param name="reference">引用。</param>
  97. public static void Release(IReference reference)
  98. {
  99. if (reference == null)
  100. {
  101. throw new GameFrameworkException("Reference is invalid.");
  102. }
  103. Type referenceType = reference.GetType();
  104. InternalCheckReferenceType(referenceType);
  105. GetReferenceCollection(referenceType).Release(reference);
  106. }
  107. /// <summary>
  108. /// 向引用池中追加指定数量的引用。
  109. /// </summary>
  110. /// <typeparam name="T">引用类型。</typeparam>
  111. /// <param name="count">追加数量。</param>
  112. public static void Add<T>(int count) where T : class, IReference, new()
  113. {
  114. GetReferenceCollection(typeof(T)).Add<T>(count);
  115. }
  116. /// <summary>
  117. /// 向引用池中追加指定数量的引用。
  118. /// </summary>
  119. /// <param name="referenceType">引用类型。</param>
  120. /// <param name="count">追加数量。</param>
  121. public static void Add(Type referenceType, int count)
  122. {
  123. InternalCheckReferenceType(referenceType);
  124. GetReferenceCollection(referenceType).Add(count);
  125. }
  126. /// <summary>
  127. /// 从引用池中移除指定数量的引用。
  128. /// </summary>
  129. /// <typeparam name="T">引用类型。</typeparam>
  130. /// <param name="count">移除数量。</param>
  131. public static void Remove<T>(int count) where T : class, IReference
  132. {
  133. GetReferenceCollection(typeof(T)).Remove(count);
  134. }
  135. /// <summary>
  136. /// 从引用池中移除指定数量的引用。
  137. /// </summary>
  138. /// <param name="referenceType">引用类型。</param>
  139. /// <param name="count">移除数量。</param>
  140. public static void Remove(Type referenceType, int count)
  141. {
  142. InternalCheckReferenceType(referenceType);
  143. GetReferenceCollection(referenceType).Remove(count);
  144. }
  145. /// <summary>
  146. /// 从引用池中移除所有的引用。
  147. /// </summary>
  148. /// <typeparam name="T">引用类型。</typeparam>
  149. public static void RemoveAll<T>() where T : class, IReference
  150. {
  151. GetReferenceCollection(typeof(T)).RemoveAll();
  152. }
  153. /// <summary>
  154. /// 从引用池中移除所有的引用。
  155. /// </summary>
  156. /// <param name="referenceType">引用类型。</param>
  157. public static void RemoveAll(Type referenceType)
  158. {
  159. InternalCheckReferenceType(referenceType);
  160. GetReferenceCollection(referenceType).RemoveAll();
  161. }
  162. private static void InternalCheckReferenceType(Type referenceType)
  163. {
  164. if (!m_EnableStrictCheck)
  165. {
  166. return;
  167. }
  168. if (referenceType == null)
  169. {
  170. throw new GameFrameworkException("Reference type is invalid.");
  171. }
  172. if (!referenceType.IsClass || referenceType.IsAbstract)
  173. {
  174. throw new GameFrameworkException("Reference type is not a non-abstract class type.");
  175. }
  176. if (!typeof(IReference).IsAssignableFrom(referenceType))
  177. {
  178. throw new GameFrameworkException(Utility.Text.Format("Reference type '{0}' is invalid.", referenceType.FullName));
  179. }
  180. }
  181. private static ReferenceCollection GetReferenceCollection(Type referenceType)
  182. {
  183. if (referenceType == null)
  184. {
  185. throw new GameFrameworkException("ReferenceType is invalid.");
  186. }
  187. ReferenceCollection referenceCollection = null;
  188. lock (s_ReferenceCollections)
  189. {
  190. if (!s_ReferenceCollections.TryGetValue(referenceType, out referenceCollection))
  191. {
  192. referenceCollection = new ReferenceCollection(referenceType);
  193. s_ReferenceCollections.Add(referenceType, referenceCollection);
  194. }
  195. }
  196. return referenceCollection;
  197. }
  198. }
  199. }