Patching v3 to support v4+ unwrapped messages.

Particular · Jun 19, 2014 · 0ea5c06 · 0ea5c06
1 parent f606444
commit 0ea5c06
Showing 1 changed file with 25 additions and 4 deletions.
diff --git a/src/impl/Serializers/NServiceBus.Serializers.Json/JsonMessageSerializerBase.cs b/src/impl/Serializers/NServiceBus.Serializers.Json/JsonMessageSerializerBase.cs
@@ -1,16 +1,20 @@
-using System.Globalization;
+using System;
+using System.Globalization;
 using System.IO;
 using System.Runtime.Serialization.Formatters;
+using Newtonsoft.Json;
+using Newtonsoft.Json.Converters;
 using NServiceBus.MessageInterfaces;
 using NServiceBus.Serialization;
 using NServiceBus.Serializers.Json.Internal;
-using Newtonsoft.Json;
-using Newtonsoft.Json.Converters;
 
 namespace NServiceBus.Serializers.Json
 {
     public abstract class JsonMessageSerializerBase : IMessageSerializer
     {
+        // From v4+ we just look for it, but don't expect it
+        const string EnclosedMessageTypes = "NServiceBus.EnclosedMessageTypes";
+
         private readonly IMessageMapper _messageMapper;
 
         protected JsonMessageSerializerBase(IMessageMapper messageMapper)
@@ -26,7 +30,7 @@ public JsonSerializerSettings JsonSerializerSettings
                                              {
                                                  TypeNameAssemblyFormat = FormatterAssemblyStyle.Simple,
                                                  TypeNameHandling = TypeNameHandling.Auto,
-                                                 Converters = { new IsoDateTimeConverter { DateTimeStyles = DateTimeStyles.RoundtripKind} }
+                                                 Converters = { new IsoDateTimeConverter { DateTimeStyles = DateTimeStyles.RoundtripKind } }
                                              };
                 return serializerSettings;
             }
@@ -49,6 +53,23 @@ public object[] Deserialize(Stream stream)
             jsonSerializer.ContractResolver = new MessageContractResolver(_messageMapper);
 
             JsonReader reader = CreateJsonReader(stream);
+            reader.Read();
+
+            JsonToken firstTokenType = reader.TokenType;
+
+            if (firstTokenType != JsonToken.StartArray)
+            {
+                string messageTypes;
+
+                var bus = Configure.Instance.Builder.Build<IBus>();
+                if (bus != null &&
+                    bus.CurrentMessageContext != null &&
+                    bus.CurrentMessageContext.Headers != null &&
+                    bus.CurrentMessageContext.Headers.TryGetValue(EnclosedMessageTypes, out messageTypes))
+                {
+                    return new[] { jsonSerializer.Deserialize(reader, Type.GetType(messageTypes.Split(';')[0])) };
+                }
+            }
 
             return jsonSerializer.Deserialize<object[]>(reader);
         }