Permissions, RateControl and Auditing

Libraries/Esiur/Core/AsyncReplyBuilderGeneric.cs

@@ -3,12 +3,14 @@ using System;
 using System.Collections.Generic;
 using System.Runtime.CompilerServices;
 using System.Text;
+using System.Threading;
 
 namespace Esiur.Core;
 
 public class AsyncReplyBuilder<T>
 {
     AsyncReply<T> reply;
+    int hasSuspended;
 
     AsyncReplyBuilder(AsyncReply<T> reply)
     {
@@ -33,7 +35,9 @@ public class AsyncReplyBuilder<T>
 
     public void SetException(Exception exception)
     {
-        reply.TriggerError(exception);
+        reply.TriggerErrorFromBuilder(
+            exception,
+            preserveSourceForAwait: Volatile.Read(ref hasSuspended) == 0);
     }
 
     public void SetResult(T result)
@@ -46,6 +50,7 @@ public class AsyncReplyBuilder<T>
         where TAwaiter : INotifyCompletion
         where TStateMachine : IAsyncStateMachine
     {
+        Volatile.Write(ref hasSuspended, 1);
         awaiter.OnCompleted(stateMachine.MoveNext);
     }
 
@@ -54,6 +59,7 @@ public class AsyncReplyBuilder<T>
         where TAwaiter : ICriticalNotifyCompletion
         where TStateMachine : IAsyncStateMachine
     {
+        Volatile.Write(ref hasSuspended, 1);
         awaiter.UnsafeOnCompleted(stateMachine.MoveNext);
     }