reacting to new http context

Author: jkotalik

src/Http/Http/src/Internal/ReusableHttpRequest.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.IO.Pipelines;
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
@@ -18,6 +19,8 @@ public sealed class ReusableHttpRequest : HttpRequest
         private readonly static Func<HttpRequest, IFormFeature> _newFormFeature = r => new FormFeature(r);
         private readonly static Func<IFeatureCollection, IRequestCookiesFeature> _newRequestCookiesFeature = f => new RequestCookiesFeature(f);
         private readonly static Func<IFeatureCollection, IRouteValuesFeature> _newRouteValuesFeature = f => new RouteValuesFeature();
+        private readonly static Func<HttpContext, IRequestBodyPipeFeature> _newRequestBodyPipeFeature = context => new RequestBodyPipeFeature(context);
+
 
         private HttpContext _context;
         private FeatureReferences<FeatureInterfaces> _features;
@@ -56,6 +59,9 @@ public void Uninitialize()
         private IRouteValuesFeature RouteValuesFeature =>
             _features.Fetch(ref _features.Cache.RouteValues, _newRouteValuesFeature);
 
+        private IRequestBodyPipeFeature RequestBodyPipeFeature =>
+            _features.Fetch(ref _features.Cache.BodyPipe, this.HttpContext, _newRequestBodyPipeFeature);
+
         public override PathString PathBase
         {
             get { return new PathString(HttpRequestFeature.PathBase); }
@@ -161,13 +167,20 @@ public override RouteValueDictionary RouteValues
             set { RouteValuesFeature.RouteValues = value; }
         }
 
+        public override PipeReader BodyPipe
+        {
+            get { return RequestBodyPipeFeature.RequestBodyPipe; }
+            set { RequestBodyPipeFeature.RequestBodyPipe = value; }
+        }
+
         struct FeatureInterfaces
         {
             public IHttpRequestFeature Request;
             public IQueryFeature Query;
             public IFormFeature Form;
             public IRequestCookiesFeature Cookies;
             public IRouteValuesFeature RouteValues;
+            public IRequestBodyPipeFeature BodyPipe;
         }
     }
 }

src/Http/Http/src/Internal/ReusableHttpResponse.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.IO.Pipelines;
 using System.Text;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Http.Features;
@@ -13,6 +14,7 @@ public sealed class ReusableHttpResponse : HttpResponse
         // Lambdas hoisted to static readonly fields to improve inlining https://github.com/dotnet/roslyn/issues/13624
         private readonly static Func<IFeatureCollection, IHttpResponseFeature> _nullResponseFeature = f => null;
         private readonly static Func<IFeatureCollection, IResponseCookiesFeature> _newResponseCookiesFeature = f => new ResponseCookiesFeature(f);
+        private readonly static Func<HttpContext, IResponseBodyPipeFeature> _newResponseBodyPipeFeature = context => new ResponseBodyPipeFeature(context);
 
         private HttpContext _context;
         private FeatureReferences<FeatureInterfaces> _features;
@@ -39,7 +41,8 @@ public void Uninitialize()
 
         private IResponseCookiesFeature ResponseCookiesFeature =>
             _features.Fetch(ref _features.Cache.Cookies, _newResponseCookiesFeature);
-
+        private IResponseBodyPipeFeature ResponseBodyPipeFeature =>
+            _features.Fetch(ref _features.Cache.BodyPipe, this.HttpContext, _newResponseBodyPipeFeature);
 
         public override HttpContext HttpContext { get { return _context; } }
 
@@ -90,6 +93,12 @@ public override IResponseCookies Cookies
             get { return ResponseCookiesFeature.Cookies; }
         }
 
+        public override PipeWriter BodyPipe
+        {
+            get { return ResponseBodyPipeFeature.ResponseBodyPipe; }
+            set { ResponseBodyPipeFeature.ResponseBodyPipe = value; }
+        }
+
         public override bool HasStarted
         {
             get { return HttpResponseFeature.HasStarted; }
@@ -133,6 +142,7 @@ struct FeatureInterfaces
         {
             public IHttpResponseFeature Response;
             public IResponseCookiesFeature Cookies;
+            public IResponseBodyPipeFeature BodyPipe;
         }
     }
 }