src/MassTransit/Internals/Reflection/ReadPropertyCache.cs - develop - vshmidt/masstransit

src/MassTransit/Internals/Reflection/ReadPropertyCache.cs
95 строк3 KB
Chris Patterson
Fixed #4697 - CorrelationId property from base classes and interfaces will be used if present.
11 окт 2023, 00:22
11 окт 2023, 00:22349c5e1
#nullable enable
namespace MassTransit.Internals
{
    using System;
    using System.Collections.Generic;
    using System.Diagnostics.CodeAnalysis;
    using System.Linq;
    using System.Reflection;


    public class ReadPropertyCache<T> :
        IReadPropertyCache<T>
        where T : class
    {
        readonly IDictionary<string, IReadProperty<T>> _properties;
        readonly IDictionary<string, PropertyInfo> _propertyIndex;

        ReadPropertyCache()
        {
            _properties = new Dictionary<string, IReadProperty<T>>(StringComparer.OrdinalIgnoreCase);
            _propertyIndex = MessageTypeCache<T>.Properties.ToDictionary(x => x.Name, StringComparer.OrdinalIgnoreCase);
        }

        IReadProperty<T, TProperty> IReadPropertyCache<T>.GetProperty<TProperty>(string? name)
        {
            return GetReadProperty<TProperty>(name ?? throw new ArgumentNullException(nameof(name)))
                ?? throw new ArgumentException($"{TypeCache<T>.ShortName} does not contain the property: {name}", nameof(name));
        }

        IReadProperty<T, TProperty> IReadPropertyCache<T>.GetProperty<TProperty>(PropertyInfo? propertyInfo)
        {
            var name = propertyInfo?.Name ?? throw new ArgumentNullException(nameof(propertyInfo));

            return GetReadProperty<TProperty>(name)
                ?? throw new ArgumentException($"{TypeCache<T>.ShortName} does not contain the property: {name}", nameof(name));
        }

        bool IReadPropertyCache<T>.TryGetProperty<TProperty>(string name, [NotNullWhen(true)] out IReadProperty<T, TProperty>? property)
        {
            IReadProperty<T, TProperty>? readProperty = GetReadProperty<TProperty>(name ?? throw new ArgumentNullException(nameof(name)));
            if (readProperty != null)
            {
                property = readProperty;
                return true;
            }

            property = null;
            return false;
        }

        IReadProperty<T, TProperty>? GetReadProperty<TProperty>(string name)
        {
            lock (_properties)
            {
                if (_properties.TryGetValue(name, out IReadProperty<T>? property))
                    return property as IReadProperty<T, TProperty>;

                if (_propertyIndex.TryGetValue(name, out var propertyInfo))
                {
                    if (propertyInfo.PropertyType != typeof(TProperty))
                        return null;

                    var readProperty = new ReadProperty<T, TProperty>(propertyInfo);

                    _properties[name] = readProperty;

                    return readProperty;
                }
            }

            return null;
        }

        public static IReadProperty<T, TProperty> GetProperty<TProperty>(string name)
        {
            return Cached.PropertyCache.Value.GetProperty<TProperty>(name);
        }

        public static bool TryGetProperty<TProperty>(string name, [NotNullWhen(true)] out IReadProperty<T, TProperty>? property)
        {
            return Cached.PropertyCache.Value.TryGetProperty(name, out property);
        }

        public static IReadProperty<T, TProperty> GetProperty<TProperty>(PropertyInfo propertyInfo)
        {
            return Cached.PropertyCache.Value.GetProperty<TProperty>(propertyInfo);
        }


        static class Cached
        {
            internal static readonly Lazy<IReadPropertyCache<T>> PropertyCache = new Lazy<IReadPropertyCache<T>>(() => new ReadPropertyCache<T>());
        }
    }
}