add incoming headers to envelope for ASB transport

JasperFx · Feb 17, 2024 · fce5b1b · fce5b1b
1 parent 46c39ef
commit fce5b1b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/Transports/Azure/Wolverine.AzureServiceBus/Internal/AzureServiceBusEnvelopeMapper.cs b/src/Transports/Azure/Wolverine.AzureServiceBus/Internal/AzureServiceBusEnvelopeMapper.cs
@@ -43,6 +43,13 @@ protected override void writeOutgoingHeader(ServiceBusMessage outgoing, string k
         outgoing.ApplicationProperties[key] = value;
     }
 
+    protected override void writeIncomingHeaders(ServiceBusReceivedMessage incoming, Envelope envelope)
+    {
+        if (incoming.ApplicationProperties == null) return;
+
+        foreach (var pair in incoming.ApplicationProperties) envelope.Headers[pair.Key] = pair.Value?.ToString();
+    }
+
     protected override bool tryReadIncomingHeader(ServiceBusReceivedMessage incoming, string key, out string? value)
     {
         if (incoming.ApplicationProperties.TryGetValue(key, out var header))