errorbrecon/shutdowndelay

2 files changed, 13 insertions, 0 deletions

diff --git a/src/SignalR/common/Http.Connections/src/ConnectionOptions.cs b/src/SignalR/common/Http.Connections/src/ConnectionOptions.cs
index 8e2c23f7d1..ba4c0226d1 100644
--- a/src/SignalR/common/Http.Connections/src/ConnectionOptions.cs
+++ b/src/SignalR/common/Http.Connections/src/ConnectionOptions.cs
@@ -15,6 +15,11 @@ public class ConnectionOptions
 
     internal List<Func<Task>> ShutdownCallbacks = new();
 
+    /// <summary>
+    /// TODO
+    /// </summary>
+    /// <param name="callback"></param>
+    /// <returns></returns>
     public IDisposable RegisterBeforeShutdown(Func<Task> callback)
     {
         ShutdownCallbacks.Add(callback);
diff --git a/src/SignalR/samples/SignalRSamples/Startup.cs b/src/SignalR/samples/SignalRSamples/Startup.cs
index 10386df942..2ddb1d1cea 100644
--- a/src/SignalR/samples/SignalRSamples/Startup.cs
+++ b/src/SignalR/samples/SignalRSamples/Startup.cs
@@ -32,6 +32,14 @@ public class Startup
         var sn = new ShutdownNotification();
         services.AddSingleton(sn);
 
+        services.AddConnections(o =>
+        {
+            o.RegisterBeforeShutdown(() =>
+            {
+                sn.Cancel();
+                return Task.CompletedTask;
+            });
+        });
         services.AddSignalR()
         .AddMessagePackProtocol();
         //.AddStackExchangeRedis();