From 6bc7df851fde79cb58ff594eb16a62f21401511c Mon Sep 17 00:00:00 2001
From: Chris R <Tratcher@outlook.com>
Date: Mon, 31 Aug 2015 09:45:21 -0700
Subject: [PATCH] Use new HttpContext.Features API.

---
 .../WebSocketMiddleware.cs                                  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs b/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
index 8d13d9be0e..c3c48d91e9 100644
--- a/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
+++ b/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
@@ -29,12 +29,12 @@ namespace Microsoft.AspNet.WebSockets.Server
         public Task Invoke(HttpContext context)
         {
             // Detect if an opaque upgrade is available. If so, add a websocket upgrade.
-            var upgradeFeature = context.GetFeature<IHttpUpgradeFeature>();
+            var upgradeFeature = context.Features.Get<IHttpUpgradeFeature>();
             if (upgradeFeature != null)
             {
-                if (_options.ReplaceFeature || context.GetFeature<IHttpWebSocketFeature>() == null)
+                if (_options.ReplaceFeature || context.Features.Get<IHttpWebSocketFeature>() == null)
                 {
-                    context.SetFeature<IHttpWebSocketFeature>(new UpgradeHandshake(context, upgradeFeature, _options));
+                    context.Features.Set<IHttpWebSocketFeature>(new UpgradeHandshake(context, upgradeFeature, _options));
                 }
             }