MassTransit State Machine Configuration

Question

How can I configure MassTransit to use quorum queues instead of classic queues when using a state machine saga? In the following code if I uncomment the cfg.ReceiveEndpoint(x => x.SetQuorumQueue()) line, it stil creates classic queues.

Also where do I use UseScheduledRedelivery and UseMessageRetry methods to configure retry and redelivery?

    public static IServiceCollection RegisterMassTransit(
        this IServiceCollection services,
        string sagaStateConnectionString,
        SerilogLoggerFactory factory,
        TransportSettings transportSettings)
    {
        services.AddDbContext(r =>
        {
            r.UseSqlServer(sagaStateConnectionString, m =>
            {
                m.MigrationsAssembly(Assembly.GetExecutingAssembly().GetName().Name);
                m.MigrationsHistoryTable($"__{nameof(OrderSagaDbContext)}");
            });
        });


        services.AddMassTransit(c =>
        {
            c.SetKebabCaseEndpointNameFormatter();

            c.AddSagaStateMachine()
                .EntityFrameworkRepository(r =>
                {
                    r.ExistingDbContext();
                    r.UseSqlServer();
                });

            c.AddBus(context => ConfigureBus(context, transportSettings));

            foreach (var q in transportSettings.Queues)
            {
                c.AddConsumers(TypeNamesToTypes(q.Processors));
                
            }
        });

        return services;
    }

    private static IBusControl ConfigureBus(
        IBusRegistrationContext context,
        TransportSettings transportSettings)
    {
        return Bus.Factory.CreateUsingRabbitMq(cfg =>
        {
            cfg.Host(transportSettings.Host, transportSettings.VirtualHost, hst =>
            {
                hst.Username(transportSettings.UserName);
                hst.Password(transportSettings.Password);
            });
            
            //cfg.ReceiveEndpoint(x => x.SetQuorumQueue());
            cfg.ConfigureEndpoints(context);

        });
    }

user_1882 · Answer

To override the receive endpoint configuration of the saga endpoint, you can create a saga definition:

public class OrderStateSagaDefinition :
    SagaDefinition
{
    protected override void ConfigureSaga(IReceiveEndpointConfigurator endpointConfigurator, ISagaConfigurator sagaConfigurator)
    {
        if(endpointConfigurator is IRabbitMqReceiveEndpointConfigurator rmq)
            rmq.SetQuorumQueue();

        endpointConfigurator.UseScheduledRedelivery(...);
        endpointConfigurator.UseMessageRetry(...);
        endpointConfigurator.UseInMemoryOutbox();
    }
}

You can also configure the message retry/redelivery/outbox in the definition as shown.

Then, add that definition to the container configuration:

c.AddSagaStateMachine()

You'll need to configure a scheduler as well. If you're using RabbitMQ delayed exchange for redelivery, change UseScheduledRedelivery to UseDelayedRedelivery.

user_19333157 · Answer

With Chris's help above I was able to figure out how to set properties for all queues in an application that uses multiple queues and a state machine saga.

Add an extension method as follows:

public static class RabbitMqReceiveEndpointConfigurationExtensions
{
    public static void Configure(
        this RabbitMqReceiveEndpointConfiguration rmq)
    {
        rmq.PublishFaults = ...;
        rmq.PrefetchCount = ...;
        rmq.SetQuorumQueue();
        rmq.UseScheduledRedelivery(...);
        rmq.UseMessageRetry(...);
    }
}

Then we can call the extension method for each queue when configuring the container as follows:

services.AddMassTransit(c =>
{
    c.SetKebabCaseEndpointNameFormatter();
    c.AddConfigureEndpointsCallback((endpoint, endpointConfigurator) =>
    {
        if (endpointConfigurator is RabbitMqReceiveEndpointConfiguration rmq)
        {
            rmq.Configure();
        }
    });
    ...
});

MassTransit State Machine Configuration

2 Answers