[Internal] Benchmark: Refactors code to make Memory Stream capacity c…

…onfigurable (#3624) Co-authored-by: Sourabh Jain <sourabhjain@microsoft.com>
Azure · Jan 4, 2023 · 5c05fbb · 5c05fbb
1 parent 814e72e
commit 5c05fbb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/Microsoft.Azure.Cosmos.Samples/Tools/Benchmark/BenchmarkConfig.cs b/Microsoft.Azure.Cosmos.Samples/Tools/Benchmark/BenchmarkConfig.cs
@@ -143,7 +143,9 @@ internal void Print()
                 Utility.TeeTraceInformation($"{nameof(BenchmarkConfig)} arguments");
                 Utility.TeeTraceInformation($"IsServerGC: {GCSettings.IsServerGC}");
                 Utility.TeeTraceInformation("--------------------------------------------------------------------- ");
-                Utility.TeeTraceInformation(JsonHelper.ToString(this));
+                Utility.TeeTraceInformation(JsonHelper.ToString(
+                    input: this, 
+                    capacity: 2048));
                 Utility.TeeTraceInformation("--------------------------------------------------------------------- ");
                 Utility.TeeTraceInformation(string.Empty);
             }

diff --git a/Microsoft.Azure.Cosmos.Samples/Tools/Benchmark/JsonHelper.cs b/Microsoft.Azure.Cosmos.Samples/Tools/Benchmark/JsonHelper.cs
@@ -17,9 +17,9 @@ internal static class JsonHelper
                 });
         private const int DefaultCapacity = 1024;
 
-        public static string ToString<T>(T input)
+        public static string ToString<T>(T input, int capacity = JsonHelper.DefaultCapacity)
         {
-            using (MemoryStream stream = JsonHelper.ToStream(input))
+            using (MemoryStream stream = JsonHelper.ToStream(input, capacity))
             using (StreamReader sr = new StreamReader(stream))
             {
                 return sr.ReadToEnd();
@@ -31,15 +31,15 @@ public static T Deserialize<T>(string payload)
             return JsonConvert.DeserializeObject<T>(payload);
         }
 
-        public static MemoryStream ToStream<T>(T input)
+        public static MemoryStream ToStream<T>(T input, int capacity = JsonHelper.DefaultCapacity)
         {
-            byte[] blob = System.Buffers.ArrayPool<byte>.Shared.Rent(JsonHelper.DefaultCapacity);
-            MemoryStream memStreamPayload = new MemoryStream(blob, 0, JsonHelper.DefaultCapacity, writable: true, publiclyVisible: true);
+            byte[] blob = System.Buffers.ArrayPool<byte>.Shared.Rent(capacity);
+            MemoryStream memStreamPayload = new MemoryStream(blob, 0, capacity, writable: true, publiclyVisible: true);
             memStreamPayload.SetLength(0);
             memStreamPayload.Position = 0;
             using (StreamWriter streamWriter = new StreamWriter(memStreamPayload,
                 encoding: JsonHelper.DefaultEncoding,
-                bufferSize: JsonHelper.DefaultCapacity,
+                bufferSize: capacity,
                 leaveOpen: true))
             {
                 using (JsonWriter writer = new JsonTextWriter(streamWriter))