Code cleanup for serializers.

Azure · Nov 11, 2024 · 86d07c5 · 86d07c5
1 parent 7230e75
commit 86d07c5
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 33 deletions.
diff --git a/Microsoft.Azure.Cosmos/src/Serializer/CosmosJsonDotNetSerializer.cs b/Microsoft.Azure.Cosmos/src/Serializer/CosmosJsonDotNetSerializer.cs
@@ -98,29 +98,28 @@ public override T FromStream<T>(Stream stream)
 
                 JsonSerializer jsonSerializer = this.GetSerializer();
 
-                if (stream is CloneableStream cloneableStream)
+                using (CosmosBufferedStreamWrapper bufferedStream = new (stream, shouldDisposeInnerStream: false))
                 {
-                    using (CosmosBufferedStreamWrapper bufferedStream = new (cloneableStream, shouldDisposeInnerStream: false))
+                    if (bufferedStream.GetJsonSerializationFormat() == Json.JsonSerializationFormat.Binary)
                     {
-                        if (bufferedStream.GetJsonSerializationFormat() == Json.JsonSerializationFormat.Binary)
-                        {
-                            byte[] content = bufferedStream.ReadAll();
+                        byte[] content = bufferedStream.ReadAll();
 
-                            using Json.Interop.CosmosDBToNewtonsoftReader reader = new (
-                                jsonReader: Json.JsonReader.Create(
-                                    jsonSerializationFormat: Json.JsonSerializationFormat.Binary,
-                                    buffer: content));
+                        using Json.Interop.CosmosDBToNewtonsoftReader reader = new (
+                            jsonReader: Json.JsonReader.Create(
+                                jsonSerializationFormat: Json.JsonSerializationFormat.Binary,
+                                buffer: content));
 
-                            return jsonSerializer.Deserialize<T>(reader);
-                        }
+                        return jsonSerializer.Deserialize<T>(reader);
                     }
-                }
-
-                using (StreamReader sr = new (stream))
-                {
-                    using (JsonTextReader jsonTextReader = new (sr))
+                    else
                     {
-                        return jsonSerializer.Deserialize<T>(jsonTextReader);
+                        using (StreamReader sr = new (bufferedStream))
+                        {
+                            using (JsonTextReader jsonTextReader = new (sr))
+                            {
+                                return jsonSerializer.Deserialize<T>(jsonTextReader);
+                            }
+                        }
                     }
                 }
             }

diff --git a/Microsoft.Azure.Cosmos/src/Serializer/CosmosSystemTextJsonSerializer.cs b/Microsoft.Azure.Cosmos/src/Serializer/CosmosSystemTextJsonSerializer.cs
@@ -52,28 +52,25 @@ public override T FromStream<T>(Stream stream)
 
             using (stream)
             {
-                if (stream is Documents.CloneableStream cloneableStream)
+                using (CosmosBufferedStreamWrapper bufferedStream = new (stream, shouldDisposeInnerStream: false))
                 {
-                    using (CosmosBufferedStreamWrapper bufferedStream = new (cloneableStream, shouldDisposeInnerStream: false))
+                    if (bufferedStream.GetJsonSerializationFormat() == JsonSerializationFormat.Binary)
                     {
-                        if (bufferedStream.GetJsonSerializationFormat() == JsonSerializationFormat.Binary)
+                        byte[] content = bufferedStream.ReadAll();
+
+                        if (CosmosObject.TryCreateFromBuffer(content, out CosmosObject cosmosObject))
+                        {
+                            return System.Text.Json.JsonSerializer.Deserialize<T>(cosmosObject.ToString(), this.jsonSerializerOptions);
+                        }
+                        else
                         {
-                            byte[] content = bufferedStream.ReadAll();
-
-                            if (CosmosObject.TryCreateFromBuffer(content, out CosmosObject cosmosObject))
-                            {
-                                return System.Text.Json.JsonSerializer.Deserialize<T>(cosmosObject.ToString(), this.jsonSerializerOptions);
-                            }
-                            else
-                            {
-                                using Stream textStream = CosmosSerializationUtil.ConvertToStreamUsingJsonSerializationFormat(content, JsonSerializationFormat.Text);
-                                return this.DeserializeStream<T>(textStream);
-                            }
+                            using Stream textStream = CosmosSerializationUtil.ConvertToStreamUsingJsonSerializationFormat(content, JsonSerializationFormat.Text);
+                            return this.DeserializeStream<T>(textStream);
                         }
                     }
-                }
 
-                return this.DeserializeStream<T>(stream);
+                    return this.DeserializeStream<T>(bufferedStream);
+                }
             }
         }