public override async Task InvokeAsync(IPipeContext context,
            CancellationToken token = new CancellationToken())
        {
            var retry = context.GetRetryInformation();
            if (context.GetMessageContext() is CorrelationContext message)
            {
                message.Retries = retry.NumberOfRetries;
            }

            await Next.InvokeAsync(context, token);
        }
        protected virtual void UpdateRetryHeaders(BasicDeliverEventArgs args, IPipeContext context)
        {
            var retryInfo = context.GetRetryInformation();

            _headerUpdater.AddOrUpdate(args, retryInfo);
        }