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

src/MassTransit.Abstractions/Transports/DictionaryTransportSetHeaderAdapter.cs

78 строк3 KB

Chris Patterson

Fixed Azure Service Bus exception stack trace exceeding header length issue

19 фев 2025, 22:23

19 фев 2025, 22:236ae1b2a

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


    public class DictionaryTransportSetHeaderAdapter :
        ITransportSetHeaderAdapter<object>
    {
        readonly IHeaderValueConverter _converter;
        readonly TransportHeaderOptions _options;

        public DictionaryTransportSetHeaderAdapter(IHeaderValueConverter converter, TransportHeaderOptions options = TransportHeaderOptions.Default)
        {
            _converter = converter;
            _options = options;
        }

        public int? MaxHeaderLength { get; set; }

        public void Set(IDictionary<string, object> 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 var result))
                        dictionary[result.Key] = TrimHeaderIfLengthExceedsLimit(result.Value);

                    break;
            }
        }

        public void Set<T>(IDictionary<string, object> 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 var result))
                        dictionary[result.Key] = TrimHeaderIfLengthExceedsLimit(result.Value);
                    break;
            }
        }

        object TrimHeaderIfLengthExceedsLimit(object value)
        {
            if (MaxHeaderLength.HasValue && value is string stringValue && stringValue.Length > MaxHeaderLength.Value)
                value = stringValue.Substring(0, MaxHeaderLength.Value);

            return value;
        }

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

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

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

            return true;
        }
    }
}