fix keep alive implementation

maksimkim · maksimkim · commit 266d847839c7 · 2018-10-10T01:11:36.000-07:00
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);
         }

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,9 @@ public override ValueTask WriteAsync(IChannelHandlerContext context, object mess`
`52`	`52`	`}`
`53`	`53`	`if (message is ILastHttpContent && !this.ShouldKeepAlive())`
`54`	`54`	`{`
`55`		`- return new ValueTask(base.WriteAsync(context, message).CloseOnComplete(context));`
	`55`	`+ Task task = base.WriteAsync(context, message).AsTask();`
	`56`	`+ task.CloseOnComplete(context.Channel);`
	`57`	`+ return new ValueTask(task);`
`56`	`58`	`}`
`57`	`59`	`return base.WriteAsync(context, message);`
`58`	`60`	`}`