Merge pull request #16676 from abpframework/AbpMemoryPoolHttpResponseStreamWriterFactory

Add `AbpMemoryPoolHttpResponseStreamWriterFactory`.

framework/src/Volo.Abp.AspNetCore.Mvc/Volo/Abp/AspNetCore/Mvc/AbpAspNetCoreMvcModule.cs

@@ -28,6 +28,7 @@ using Volo.Abp.AspNetCore.Mvc.ApiExploring;
 using Volo.Abp.AspNetCore.Mvc.Conventions;
 using Volo.Abp.AspNetCore.Mvc.DataAnnotations;
 using Volo.Abp.AspNetCore.Mvc.DependencyInjection;
+using Volo.Abp.AspNetCore.Mvc.Infrastructure;
 using Volo.Abp.AspNetCore.Mvc.Json;
 using Volo.Abp.AspNetCore.Mvc.Localization;
 using Volo.Abp.AspNetCore.VirtualFileSystem;
@@ -201,6 +202,8 @@ public class AbpAspNetCoreMvcModule : AbpModule
         {
             options.DisableModule("abp");
         });
+
+        context.Services.Replace(ServiceDescriptor.Singleton<IHttpResponseStreamWriterFactory, AbpMemoryPoolHttpResponseStreamWriterFactory>());
     }
 
     public override void PostConfigureServices(ServiceConfigurationContext context)

framework/src/Volo.Abp.AspNetCore.Mvc/Volo/Abp/AspNetCore/Mvc/Infrastructure/AbpMemoryPoolHttpResponseStreamWriterFactory.cs

@@ -0,0 +1,40 @@
+using System;
+using System.Buffers;
+using System.IO;
+using System.Text;
+using Microsoft.AspNetCore.Mvc.Infrastructure;
+using Microsoft.AspNetCore.WebUtilities;
+
+namespace Volo.Abp.AspNetCore.Mvc.Infrastructure;
+
+/// <summary>
+/// https://github.com/dotnet/aspnetcore/issues/40928#issuecomment-1450063613
+/// </summary>
+public class AbpMemoryPoolHttpResponseStreamWriterFactory : IHttpResponseStreamWriterFactory
+{
+    public const int DefaultBufferSize = 32 * 1024;
+
+    private readonly ArrayPool<byte> _bytePool;
+    private readonly ArrayPool<char> _charPool;
+
+    public AbpMemoryPoolHttpResponseStreamWriterFactory(ArrayPool<byte> bytePool, ArrayPool<char> charPool)
+    {
+        _bytePool = bytePool ?? throw new ArgumentNullException(nameof(bytePool));
+        _charPool = charPool ?? throw new ArgumentNullException(nameof(charPool));
+    }
+
+    public TextWriter CreateWriter(Stream stream, Encoding encoding)
+    {
+        if (stream == null)
+        {
+            throw new ArgumentNullException(nameof(stream));
+        }
+
+        if (encoding == null)
+        {
+            throw new ArgumentNullException(nameof(encoding));
+        }
+
+        return new HttpResponseStreamWriter(stream, encoding, DefaultBufferSize, _bytePool, _charPool);
+    }
+}