tests/MassTransit.Tests/Middleware/RetryCommandContext.cs - develop - vshmidt/masstransit

tests/MassTransit.Tests/Middleware/RetryCommandContext.cs

82 строки2 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.Tests.Middleware
{
    using System;
    using System.Threading;
    using Contracts;


    public class RetryCommandContext :
        CommandContext,
        CommandRetryContext
    {
        readonly CommandContext _context;

        public RetryCommandContext(CommandContext context)
        {
            _context = context;
        }

        public DateTime Timestamp => _context.Timestamp;

        public CancellationToken CancellationToken => _context.CancellationToken;

        public bool HasPayloadType(Type payloadType)
        {
            return _context.HasPayloadType(payloadType);
        }

        public bool TryGetPayload<TPayload>(out TPayload payload)
            where TPayload : class
        {
            return _context.TryGetPayload(out payload);
        }

        public TPayload GetOrAddPayload<TPayload>(PayloadFactory<TPayload> payloadFactory)
            where TPayload : class
        {
            return _context.GetOrAddPayload(payloadFactory);
        }

        public T AddOrUpdatePayload<T>(PayloadFactory<T> addFactory, UpdatePayloadFactory<T> updateFactory)
            where T : class
        {
            return _context.AddOrUpdatePayload(addFactory, updateFactory);
        }

        public int RetryAttempt { get; set; }

        public RetryCommandContext CreateNext()
        {
            return new RetryCommandContext(_context);
        }

        public virtual TContext CreateNext<TContext>()
            where TContext : RetryCommandContext
        {
            throw new InvalidOperationException("This is only supported by a derived type");
        }
    }


    public class RetryCommandContext<T> :
        RetryCommandContext,
        CommandContext<T>
        where T : class
    {
        readonly CommandContext<T> _context;

        public RetryCommandContext(CommandContext<T> context)
            : base(context)
        {
            _context = context;
        }

        T CommandContext<T>.Command => _context.Command;

        public override TContext CreateNext<TContext>()
        {
            return new RetryCommandContext<T>(_context) as TContext
                ?? throw new ArgumentException($"The context type is not valid: {TypeCache<T>.ShortName}");
        }
    }
}