src/MassTransit/RoutingSlipEventExtensions.cs - develop - vshmidt/masstransit

src/MassTransit/RoutingSlipEventExtensions.cs
207 строк9 KB
Chris Patterson
Fixed #3868 - default value routing slip arguments were not overridden by variables because default/null values should not be serialized.
22 ноя 2022, 18:47
22 ноя 2022, 18:4757f37a0
#nullable enable
namespace MassTransit
{
    using System.Collections.Generic;
    using Courier.Contracts;
    using Internals;


    public static class RoutingSlipEventExtensions
    {
        public static T? GetVariable<T>(this ConsumeContext<RoutingSlip> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlip> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompensated> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompensated> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompensated> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompensated> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompensationFailed> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompensationFailed> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompensationFailed> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompensationFailed> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetArgument<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : class
        {
            return GetDictionaryValue(context.SerializerContext, context.Message.Arguments, context.Message.Variables, key, defaultValue);
        }

        public static T? GetArgument<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return GetDictionaryValue(context.SerializerContext, context.Message.Arguments, context.Message.Variables, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetResult<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Data, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityCompleted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetArgument<T>(this ConsumeContext<RoutingSlipActivityFaulted> context, string key, T? defaultValue = null)
            where T : class
        {
            return GetDictionaryValue(context.SerializerContext, context.Message.Arguments, context.Message.Variables, key, defaultValue);
        }

        public static T? GetArgument<T>(this ConsumeContext<RoutingSlipActivityFaulted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return GetDictionaryValue(context.SerializerContext, context.Message.Arguments, context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityFaulted> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipActivityFaulted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipCompensationFailed> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipCompensationFailed> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipCompleted> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipCompleted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipFaulted> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipFaulted> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipTerminated> context, string key, T? defaultValue = null)
            where T : class
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        public static T? GetVariable<T>(this ConsumeContext<RoutingSlipTerminated> context, string key, T? defaultValue = null)
            where T : struct
        {
            return context.SerializerContext.GetValue(context.Message.Variables, key, defaultValue);
        }

        static T? GetDictionaryValue<T>(IObjectDeserializer context, IDictionary<string, object>? arguments, IDictionary<string, object>? variables,
            string key, T? defaultValue = null)
            where T : class
        {
            IDictionary<string, object>? argumentsDictionary = variables?.Count > 0
                ? arguments?.Count > 0 ? variables.MergeLeft(arguments) : variables
                : arguments;

            if (argumentsDictionary == null)
                return defaultValue;

            return context.GetValue(argumentsDictionary, key, defaultValue);
        }

        static T? GetDictionaryValue<T>(IObjectDeserializer context, IDictionary<string, object>? arguments, IDictionary<string, object>? variables,
            string key, T? defaultValue = null)
            where T : struct
        {
            IDictionary<string, object>? argumentsDictionary = variables?.Count > 0
                ? arguments?.Count > 0 ? variables.MergeLeft(arguments) : variables
                : arguments;

            if (argumentsDictionary == null)
                return defaultValue;

            return context.GetValue(argumentsDictionary, key, defaultValue);
        }
    }
}