Use PoolingAsyncValueTaskMethodBuilder on various ReadAsync methods. (#35011)

Author: adityamandaleeka

src/Servers/Kestrel/Core/src/Internal/Http/Http1ContentLengthMessageBody.cs

@@ -5,6 +5,7 @@
 using System.Diagnostics;
 using System.Globalization;
 using System.IO.Pipelines;
+using System.Runtime.CompilerServices;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Connections;
@@ -29,6 +30,7 @@ public Http1ContentLengthMessageBody(Http1Connection context, long contentLength
             _unexaminedInputLength = contentLength;
         }
 
+        [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
         public override async ValueTask<ReadResult> ReadAsyncInternal(CancellationToken cancellationToken = default)
         {
             VerifyIsNotReading();

src/Servers/Kestrel/Core/src/Internal/Http/Http1UpgradeMessageBody.cs

@@ -3,6 +3,7 @@
 
 using System;
 using System.IO.Pipelines;
+using System.Runtime.CompilerServices;
 using System.Threading;
 using System.Threading.Tasks;
 
@@ -92,6 +93,7 @@ public override ValueTask<ReadResult> ReadAsyncInternal(CancellationToken cancel
             return new ValueTask<ReadResult>(readResult);
         }
 
+        [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
         private async ValueTask<ReadResult> ReadAsyncInternalAwaited(ValueTask<ReadResult> readTask, CancellationToken cancellationToken = default)
         {
             var readResult = await readTask;

src/Servers/Kestrel/Core/src/Internal/Http2/Http2MessageBody.cs

@@ -5,6 +5,7 @@
 using System.Diagnostics;
 using System.Globalization;
 using System.IO.Pipelines;
+using System.Runtime.CompilerServices;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Connections;
@@ -89,6 +90,7 @@ public override bool TryRead(out ReadResult readResult)
             return hasResult;
         }
 
+        [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
         public override async ValueTask<ReadResult> ReadAsync(CancellationToken cancellationToken = default)
         {
             await TryStartAsync();

src/Servers/Kestrel/Core/src/Internal/Http3/Http3MessageBody.cs

@@ -4,6 +4,7 @@
 using System;
 using System.Globalization;
 using System.IO.Pipelines;
+using System.Runtime.CompilerServices;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Connections;
@@ -68,6 +69,7 @@ public override bool TryRead(out ReadResult readResult)
             return hasResult;
         }
 
+        [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
         public override async ValueTask<ReadResult> ReadAsync(CancellationToken cancellationToken = default)
         {
             await TryStartAsync();

src/Shared/ServerInfrastructure/DuplexPipeStream.cs

@@ -8,6 +8,7 @@
 using System.Threading.Tasks;
 using System.Buffers;
 using Microsoft.AspNetCore.Internal;
+using System.Runtime.CompilerServices;
 
 #nullable enable
 
@@ -112,6 +113,7 @@ public override Task FlushAsync(CancellationToken cancellationToken)
             return _output.FlushAsync(cancellationToken).GetAsTask();
         }
 
+        [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
         private async ValueTask<int> ReadAsyncInternal(Memory<byte> destination, CancellationToken cancellationToken)
         {
             while (true)