Added OnResponseCompleted to HttpResponse

Author: ajaybhargavb

src/Microsoft.AspNet.Http.Core/DefaultHttpResponse.cs

@@ -6,9 +6,6 @@
 using System.IO;
 using System.Linq;
 using System.Security.Claims;
-using System.Text;
-using System.Threading.Tasks;
-using Microsoft.AspNet.Http;
 using Microsoft.AspNet.Http.Infrastructure;
 using Microsoft.AspNet.Http.Security;
 using Microsoft.AspNet.FeatureModel;
@@ -115,6 +112,11 @@ public override void OnSendingHeaders(Action<object> callback, object state)
             HttpResponseFeature.OnSendingHeaders(callback, state);
         }
 
+        public override void OnResponseCompleted(Action<object> callback, object state)
+        {
+            HttpResponseFeature.OnResponseCompleted(callback, state);
+        }
+
         public override void Redirect(string location, bool permanent)
         {
             if (permanent)

src/Microsoft.AspNet.Http.Core/HttpResponseFeature.cs

@@ -34,5 +34,10 @@ public void OnSendingHeaders(Action<object> callback, object state)
         {
             throw new NotSupportedException();
         }
+
+        public void OnResponseCompleted(Action<object> callback, object state)
+        {
+            throw new NotSupportedException();
+        }
     }
 }

src/Microsoft.AspNet.Http.Interfaces/IHttpResponseFeature.cs

@@ -15,5 +15,6 @@ public interface IHttpResponseFeature
         Stream Body { get; set; }
         bool HeadersSent { get; }
         void OnSendingHeaders(Action<object> callback, object state);
+        void OnResponseCompleted(Action<object> callback, object state);
     }
 }

src/Microsoft.AspNet.Http/HttpResponse.cs

@@ -27,6 +27,8 @@ public abstract class HttpResponse
 
         public abstract void OnSendingHeaders(Action<object> callback, object state);
 
+        public abstract void OnResponseCompleted(Action<object> callback, object state);
+
         public virtual void Redirect(string location)
         {
             Redirect(location, permanent: false);

src/Microsoft.AspNet.Owin/OwinConstants.cs

@@ -70,6 +70,7 @@ internal static class CommonKeys
             public const string AppName = "host.AppName";
             public const string Capabilities = "server.Capabilities";
             public const string OnSendingHeaders = "server.OnSendingHeaders";
+            public const string OnResponseCompleted = "server.OnResponseCompleted";
             public const string OnAppDisposing = "host.OnAppDisposing";
             public const string Scheme = "scheme";
             public const string Host = "host";

src/Microsoft.AspNet.Owin/OwinFeatureCollection.cs

@@ -155,6 +155,16 @@ void IHttpResponseFeature.OnSendingHeaders(Action<object> callback, object state
             register(callback, state);
         }
 
+        void IHttpResponseFeature.OnResponseCompleted(Action<object> callback, object state)
+        {
+            var register = Prop<Action<Action<object>, object>>(OwinConstants.CommonKeys.OnResponseCompleted);
+            if (register == null)
+            {
+                throw new NotSupportedException(OwinConstants.CommonKeys.OnResponseCompleted);
+            }
+            register(callback, state);
+        }
+
         IPAddress IHttpConnectionFeature.RemoteIpAddress
         {
             get { return IPAddress.Parse(Prop<string>(OwinConstants.CommonKeys.RemoteIpAddress)); }