src/MassTransit/Initializers/PropertyProviders/AsyncPropertyProvider.cs - develop - vshmidt/masstransit

src/MassTransit/Initializers/PropertyProviders/AsyncPropertyProvider.cs

92 строки3 KB

Chris Patterson

Integration of all external packages into MassTransit, split out into Abstractions, Middleware, and the core MassTransit assembly.

22 янв 2022, 18:24

22 янв 2022, 18:24b5b4f50

namespace MassTransit.Initializers.PropertyProviders
{
    using System.Threading.Tasks;
    using Util;


    /// <summary>
    /// Awaits a <see cref="Task{TProperty}" /> property, returning the property value.
    /// </summary>
    /// <typeparam name="TInput"></typeparam>
    /// <typeparam name="TProperty"></typeparam>
    public class AsyncPropertyProvider<TInput, TProperty> :
        IPropertyProvider<TInput, TProperty>
        where TInput : class
    {
        readonly IPropertyProvider<TInput, Task<TProperty>> _provider;

        public AsyncPropertyProvider(IPropertyProvider<TInput, Task<TProperty>> provider)
        {
            _provider = provider;
        }

        Task<TProperty> IPropertyProvider<TInput, TProperty>.GetProperty<T>(InitializeContext<T, TInput> context)
        {
            if (!context.HasInput)
                return TaskUtil.Default<TProperty>();

            Task<Task<TProperty>> propertyTask = _provider.GetProperty(context);
            if (propertyTask.Status == TaskStatus.RanToCompletion)
            {
                Task<TProperty> valueTask = propertyTask.Result;
                if (valueTask.Status == TaskStatus.RanToCompletion)
                    return valueTask;
            }

            async Task<TProperty> GetPropertyAsync()
            {
                Task<TProperty> valueTask = await propertyTask.ConfigureAwait(false);

                return await valueTask.ConfigureAwait(false);
            }

            return GetPropertyAsync();
        }
    }


    /// <summary>
    /// Awaits a <see cref="Task{TProperty}" /> property, returning the property value.
    /// </summary>
    /// <typeparam name="TInput"></typeparam>
    /// <typeparam name="TProperty"></typeparam>
    /// <typeparam name="TTask"></typeparam>
    public class AsyncPropertyProvider<TInput, TProperty, TTask> :
        IPropertyProvider<TInput, TProperty>
        where TInput : class
    {
        readonly IPropertyConverter<TProperty, TTask> _converter;
        readonly IPropertyProvider<TInput, Task<TTask>> _provider;

        public AsyncPropertyProvider(IPropertyProvider<TInput, Task<TTask>> provider, IPropertyConverter<TProperty, TTask> converter)
        {
            _provider = provider;
            _converter = converter;
        }

        Task<TProperty> IPropertyProvider<TInput, TProperty>.GetProperty<T>(InitializeContext<T, TInput> context)
        {
            if (!context.HasInput)
                return TaskUtil.Default<TProperty>();

            Task<Task<TTask>> propertyTask = _provider.GetProperty(context);
            if (propertyTask.Status == TaskStatus.RanToCompletion)
            {
                Task<TTask> valueTask = propertyTask.Result;
                if (valueTask.Status == TaskStatus.RanToCompletion)
                    return _converter.Convert(context, valueTask.Result);
            }

            async Task<TProperty> GetPropertyAsync()
            {
                Task<TTask> valueTask = await propertyTask.ConfigureAwait(false);

                var value = await valueTask.ConfigureAwait(false);

                return await _converter.Convert(context, value).ConfigureAwait(false);
            }

            return GetPropertyAsync();
        }
    }
}