NServiceBus - Add message context identifiers to exception;

AddMessageContextIdentifierstoExceptionBehavior.cs

//
// var pipeline = endpointConfiguration.Pipeline;
// pipeline.Register(behavior: new AddMessageContextIdentifierstoExceptionBehavior(), description: "Assigns the incoming message context identifiers to the exception for diagnostical purposes.");
//
class AddMessageContextIdentifierstoExceptionBehavior : Behavior<ITransportReceiveContext>
{
    public override async Task Invoke(ITransportReceiveContext context, Func<Task> next)
    {
        try
        {
            await next().ConfigureAwait(false);
        }
        catch(Exception ex)
        {
            ex.Data[Headers.MessageId] = context.Message.MessageId;
            var headers = context.Message.Headers;
            if(headers.TryGet(Headers.ConversationId, out string conversationId)) ex.Data[Headers.ConversationId] = conversationId;
            if(headers.TryGet(Headers.CorrelationId, out string correlationId)) ex.Data[Headers.CorrelationId] = correlationId;
            throw;
        }
    }
}