fix keep alive implementation

Azure · Oct 10, 2018 · 266d847 · 266d847
1 parent 1236174
commit 266d847
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/DotNetty.Codecs.Http/HttpServerKeepAliveHandler.cs b/src/DotNetty.Codecs.Http/HttpServerKeepAliveHandler.cs
@@ -52,7 +52,9 @@ public override ValueTask WriteAsync(IChannelHandlerContext context, object mess
             }
             if (message is ILastHttpContent && !this.ShouldKeepAlive())
             {
-                return new ValueTask(base.WriteAsync(context, message).CloseOnComplete(context));
+                Task task = base.WriteAsync(context, message).AsTask();
+                task.CloseOnComplete(context.Channel);
+                return new ValueTask(task);
             }
             return base.WriteAsync(context, message);
         }