src/MassTransit.Abstractions/Transports/TransportSetHeaderAdapter.cs - develop - vshmidt/masstransit

src/MassTransit.Abstractions/Transports/TransportSetHeaderAdapter.cs

70 строк2 KB

Chris Patterson

Added fault header for the original input address of a faulted message when moved to the error queue.

09 сен 2022, 15:29

09 сен 2022, 15:297a94111

namespace MassTransit.Transports
{
    using System.Collections.Generic;


    public class TransportSetHeaderAdapter<TValueType> :
        ITransportSetHeaderAdapter<TValueType>
    {
        readonly IHeaderValueConverter<TValueType> _converter;
        readonly TransportHeaderOptions _options;

        public TransportSetHeaderAdapter(IHeaderValueConverter<TValueType> converter, TransportHeaderOptions options = TransportHeaderOptions.Default)
        {
            _converter = converter;
            _options = options;
        }

        public void Set(IDictionary<string, TValueType> dictionary, in HeaderValue headerValue)
        {
            switch (headerValue.Value)
            {
                case null:
                    if (dictionary.ContainsKey(headerValue.Key))
                        dictionary.Remove(headerValue.Key);
                    break;

                default:
                    if (IsHeaderIncluded(headerValue.Key) && _converter.TryConvert(headerValue, out HeaderValue<TValueType> result))
                        dictionary[result.Key] = result.Value;
                    break;
            }
        }

        public void Set<T>(IDictionary<string, TValueType> dictionary, in HeaderValue<T> headerValue)
        {
            switch (headerValue.Value)
            {
                case null:
                case string s when string.IsNullOrWhiteSpace(s):
                    if (dictionary.ContainsKey(headerValue.Key))
                        dictionary.Remove(headerValue.Key);
                    break;

                default:
                    if (IsHeaderIncluded(headerValue.Key) && _converter.TryConvert(headerValue, out HeaderValue<TValueType> result))
                        dictionary[result.Key] = result.Value;
                    break;
            }
        }

        bool IsHeaderIncluded(string key)
        {
            if (key.StartsWith("MT-Host-"))
                return _options.HasFlag(TransportHeaderOptions.IncludeHost);

            if (key.Equals(MessageHeaders.FaultInputAddress))
                return true;

            if (key.StartsWith("MT-Fault-"))
            {
                if (_options.HasFlag(TransportHeaderOptions.IncludeFaultDetail))
                    return true;

                return _options.HasFlag(TransportHeaderOptions.IncludeFaultMessage) && key.Equals(MessageHeaders.FaultMessage);
            }

            return true;
        }
    }
}