diff --git a/src/Tgstation.Server.Host/GraphQL/Subscription.cs b/src/Tgstation.Server.Host/GraphQL/Subscription.cs index d47d83a8d6..c7c9c80669 100644 --- a/src/Tgstation.Server.Host/GraphQL/Subscription.cs +++ b/src/Tgstation.Server.Host/GraphQL/Subscription.cs @@ -1,4 +1,5 @@ using System; +using System.Threading; using System.Threading.Tasks; using HotChocolate; @@ -34,16 +35,18 @@ namespace Tgstation.Server.Host.GraphQL /// The . /// The . /// The for the request. + /// The for the operation. /// A resulting in a of the for the . public ValueTask> SessionInvalidatedStream( [Service] ITopicEventReceiver receiver, [Service] ISessionInvalidationTracker invalidationTracker, - [Service] IAuthenticationContext authenticationContext) + [Service] IAuthenticationContext authenticationContext, + CancellationToken cancellationToken) { ArgumentNullException.ThrowIfNull(receiver); ArgumentNullException.ThrowIfNull(invalidationTracker); - var subscription = receiver.SubscribeAsync(SessionInvalidatedTopic(authenticationContext)); + var subscription = receiver.SubscribeAsync(SessionInvalidatedTopic(authenticationContext), cancellationToken); invalidationTracker.TrackSession(authenticationContext); return subscription; }