src/MassTransit/Initializers/Contexts/DynamicInitializeContext.cs - develop - vshmidt/masstransit

src/MassTransit/Initializers/Contexts/DynamicInitializeContext.cs

71 строка2 KB

Chris Patterson

Massive file cleanup

27 мар 2024, 02:43

27 мар 2024, 02:432ac69e8

namespace MassTransit.Initializers.Contexts
{
    using System;
    using Middleware;


    public class DynamicInitializeContext<TMessage> :
        ProxyPipeContext,
        InitializeContext<TMessage>
        where TMessage : class
    {
        public DynamicInitializeContext(InitializeContext context, TMessage message)
            : base(context)
        {
            Message = message;
            MessageType = message.GetType();

            Depth = context.Depth + 1;
            Parent = context;
        }

        public TMessage Message { get; }
        public Type MessageType { get; }

        public int Depth { get; }

        public InitializeContext Parent { get; }

        public InitializeContext<TMessage, T> CreateInputContext<T>(T input)
            where T : class
        {
            return new DynamicInitializeContext<TMessage, T>(this, Message, input);
        }

        public bool TryGetParent<T>(out InitializeContext<T> parentContext)
            where T : class
        {
            if (this is InitializeContext<T> parent || (Parent != null && Parent.TryGetParent(out parent)))
            {
                parentContext = parent;
                return true;
            }

            parentContext = default;
            return false;
        }

        public InitializeContext<T> CreateMessageContext<T>(T message)
            where T : class
        {
            return new DynamicInitializeContext<T>(this, message);
        }
    }


    public class DynamicInitializeContext<TMessage, TInput> :
        DynamicInitializeContext<TMessage>,
        InitializeContext<TMessage, TInput>
        where TMessage : class
        where TInput : class
    {
        public DynamicInitializeContext(InitializeContext context, TMessage message, TInput input)
            : base(context, message)
        {
            HasInput = (Input = input) != null;
        }

        public bool HasInput { get; }
        public TInput Input { get; }
    }
}