src/Transports/MassTransit.Azure.ServiceBus.Core/ServiceBusSendContextExtensions.cs - develop - vshmidt/masstransit

src/Transports/MassTransit.Azure.ServiceBus.Core/ServiceBusSendContextExtensions.cs

59 строк2 KB

Chris Patterson

Added ReplyTo as a settable property on the ServiceBusSendContext

03 май 2024, 02:39

03 май 2024, 02:3925d6373

namespace MassTransit
{
    using System;


    public static class ServiceBusSendContextExtensions
    {
        /// <summary>
        /// Set the time at which the message should be delivered to the queue
        /// </summary>
        /// <param name="context"></param>
        /// <param name="scheduledTime">The scheduled time for the message</param>
        public static void SetScheduledEnqueueTime(this SendContext context, DateTime scheduledTime)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
            {
                if (scheduledTime.Kind == DateTimeKind.Local)
                    scheduledTime = scheduledTime.ToUniversalTime();

                sendContext.ScheduledEnqueueTimeUtc = scheduledTime;
            }
        }

        /// <summary>
        /// Set the time at which the message should be delivered to the queue
        /// </summary>
        /// <param name="context"></param>
        /// <param name="delay">The time to wait before the message should be enqueued</param>
        public static void SetScheduledEnqueueTime(this SendContext context, TimeSpan delay)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
                sendContext.ScheduledEnqueueTimeUtc = DateTime.UtcNow + delay;
        }

        public static void SetSessionId(this SendContext context, string sessionId)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
                sendContext.SessionId = sessionId;
        }

        public static void SetReplyToSessionId(this SendContext context, string sessionId)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
                sendContext.ReplyToSessionId = sessionId;
        }

        public static void SetReplyTo(this SendContext context, string replyTo)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
                sendContext.ReplyTo = replyTo;
        }

        public static void SetLabel(this SendContext context, string label)
        {
            if (context.TryGetPayload(out ServiceBusSendContext sendContext))
                sendContext.Label = label;
        }
    }
}