span

BrennanConroy · BrennanConroy · commit e405b3091396 · 2024-10-23T12:10:53.000-07:00
diff --git a/src/Benchmarks/Middleware/MiddlewareHelpers.cs b/src/Benchmarks/Middleware/MiddlewareHelpers.cs
@@ -40,24 +40,26 @@ public static async Task RenderFortunesHtml(IEnumerable<Fortune> model, HttpCont
 
             await httpContext.Response.StartAsync();
 
+            var writer = new BufferWriter<byte>(httpContext.Response.BodyWriter);
+
             Encoding.UTF8.GetBytes(
                 "<!DOCTYPE html><html><head><title>Fortunes</title></head><body><table><tr><th>id</th><th>message</th></tr>",
-                httpContext.Response.BodyWriter);
+                writer);
 
             foreach (var item in model)
             {
-                Encoding.UTF8.GetBytes("<tr><td>", httpContext.Response.BodyWriter);
-                Encoding.UTF8.GetBytes(item.Id.ToString(CultureInfo.InvariantCulture), httpContext.Response.BodyWriter);
-                Encoding.UTF8.GetBytes("</td><td>", httpContext.Response.BodyWriter);
-                EncodeToPipe(httpContext.Response.BodyWriter, htmlEncoder, item.Message);
-                Encoding.UTF8.GetBytes("</td></tr>", httpContext.Response.BodyWriter);
+                Encoding.UTF8.GetBytes("<tr><td>", writer);
+                Encoding.UTF8.GetBytes(item.Id.ToString(CultureInfo.InvariantCulture), writer);
+                Encoding.UTF8.GetBytes("</td><td>", writer);
+                EncodeToPipe(writer, htmlEncoder, item.Message);
+                Encoding.UTF8.GetBytes("</td></tr>", writer);
             }
 
-            Encoding.UTF8.GetBytes("</table></body></html>", httpContext.Response.BodyWriter);
+            Encoding.UTF8.GetBytes("</table></body></html>", writer);
 
             await httpContext.Response.BodyWriter.FlushAsync();
 
-            static void EncodeToPipe(PipeWriter writer, HtmlEncoder htmlEncoder, string item)
+            static void EncodeToPipe(IBufferWriter<byte> writer, HtmlEncoder htmlEncoder, string item)
             {
                 Span<char> buffer = stackalloc char[256];
                 int remaining = item.Length;
@@ -70,4 +72,35 @@ static void EncodeToPipe(PipeWriter writer, HtmlEncoder htmlEncoder, string item
             }
         }
     }
+
+    internal class BufferWriter<T> : IBufferWriter<T>
+    {
+        private readonly IBufferWriter<T> _inner;
+        private Memory<T> _memory;
+
+        public BufferWriter(IBufferWriter<T> writer)
+        {
+            _inner = writer;
+        }
+
+        public void Advance(int count)
+        {
+            _memory = _memory.Slice(count);
+            _inner.Advance(count);
+        }
+
+        public Memory<T> GetMemory(int sizeHint = 0)
+        {
+            if (_memory.Length == 0 || _memory.Length < sizeHint)
+            {
+                _memory = _inner.GetMemory(sizeHint);
+            }
+            return _memory;
+        }
+
+        public Span<T> GetSpan(int sizeHint = 0)
+        {
+            return GetMemory(sizeHint).Span;
+        }
+    }
 }
