WorkItemsGroupBase.cs 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489
  1. using System;
  2. using System.Threading;
  3. namespace Amib.Threading.Internal
  4. {
  5. public abstract class WorkItemsGroupBase : IWorkItemsGroup
  6. {
  7. #region Private Fields
  8. /// <summary>
  9. /// Contains the name of this instance of SmartThreadPool.
  10. /// Can be changed by the user.
  11. /// </summary>
  12. private string _name = "WorkItemsGroupBase";
  13. public WorkItemsGroupBase()
  14. {
  15. IsIdle = true;
  16. }
  17. #endregion
  18. #region IWorkItemsGroup Members
  19. #region Public Methods
  20. /// <summary>
  21. /// Get/Set the name of the SmartThreadPool/WorkItemsGroup instance
  22. /// </summary>
  23. public string Name
  24. {
  25. get { return _name; }
  26. set { _name = value; }
  27. }
  28. #endregion
  29. #region Abstract Methods
  30. public abstract int Concurrency { get; set; }
  31. public abstract int WaitingCallbacks { get; }
  32. public abstract object[] GetStates();
  33. public abstract WIGStartInfo WIGStartInfo { get; }
  34. public abstract void Start();
  35. public abstract void Cancel(bool abortExecution);
  36. public abstract bool WaitForIdle(int millisecondsTimeout);
  37. public abstract event WorkItemsGroupIdleHandler OnIdle;
  38. internal abstract void Enqueue(WorkItem workItem);
  39. internal virtual void PreQueueWorkItem() { }
  40. #endregion
  41. #region Common Base Methods
  42. /// <summary>
  43. /// Cancel all the work items.
  44. /// Same as Cancel(false)
  45. /// </summary>
  46. public virtual void Cancel()
  47. {
  48. Cancel(false);
  49. }
  50. /// <summary>
  51. /// Wait for the SmartThreadPool/WorkItemsGroup to be idle
  52. /// </summary>
  53. public void WaitForIdle()
  54. {
  55. WaitForIdle(Timeout.Infinite);
  56. }
  57. /// <summary>
  58. /// Wait for the SmartThreadPool/WorkItemsGroup to be idle
  59. /// </summary>
  60. public bool WaitForIdle(TimeSpan timeout)
  61. {
  62. return WaitForIdle((int)timeout.TotalMilliseconds);
  63. }
  64. /// <summary>
  65. /// IsIdle is true when there are no work items running or queued.
  66. /// </summary>
  67. public bool IsIdle { get; protected set; }
  68. #endregion
  69. #region QueueWorkItem
  70. public IWorkItemResult QueueWorkItem(WaitCallback callback)
  71. {
  72. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, null);
  73. Enqueue(workItem);
  74. return workItem.GetWorkItemResult();
  75. }
  76. public IWorkItemResult QueueWorkItem(WaitCallback callback, object state)
  77. {
  78. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state);
  79. Enqueue(workItem);
  80. return workItem.GetWorkItemResult();
  81. }
  82. public IWorkItemResult QueueWorkItem(WorkItemInfo workItemInfo, WaitCallback callback)
  83. {
  84. PreQueueWorkItem();
  85. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, workItemInfo, callback, null);
  86. Enqueue(workItem);
  87. return workItem.GetWorkItemResult();
  88. }
  89. public IWorkItemResult QueueWorkItem(WorkItemInfo workItemInfo, WaitCallback callback, object state)
  90. {
  91. PreQueueWorkItem();
  92. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, workItemInfo, callback, state);
  93. Enqueue(workItem);
  94. return workItem.GetWorkItemResult();
  95. }
  96. /// <summary>
  97. /// Queue a work item
  98. /// </summary>
  99. /// <param name="callback">A callback to execute</param>
  100. /// <returns>Returns a work item result</returns>
  101. public IWorkItemResult QueueWorkItem(WorkItemCallback callback)
  102. {
  103. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback);
  104. Enqueue(workItem);
  105. return workItem.GetWorkItemResult();
  106. }
  107. /// <summary>
  108. /// Queue a work item
  109. /// </summary>
  110. /// <param name="callback">A callback to execute</param>
  111. /// <param name="workItemPriority">The priority of the work item</param>
  112. /// <returns>Returns a work item result</returns>
  113. public IWorkItemResult QueueWorkItem(WorkItemCallback callback, WorkItemPriority workItemPriority)
  114. {
  115. PreQueueWorkItem();
  116. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, workItemPriority);
  117. Enqueue(workItem);
  118. return workItem.GetWorkItemResult();
  119. }
  120. /// <summary>
  121. /// Queue a work item
  122. /// </summary>
  123. /// <param name="workItemInfo">Work item info</param>
  124. /// <param name="callback">A callback to execute</param>
  125. /// <returns>Returns a work item result</returns>
  126. public IWorkItemResult QueueWorkItem(WorkItemInfo workItemInfo, WorkItemCallback callback)
  127. {
  128. PreQueueWorkItem();
  129. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, workItemInfo, callback);
  130. Enqueue(workItem);
  131. return workItem.GetWorkItemResult();
  132. }
  133. /// <summary>
  134. /// Queue a work item
  135. /// </summary>
  136. /// <param name="callback">A callback to execute</param>
  137. /// <param name="state">
  138. /// The context object of the work item. Used for passing arguments to the work item.
  139. /// </param>
  140. /// <returns>Returns a work item result</returns>
  141. public IWorkItemResult QueueWorkItem(WorkItemCallback callback, object state)
  142. {
  143. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state);
  144. Enqueue(workItem);
  145. return workItem.GetWorkItemResult();
  146. }
  147. /// <summary>
  148. /// Queue a work item
  149. /// </summary>
  150. /// <param name="callback">A callback to execute</param>
  151. /// <param name="state">
  152. /// The context object of the work item. Used for passing arguments to the work item.
  153. /// </param>
  154. /// <param name="workItemPriority">The work item priority</param>
  155. /// <returns>Returns a work item result</returns>
  156. public IWorkItemResult QueueWorkItem(WorkItemCallback callback, object state, WorkItemPriority workItemPriority)
  157. {
  158. PreQueueWorkItem();
  159. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state, workItemPriority);
  160. Enqueue(workItem);
  161. return workItem.GetWorkItemResult();
  162. }
  163. /// <summary>
  164. /// Queue a work item
  165. /// </summary>
  166. /// <param name="workItemInfo">Work item information</param>
  167. /// <param name="callback">A callback to execute</param>
  168. /// <param name="state">
  169. /// The context object of the work item. Used for passing arguments to the work item.
  170. /// </param>
  171. /// <returns>Returns a work item result</returns>
  172. public IWorkItemResult QueueWorkItem(WorkItemInfo workItemInfo, WorkItemCallback callback, object state)
  173. {
  174. PreQueueWorkItem();
  175. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, workItemInfo, callback, state);
  176. Enqueue(workItem);
  177. return workItem.GetWorkItemResult();
  178. }
  179. /// <summary>
  180. /// Queue a work item
  181. /// </summary>
  182. /// <param name="callback">A callback to execute</param>
  183. /// <param name="state">
  184. /// The context object of the work item. Used for passing arguments to the work item.
  185. /// </param>
  186. /// <param name="postExecuteWorkItemCallback">
  187. /// A delegate to call after the callback completion
  188. /// </param>
  189. /// <returns>Returns a work item result</returns>
  190. public IWorkItemResult QueueWorkItem( WorkItemCallback callback, object state,
  191. PostExecuteWorkItemCallback postExecuteWorkItemCallback)
  192. {
  193. PreQueueWorkItem();
  194. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state, postExecuteWorkItemCallback);
  195. Enqueue(workItem);
  196. return workItem.GetWorkItemResult();
  197. }
  198. /// <summary>
  199. /// Queue a work item
  200. /// </summary>
  201. /// <param name="callback">A callback to execute</param>
  202. /// <param name="state">
  203. /// The context object of the work item. Used for passing arguments to the work item.
  204. /// </param>
  205. /// <param name="postExecuteWorkItemCallback">
  206. /// A delegate to call after the callback completion
  207. /// </param>
  208. /// <param name="workItemPriority">The work item priority</param>
  209. /// <returns>Returns a work item result</returns>
  210. public IWorkItemResult QueueWorkItem( WorkItemCallback callback, object state, PostExecuteWorkItemCallback postExecuteWorkItemCallback,
  211. WorkItemPriority workItemPriority)
  212. {
  213. PreQueueWorkItem();
  214. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state, postExecuteWorkItemCallback, workItemPriority);
  215. Enqueue(workItem);
  216. return workItem.GetWorkItemResult();
  217. }
  218. /// <summary>
  219. /// Queue a work item
  220. /// </summary>
  221. /// <param name="callback">A callback to execute</param>
  222. /// <param name="state">
  223. /// The context object of the work item. Used for passing arguments to the work item.
  224. /// </param>
  225. /// <param name="postExecuteWorkItemCallback">
  226. /// A delegate to call after the callback completion
  227. /// </param>
  228. /// <param name="callToPostExecute">Indicates on which cases to call to the post execute callback</param>
  229. /// <returns>Returns a work item result</returns>
  230. public IWorkItemResult QueueWorkItem( WorkItemCallback callback, object state,
  231. PostExecuteWorkItemCallback postExecuteWorkItemCallback, CallToPostExecute callToPostExecute)
  232. {
  233. PreQueueWorkItem();
  234. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state, postExecuteWorkItemCallback, callToPostExecute);
  235. Enqueue(workItem);
  236. return workItem.GetWorkItemResult();
  237. }
  238. /// <summary>
  239. /// Queue a work item
  240. /// </summary>
  241. /// <param name="callback">A callback to execute</param>
  242. /// <param name="state">
  243. /// The context object of the work item. Used for passing arguments to the work item.
  244. /// </param>
  245. /// <param name="postExecuteWorkItemCallback">
  246. /// A delegate to call after the callback completion
  247. /// </param>
  248. /// <param name="callToPostExecute">Indicates on which cases to call to the post execute callback</param>
  249. /// <param name="workItemPriority">The work item priority</param>
  250. /// <returns>Returns a work item result</returns>
  251. public IWorkItemResult QueueWorkItem( WorkItemCallback callback, object state,
  252. PostExecuteWorkItemCallback postExecuteWorkItemCallback, CallToPostExecute callToPostExecute,
  253. WorkItemPriority workItemPriority)
  254. {
  255. PreQueueWorkItem();
  256. WorkItem workItem = WorkItemFactory.CreateWorkItem(this, WIGStartInfo, callback, state, postExecuteWorkItemCallback, callToPostExecute, workItemPriority);
  257. Enqueue(workItem);
  258. return workItem.GetWorkItemResult();
  259. }
  260. #endregion
  261. #region QueueWorkItem(Action<...>)
  262. public IWorkItemResult QueueWorkItem(Action action)
  263. {
  264. return QueueWorkItem (action, SmartThreadPool.DefaultWorkItemPriority);
  265. }
  266. public IWorkItemResult QueueWorkItem (Action action, WorkItemPriority priority)
  267. {
  268. PreQueueWorkItem ();
  269. WorkItem workItem = WorkItemFactory.CreateWorkItem (
  270. this,
  271. WIGStartInfo,
  272. delegate
  273. {
  274. action.Invoke ();
  275. return null;
  276. }, priority);
  277. Enqueue (workItem);
  278. return workItem.GetWorkItemResult ();
  279. }
  280. public IWorkItemResult QueueWorkItem<T>(Action<T> action, T arg)
  281. {
  282. return QueueWorkItem<T> (action, arg, SmartThreadPool.DefaultWorkItemPriority);
  283. }
  284. public IWorkItemResult QueueWorkItem<T> (Action<T> action, T arg, WorkItemPriority priority)
  285. {
  286. PreQueueWorkItem ();
  287. WorkItem workItem = WorkItemFactory.CreateWorkItem (
  288. this,
  289. WIGStartInfo,
  290. state =>
  291. {
  292. action.Invoke (arg);
  293. return null;
  294. },
  295. WIGStartInfo.FillStateWithArgs ? new object[] { arg } : null, priority);
  296. Enqueue (workItem);
  297. return workItem.GetWorkItemResult ();
  298. }
  299. public IWorkItemResult QueueWorkItem<T1, T2>(Action<T1, T2> action, T1 arg1, T2 arg2)
  300. {
  301. return QueueWorkItem<T1, T2> (action, arg1, arg2, SmartThreadPool.DefaultWorkItemPriority);
  302. }
  303. public IWorkItemResult QueueWorkItem<T1, T2> (Action<T1, T2> action, T1 arg1, T2 arg2, WorkItemPriority priority)
  304. {
  305. PreQueueWorkItem ();
  306. WorkItem workItem = WorkItemFactory.CreateWorkItem (
  307. this,
  308. WIGStartInfo,
  309. state =>
  310. {
  311. action.Invoke (arg1, arg2);
  312. return null;
  313. },
  314. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2 } : null, priority);
  315. Enqueue (workItem);
  316. return workItem.GetWorkItemResult ();
  317. }
  318. public IWorkItemResult QueueWorkItem<T1, T2, T3>(Action<T1, T2, T3> action, T1 arg1, T2 arg2, T3 arg3)
  319. {
  320. return QueueWorkItem<T1, T2, T3> (action, arg1, arg2, arg3, SmartThreadPool.DefaultWorkItemPriority);
  321. ;
  322. }
  323. public IWorkItemResult QueueWorkItem<T1, T2, T3> (Action<T1, T2, T3> action, T1 arg1, T2 arg2, T3 arg3, WorkItemPriority priority)
  324. {
  325. PreQueueWorkItem ();
  326. WorkItem workItem = WorkItemFactory.CreateWorkItem (
  327. this,
  328. WIGStartInfo,
  329. state =>
  330. {
  331. action.Invoke (arg1, arg2, arg3);
  332. return null;
  333. },
  334. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2, arg3 } : null, priority);
  335. Enqueue (workItem);
  336. return workItem.GetWorkItemResult ();
  337. }
  338. public IWorkItemResult QueueWorkItem<T1, T2, T3, T4>(
  339. Action<T1, T2, T3, T4> action, T1 arg1, T2 arg2, T3 arg3, T4 arg4)
  340. {
  341. return QueueWorkItem<T1, T2, T3, T4> (action, arg1, arg2, arg3, arg4,
  342. SmartThreadPool.DefaultWorkItemPriority);
  343. }
  344. public IWorkItemResult QueueWorkItem<T1, T2, T3, T4> (
  345. Action<T1, T2, T3, T4> action, T1 arg1, T2 arg2, T3 arg3, T4 arg4, WorkItemPriority priority)
  346. {
  347. PreQueueWorkItem ();
  348. WorkItem workItem = WorkItemFactory.CreateWorkItem (
  349. this,
  350. WIGStartInfo,
  351. state =>
  352. {
  353. action.Invoke (arg1, arg2, arg3, arg4);
  354. return null;
  355. },
  356. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2, arg3, arg4 } : null, priority);
  357. Enqueue (workItem);
  358. return workItem.GetWorkItemResult ();
  359. }
  360. #endregion
  361. #region QueueWorkItem(Func<...>)
  362. public IWorkItemResult<TResult> QueueWorkItem<TResult>(Func<TResult> func)
  363. {
  364. PreQueueWorkItem();
  365. WorkItem workItem = WorkItemFactory.CreateWorkItem(
  366. this,
  367. WIGStartInfo,
  368. state =>
  369. {
  370. return func.Invoke();
  371. });
  372. Enqueue(workItem);
  373. return new WorkItemResultTWrapper<TResult>(workItem.GetWorkItemResult());
  374. }
  375. public IWorkItemResult<TResult> QueueWorkItem<T, TResult>(Func<T, TResult> func, T arg)
  376. {
  377. PreQueueWorkItem();
  378. WorkItem workItem = WorkItemFactory.CreateWorkItem(
  379. this,
  380. WIGStartInfo,
  381. state =>
  382. {
  383. return func.Invoke(arg);
  384. },
  385. WIGStartInfo.FillStateWithArgs ? new object[] { arg } : null);
  386. Enqueue(workItem);
  387. return new WorkItemResultTWrapper<TResult>(workItem.GetWorkItemResult());
  388. }
  389. public IWorkItemResult<TResult> QueueWorkItem<T1, T2, TResult>(Func<T1, T2, TResult> func, T1 arg1, T2 arg2)
  390. {
  391. PreQueueWorkItem();
  392. WorkItem workItem = WorkItemFactory.CreateWorkItem(
  393. this,
  394. WIGStartInfo,
  395. state =>
  396. {
  397. return func.Invoke(arg1, arg2);
  398. },
  399. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2 } : null);
  400. Enqueue(workItem);
  401. return new WorkItemResultTWrapper<TResult>(workItem.GetWorkItemResult());
  402. }
  403. public IWorkItemResult<TResult> QueueWorkItem<T1, T2, T3, TResult>(
  404. Func<T1, T2, T3, TResult> func, T1 arg1, T2 arg2, T3 arg3)
  405. {
  406. PreQueueWorkItem();
  407. WorkItem workItem = WorkItemFactory.CreateWorkItem(
  408. this,
  409. WIGStartInfo,
  410. state =>
  411. {
  412. return func.Invoke(arg1, arg2, arg3);
  413. },
  414. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2, arg3 } : null);
  415. Enqueue(workItem);
  416. return new WorkItemResultTWrapper<TResult>(workItem.GetWorkItemResult());
  417. }
  418. public IWorkItemResult<TResult> QueueWorkItem<T1, T2, T3, T4, TResult>(
  419. Func<T1, T2, T3, T4, TResult> func, T1 arg1, T2 arg2, T3 arg3, T4 arg4)
  420. {
  421. PreQueueWorkItem();
  422. WorkItem workItem = WorkItemFactory.CreateWorkItem(
  423. this,
  424. WIGStartInfo,
  425. state =>
  426. {
  427. return func.Invoke(arg1, arg2, arg3, arg4);
  428. },
  429. WIGStartInfo.FillStateWithArgs ? new object[] { arg1, arg2, arg3, arg4 } : null);
  430. Enqueue(workItem);
  431. return new WorkItemResultTWrapper<TResult>(workItem.GetWorkItemResult());
  432. }
  433. #endregion
  434. #endregion
  435. }
  436. }