Ensure shutdown handler runs asynchronously.

Author: tintoy

src/Server/Handlers/ShutdownHandler.cs

@@ -13,12 +13,19 @@ public class ShutdownHandler : IShutdownHandler
 
         private readonly TaskCompletionSource<bool> _shutdownSource = new TaskCompletionSource<bool>(TaskContinuationOptions.LongRunning);
         public Task WasShutDown => _shutdownSource.Task;
-        public Task Handle(object request, CancellationToken token)
+        public async Task Handle(object request, CancellationToken token)
         {
+            await Task.Yield(); // Ensure shutdown handler runs asynchronously.
+
             ShutdownRequested = true;
-            Shutdown?.Invoke(ShutdownRequested);
-            _shutdownSource.SetResult(true); // after all event sinks were notified
-            return Task.CompletedTask;
+            try
+            {
+                Shutdown?.Invoke(ShutdownRequested);
+            }
+            finally
+            {
+                _shutdownSource.SetResult(true); // after all event sinks were notified
+            }
         }
     }
 }