NServiceBus behavior to share persistence database connection

gistfile1.cs

using System;
using System.Data;
using System.Diagnostics;
using NServiceBus;
using NServiceBus.Persistence.NHibernate;
using NServiceBus.Pipeline;
using NServiceBus.Pipeline.Contexts;

class MyContextBehaviorRegistration : INeedInitialization
{
	public void Customize(BusConfiguration busConfiguration)
	{
		busConfiguration.RegisterComponents(r => r.ConfigureComponent<MyContext>(DependencyLifecycle.InstancePerCall));
		busConfiguration.RegisterComponents(r => r.ConfigureComponent<IDbConnection>(b => b.Build<NHibernateStorageContext>().Connection, DependencyLifecycle.InstancePerCall));
		// Register the new step in the pipeline
		busConfiguration.Pipeline.Register<MyContextBehavior.Registration>();
	}
}

class MyContextBehavior : IBehavior<IncomingContext>
{
	public void Invoke(IncomingContext context, Action next)
	{
		var ctx = context.Builder.Build<MyContext>();
		next();
		ctx.SaveChanges();
	}

	internal class Registration : RegisterStep
	{
		public Registration()
			: base(typeof(MyContextBehavior).Name, typeof(MyContextBehavior), "Database context behavior yo!")
		{
			InsertAfter(WellKnownStep.MutateIncomingMessages);
		}
	}
}

class MyContext
{
	public MyContext(IDbConnection connection)
	{
		Console.WriteLine("Constructor | IDbConnection");
		Debug.Assert(connection != null, "No shared connection injected.");
		// Pass it to the base class here or via the constructor
	}

	public void SaveChanges()
	{
		Console.WriteLine("SaveChanges");
	}
}