Unit of Work

Using a transaction scope

If a business transaction is spread across multiple handlers there is always a risk of partial updates since one handler might succeed in updating the data while other won't. To avoid this it is possible to use a unit of work that wraps all handlers in a TransactionScope and makes sure that there are no partial updates. Use following code to enable a wrapping scope:

var unitOfWork = endpointConfiguration.UnitOfWork();
unitOfWork.WrapHandlersInATransactionScope();

Controlling transaction scope options

The following options for transaction scopes used to wrap all handlers can be configured.

Isolation level

NServiceBus will by default use the ReadCommitted isolation level.

Change the isolation level using

var unitOfWork = endpointConfiguration.UnitOfWork();
unitOfWork.WrapHandlersInATransactionScope(
    isolationLevel: IsolationLevel.RepeatableRead);

Transaction timeout

NServiceBus will use the default transaction timeout of the machine the endpoint is running on.

Change the transaction timeout using

var unitOfWork = endpointConfiguration.UnitOfWork();
unitOfWork.WrapHandlersInATransactionScope(
    timeout: TimeSpan.FromSeconds(30));

Or via .config file using a example DefaultSettingsSection.

Implementing custom unit of work

A unit of work allows for shared code, that wraps handlers, to be reused in a way that doesn't pollute the individual handler code. For example, committing NHibernate transactions, or calling SaveChanges on a RavenDB session.

IManageUnitsOfWork

To create a unit of work, implement the IManageUnitsOfWork interface.

public class MyUnitOfWork :
    IManageUnitsOfWork
{
    public Task Begin()
    {
        // Do custom work here
        return Task.CompletedTask;
    }

    public Task End(Exception ex = null)
    {
        // Do custom work here
        return Task.CompletedTask;
    }
}

The semantics are that Begin() is called when the transport messages enters the pipeline. A transport message can consist of multiple application messages. This allows any setup that is required.

The End() method is called when the processing is complete. If there is an exception, it is passed into the method.

This gives a way to perform different actions depending on the outcome of the message(s).

partial: access-to-context

partial: imanageunitsofwork-outbox

partial: nulltask

Registering custom unit of work

After implementing an IManageUnitsOfWork, it needs to be registered:

endpointConfiguration.RegisterComponents(
    registration: components =>
    {
        components.ConfigureComponent<MyUnitOfWork>(DependencyLifecycle.InstancePerUnitOfWork);
    });