Switch from BinaryFormatter to BsonFormatter for interpreter messages

This commit is contained in:
Yoshi Askharoun
2023-05-25 02:20:01 -05:00
parent 43868e4ae5
commit 81beb67fb2
5 changed files with 92 additions and 57 deletions
+87
View File
@@ -0,0 +1,87 @@
using Newtonsoft.Json;
using Newtonsoft.Json.Bson;
using System;
using System.IO;
using System.Runtime.Serialization;
namespace Microsoft.Iris.Debug;
internal class BsonFormatter : IFormatter
{
private readonly JsonSerializer _serializer;
public ISurrogateSelector SurrogateSelector { get; set; }
public SerializationBinder Binder { get; set; }
public StreamingContext Context { get; set; }
public BsonFormatter(JsonSerializerSettings settings = null) : this(new StreamingContext(StreamingContextStates.All), settings)
{
}
public BsonFormatter(StreamingContext context, JsonSerializerSettings settings = null)
{
Context = context;
settings ??= new()
{
ReferenceLoopHandling = ReferenceLoopHandling.Ignore,
Converters = new JsonConverter[]
{
new BsonFallbackConverter()
}
};
_serializer = JsonSerializer.Create(settings);
}
public object Deserialize(Stream serializationStream)
{
using BsonDataReader reader = new(serializationStream)
{
CloseInput = false
};
return _serializer.Deserialize(reader);
}
public void Serialize(Stream serializationStream, object graph)
{
using BsonDataWriter writer = new(serializationStream);
_serializer.Serialize(writer, graph);
writer.Flush();
}
}
internal class BsonFallbackConverter : JsonConverter
{
public override bool CanConvert(Type objectType)
{
return !objectType.Attributes.HasFlag(System.Reflection.TypeAttributes.Serializable)
|| objectType == typeof(uint) || objectType == typeof(ulong);
}
public override object ReadJson(JsonReader reader, Type objectType, object existingValue, JsonSerializer serializer)
{
if (objectType == typeof(uint))
return Convert.ToUInt32(existingValue);
else if (objectType == typeof(ulong))
return Convert.ToUInt64(existingValue);
else
return existingValue?.ToString();
}
public override void WriteJson(JsonWriter writer, object value, JsonSerializer serializer)
{
var objectType = value?.GetType();
object serializedValue;
if (objectType == typeof(uint))
serializedValue = BitConverter.GetBytes((uint)value);
else if (objectType == typeof(ulong))
serializedValue = BitConverter.GetBytes((ulong)value);
else
serializedValue = value?.ToString();
serializer.Serialize(writer, serializedValue);
}
}