From 57594803fce3c1708e48d35bfc98a104ec6c1d7a Mon Sep 17 00:00:00 2001 From: Chris R Date: Tue, 12 May 2015 11:34:21 -0700 Subject: [PATCH] React to IHttpRequestLifetimeFeature change. --- .../FeatureContext.cs | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/Microsoft.AspNet.Server.WebListener/FeatureContext.cs b/src/Microsoft.AspNet.Server.WebListener/FeatureContext.cs index e8a2120e20..4d42a0db5e 100644 --- a/src/Microsoft.AspNet.Server.WebListener/FeatureContext.cs +++ b/src/Microsoft.AspNet.Server.WebListener/FeatureContext.cs @@ -63,6 +63,7 @@ namespace Microsoft.AspNet.Server.WebListener private X509Certificate2 _clientCert; private ClaimsPrincipal _user; private IAuthenticationHandler _authHandler; + private CancellationToken? _disconnectToken; private Stream _responseStream; private IDictionary _responseHeaders; @@ -375,7 +376,15 @@ namespace Microsoft.AspNet.Server.WebListener CancellationToken IHttpRequestLifetimeFeature.RequestAborted { - get { return _requestContext.DisconnectToken; } + get + { + if (!_disconnectToken.HasValue) + { + _disconnectToken = _requestContext.DisconnectToken; + } + return _disconnectToken.Value; + } + set { _disconnectToken = value; } } void IHttpRequestLifetimeFeature.Abort()