TaskScheduler.cs 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. #if NET_4_0
  2. //
  3. // TaskScheduler.cs
  4. //
  5. // Author:
  6. // Jérémie "Garuma" Laval <[email protected]>
  7. //
  8. // Copyright (c) 2009 Jérémie "Garuma" Laval
  9. //
  10. // Permission is hereby granted, free of charge, to any person obtaining a copy
  11. // of this software and associated documentation files (the "Software"), to deal
  12. // in the Software without restriction, including without limitation the rights
  13. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  14. // copies of the Software, and to permit persons to whom the Software is
  15. // furnished to do so, subject to the following conditions:
  16. //
  17. // The above copyright notice and this permission notice shall be included in
  18. // all copies or substantial portions of the Software.
  19. //
  20. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  23. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  25. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  26. // THE SOFTWARE.
  27. using System;
  28. using System.Threading;
  29. using System.Collections.Generic;
  30. namespace System.Threading.Tasks
  31. {
  32. public abstract class TaskScheduler
  33. {
  34. static TaskScheduler defaultScheduler = new Scheduler ();
  35. [ThreadStatic]
  36. static TaskScheduler currentScheduler;
  37. int id;
  38. static int lastId = int.MinValue;
  39. protected TaskScheduler ()
  40. {
  41. this.id = Interlocked.Increment (ref lastId);
  42. }
  43. // FIXME: Probably not correct
  44. public static TaskScheduler FromCurrentSynchronizationContext ()
  45. {
  46. return Current;
  47. }
  48. public static TaskScheduler Default {
  49. get {
  50. return defaultScheduler;
  51. }
  52. }
  53. public static TaskScheduler Current {
  54. get {
  55. if (Task.Current != null && currentScheduler != null)
  56. return currentScheduler;
  57. return defaultScheduler;
  58. }
  59. internal set {
  60. currentScheduler = value;
  61. }
  62. }
  63. public int Id {
  64. get {
  65. return id;
  66. }
  67. }
  68. public virtual int MaximumConcurrencyLevel {
  69. get {
  70. return Environment.ProcessorCount;
  71. }
  72. }
  73. protected abstract IEnumerable<Task> GetScheduledTasks ();
  74. protected internal abstract void QueueTask (Task task);
  75. protected internal virtual bool TryDequeue (Task task)
  76. {
  77. throw new NotSupportedException ();
  78. }
  79. internal protected bool TryExecuteTask (Task task)
  80. {
  81. throw new NotSupportedException ();
  82. }
  83. protected abstract bool TryExecuteTaskInline (Task task, bool taskWasPreviouslyQueued);
  84. }
  85. }
  86. #endif