AsyncETTaskMethodBuilder.cs 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. using System;
  2. using System.Diagnostics;
  3. using System.Runtime.CompilerServices;
  4. using System.Security;
  5. namespace ET
  6. {
  7. public struct ETAsyncTaskMethodBuilder
  8. {
  9. private IStateMachineWrap iStateMachineWrap;
  10. private ETTask tcs;
  11. // 1. Static Create method.
  12. [DebuggerHidden]
  13. public static ETAsyncTaskMethodBuilder Create()
  14. {
  15. ETAsyncTaskMethodBuilder builder = new() { tcs = ETTask.Create(true) };
  16. return builder;
  17. }
  18. // 2. TaskLike Task property.
  19. [DebuggerHidden]
  20. public ETTask Task => this.tcs;
  21. // 3. SetException
  22. [DebuggerHidden]
  23. public void SetException(Exception exception)
  24. {
  25. if (this.iStateMachineWrap != null)
  26. {
  27. this.iStateMachineWrap.Recycle();
  28. this.iStateMachineWrap = null;
  29. }
  30. this.tcs.SetException(exception);
  31. }
  32. // 4. SetResult
  33. [DebuggerHidden]
  34. public void SetResult()
  35. {
  36. if (this.iStateMachineWrap != null)
  37. {
  38. this.iStateMachineWrap.Recycle();
  39. this.iStateMachineWrap = null;
  40. }
  41. this.tcs.SetResult();
  42. }
  43. // 5. AwaitOnCompleted
  44. [DebuggerHidden]
  45. public void AwaitOnCompleted<TAwaiter, TStateMachine>(ref TAwaiter awaiter, ref TStateMachine stateMachine) where TAwaiter : INotifyCompletion where TStateMachine : IAsyncStateMachine
  46. {
  47. this.iStateMachineWrap ??= StateMachineWrap<TStateMachine>.Fetch(ref stateMachine);
  48. awaiter.OnCompleted(this.iStateMachineWrap.MoveNext);
  49. }
  50. // 6. AwaitUnsafeOnCompleted
  51. [DebuggerHidden]
  52. [SecuritySafeCritical]
  53. public void AwaitUnsafeOnCompleted<TAwaiter, TStateMachine>(ref TAwaiter awaiter, ref TStateMachine stateMachine) where TAwaiter : ICriticalNotifyCompletion where TStateMachine : IAsyncStateMachine
  54. {
  55. this.iStateMachineWrap ??= StateMachineWrap<TStateMachine>.Fetch(ref stateMachine);
  56. awaiter.UnsafeOnCompleted(this.iStateMachineWrap.MoveNext);
  57. }
  58. // 7. Start
  59. [DebuggerHidden]
  60. public void Start<TStateMachine>(ref TStateMachine stateMachine) where TStateMachine : IAsyncStateMachine
  61. {
  62. stateMachine.MoveNext();
  63. }
  64. // 8. SetStateMachine
  65. [DebuggerHidden]
  66. public void SetStateMachine(IAsyncStateMachine stateMachine)
  67. {
  68. }
  69. }
  70. public struct ETAsyncTaskMethodBuilder<T>
  71. {
  72. private IStateMachineWrap iStateMachineWrap;
  73. private ETTask<T> tcs;
  74. // 1. Static Create method.
  75. [DebuggerHidden]
  76. public static ETAsyncTaskMethodBuilder<T> Create()
  77. {
  78. ETAsyncTaskMethodBuilder<T> builder = new ETAsyncTaskMethodBuilder<T>() { tcs = ETTask<T>.Create(true) };
  79. return builder;
  80. }
  81. // 2. TaskLike Task property.
  82. [DebuggerHidden]
  83. public ETTask<T> Task => this.tcs;
  84. // 3. SetException
  85. [DebuggerHidden]
  86. public void SetException(Exception exception)
  87. {
  88. if (this.iStateMachineWrap != null)
  89. {
  90. this.iStateMachineWrap.Recycle();
  91. this.iStateMachineWrap = null;
  92. }
  93. this.tcs.SetException(exception);
  94. }
  95. // 4. SetResult
  96. [DebuggerHidden]
  97. public void SetResult(T ret)
  98. {
  99. if (this.iStateMachineWrap != null)
  100. {
  101. this.iStateMachineWrap.Recycle();
  102. this.iStateMachineWrap = null;
  103. }
  104. this.tcs.SetResult(ret);
  105. }
  106. // 5. AwaitOnCompleted
  107. [DebuggerHidden]
  108. public void AwaitOnCompleted<TAwaiter, TStateMachine>(ref TAwaiter awaiter, ref TStateMachine stateMachine) where TAwaiter : INotifyCompletion where TStateMachine : IAsyncStateMachine
  109. {
  110. this.iStateMachineWrap ??= StateMachineWrap<TStateMachine>.Fetch(ref stateMachine);
  111. awaiter.OnCompleted(this.iStateMachineWrap.MoveNext);
  112. }
  113. // 6. AwaitUnsafeOnCompleted
  114. [DebuggerHidden]
  115. [SecuritySafeCritical]
  116. public void AwaitUnsafeOnCompleted<TAwaiter, TStateMachine>(ref TAwaiter awaiter, ref TStateMachine stateMachine) where TAwaiter : ICriticalNotifyCompletion where TStateMachine : IAsyncStateMachine
  117. {
  118. this.iStateMachineWrap ??= StateMachineWrap<TStateMachine>.Fetch(ref stateMachine);
  119. awaiter.UnsafeOnCompleted(this.iStateMachineWrap.MoveNext);
  120. }
  121. // 7. Start
  122. [DebuggerHidden]
  123. public void Start<TStateMachine>(ref TStateMachine stateMachine) where TStateMachine : IAsyncStateMachine
  124. {
  125. stateMachine.MoveNext();
  126. }
  127. // 8. SetStateMachine
  128. [DebuggerHidden]
  129. public void SetStateMachine(IAsyncStateMachine stateMachine)
  130. {
  131. }
  132. }
  133. }