src/MassTransit/Serialization/DictionarySendHeaders.cs - develop - vshmidt/masstransit

src/MassTransit/Serialization/DictionarySendHeaders.cs

109 строк3 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

#nullable enable
namespace MassTransit.Serialization
{
    using System;
    using System.Collections;
    using System.Collections.Generic;
    using System.Diagnostics.CodeAnalysis;
    using System.Linq;


    public class DictionarySendHeaders :
        SendHeaders
    {
        readonly IDictionary<string, object> _headers;

        public DictionarySendHeaders()
        {
            _headers = new Dictionary<string, object>(StringComparer.OrdinalIgnoreCase);
        }

        public DictionarySendHeaders(IDictionary<string, object?>? headers)
            : this()
        {
            if (headers == null)
                return;

            foreach (KeyValuePair<string, object?> header in headers)
            {
                if (header.Value != null)
                    _headers.Add(header.Key, header.Value);
            }
        }

        public DictionarySendHeaders(IDictionary<string, object> headers, bool useExistingDictionary)
        {
            if (headers == null)
                throw new ArgumentNullException(nameof(headers));

            if (useExistingDictionary)
                _headers = headers;
            else
            {
                _headers = new Dictionary<string, object>(headers, StringComparer.OrdinalIgnoreCase);

                foreach (KeyValuePair<string, object> header in headers)
                    _headers.Add(header.Key, header.Value);
            }
        }

        public void Set(string key, string? value)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));

            if (value == null)
                _headers.Remove(key);
            else
                _headers[key] = value;
        }

        public void Set(string key, object? value, bool overwrite = true)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));

            if (overwrite)
            {
                if (value == null)
                    _headers.Remove(key);
                else
                    _headers[key] = value;
            }
            else if (!_headers.ContainsKey(key) && value != null)
                _headers.Add(key, value);
        }

        public bool TryGetHeader(string key, [NotNullWhen(true)] out object? value)
        {
            return _headers.TryGetValue(key, out value);
        }

        public IEnumerable<KeyValuePair<string, object>> GetAll()
        {
            return _headers;
        }

        public T? Get<T>(string key, T? defaultValue)
            where T : class
        {
            return SystemTextJsonMessageSerializer.Instance.GetValue((IReadOnlyDictionary<string, object>)_headers, key, defaultValue);
        }

        public T? Get<T>(string key, T? defaultValue)
            where T : struct
        {
            return SystemTextJsonMessageSerializer.Instance.GetValue((IReadOnlyDictionary<string, object>)_headers, key, defaultValue);
        }

        public IEnumerator<HeaderValue> GetEnumerator()
        {
            return _headers.Select(x => new HeaderValue(x.Key, x.Value)).GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }
    }
}