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

src/MassTransit.Abstractions/Transports/TransportSetHeaderAdapterExtensions.cs
141 строка6 KB
Chris Patterson
Added ExceptionHeaders to ExceptionReceiveContext so that additional transport headers can be specified (or existing headers replaced/removed) when moving messages to the error queue.
27 ноя 2024, 17:49
27 ноя 2024, 17:49f8484cd
namespace MassTransit.Transports
{
    using System;
    using System.Collections.Generic;


    public static class TransportSetHeaderAdapterExtensions
    {
        static readonly ITransportSetHeaderAdapter<object> _adapter = new DictionaryTransportSetHeaderAdapter(new StringHeaderValueConverter());

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            string? value)
        {
            if (!string.IsNullOrWhiteSpace(value))
                adapter.Set(dictionary, new HeaderValue<string>(key, value!));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            Uri? value)
        {
            if (value != null)
                adapter.Set(dictionary, new HeaderValue<string>(key, value.ToString()));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            Guid? value)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, ToString(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            Guid? value, Func<Guid, string> formatter)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, formatter(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            int? value)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, ToString(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            int? value, Func<int, string> formatter)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, formatter(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            TimeSpan? value)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, ToString(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            TimeSpan? value, Func<TimeSpan, string> formatter)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, formatter(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            DateTime? value)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, ToString(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary, string key,
            DateTime? value, Func<DateTime, string> formatter)
        {
            if (value.HasValue)
                adapter.Set(dictionary, new HeaderValue<string>(key, formatter(value.Value)));
        }

        public static void Set<TValueType>(this ITransportSetHeaderAdapter<TValueType> adapter, IDictionary<string, TValueType> dictionary,
            IEnumerable<HeaderValue> headerValues)
        {
            foreach (var header in headerValues)
                adapter.Set(dictionary, header);
        }

        public static void Set(this IDictionary<string, object> dictionary, IEnumerable<HeaderValue> headerValues)
        {
            foreach (var header in headerValues)
                _adapter.Set(dictionary, header);
        }

        public static bool TryGetInt(this IDictionary<string, string> dictionary, string key, out int value)
        {
            if (dictionary.TryGetValue(key, out var text))
                return int.TryParse(text, out value);

            value = default;
            return false;
        }

        public static void Set(this IDictionary<string, object> dictionary, params HeaderValue[] headerValues)
        {
            foreach (var header in headerValues)
                _adapter.Set(dictionary, header);
        }

        [Obsolete("Removed with change to exception headers")]
        public static void SetExceptionHeaders(this IDictionary<string, object> dictionary, ExceptionReceiveContext exceptionContext)
        {
            _adapter.SetExceptionHeaders(dictionary, exceptionContext);
        }

        [Obsolete("Removed with change to exception headers")]
        public static void SetHostHeaders(this IDictionary<string, object> dictionary)
        {
            _adapter.SetHostHeaders(dictionary);
        }

        static string ToString(TimeSpan timeSpan)
        {
            return timeSpan.TotalMilliseconds.ToString("F0");
        }

        static string ToString(Guid guid)
        {
            return guid.ToString();
        }

        static string ToString(int value)
        {
            return value.ToString();
        }

        static string ToString(DateTime dateTime)
        {
            return dateTime.ToString("O");
        }
    }
}