src/Persistence/MassTransit.MongoDbIntegration/Configuration/MongoDbSagaRepositoryRegistrationExtensions.cs - develop - vshmidt/masstransit

src/Persistence/MassTransit.MongoDbIntegration/Configuration/MongoDbSagaRepositoryRegistrationExtensions.cs
174 строки7 KB
Chris Patterson
Added Guid mappings for MongoDB job saga, job attempt saga, and job type saga.
29 июн 2025, 23:11
29 июн 2025, 23:11cc49a66
namespace MassTransit
{
    using System;
    using Configuration;
    using MongoDB.Driver;


    public static class MongoDbSagaRepositoryRegistrationExtensions
    {
        /// <summary>
        /// Adds a MongoDB saga repository to the registration
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="configure"></param>
        /// <typeparam name="TSaga"></typeparam>
        /// <returns></returns>
        public static ISagaRegistrationConfigurator<TSaga> MongoDbRepository<TSaga>(this ISagaRegistrationConfigurator<TSaga> configurator,
            Action<IMongoDbSagaRepositoryConfigurator<TSaga>> configure = null)
            where TSaga : class, ISagaVersion
        {
            var mongoDbConfigurator = new MongoDbSagaRepositoryConfigurator<TSaga>();

            configure?.Invoke(mongoDbConfigurator);

            mongoDbConfigurator.Validate().ThrowIfContainsFailure("The MongoDB saga repository configuration is invalid:");

            configurator.Repository(mongoDbConfigurator.Register);

            return configurator;
        }

        /// <summary>
        /// Adds a MongoDB saga repository to the registration
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="connectionString"></param>
        /// <param name="configure"></param>
        /// <typeparam name="TSaga"></typeparam>
        /// <returns></returns>
        public static ISagaRegistrationConfigurator<TSaga> MongoDbRepository<TSaga>(this ISagaRegistrationConfigurator<TSaga> configurator,
            string connectionString, Action<IMongoDbSagaRepositoryConfigurator<TSaga>> configure)
            where TSaga : class, ISagaVersion
        {
            return configurator.MongoDbRepository(cfg =>
            {
                cfg.Connection = connectionString;
                configure?.Invoke(cfg);
            });
        }

        /// <summary>
        /// Adds a MongoDB saga repository to the registration
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="database"></param>
        /// <param name="configure"></param>
        /// <typeparam name="TSaga"></typeparam>
        /// <returns></returns>
        public static ISagaRegistrationConfigurator<TSaga> MongoDbRepository<TSaga>(this ISagaRegistrationConfigurator<TSaga> configurator,
            IMongoDatabase database, Action<IMongoDbSagaRepositoryConfigurator<TSaga>> configure)
            where TSaga : class, ISagaVersion
        {
            return configurator.MongoDbRepository(cfg =>
            {
                cfg.Database(database);
                configure?.Invoke(cfg);
            });
        }

        /// <summary>
        /// Configure the Job Service saga state machines to use MongoDB
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="configure"></param>
        /// <returns></returns>
        public static IJobSagaRegistrationConfigurator MongoDbRepository(this IJobSagaRegistrationConfigurator configurator,
            Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            var registrationProvider = new MongoDbSagaRepositoryRegistrationProvider(configure);

            configurator.UseRepositoryRegistrationProvider(registrationProvider);

            return configurator;
        }

        /// <summary>
        /// Configure the Job Service saga state machines to use MongoDB
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="connectionString">The connection string for the MongoDB database</param>
        /// <param name="configure"></param>
        /// <returns></returns>
        public static IJobSagaRegistrationConfigurator MongoDbRepository(this IJobSagaRegistrationConfigurator configurator, string connectionString,
            Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            var registrationProvider = new MongoDbSagaRepositoryRegistrationProvider(r =>
            {
                r.Connection = connectionString;

                configure?.Invoke(r);
            });

            configurator.UseRepositoryRegistrationProvider(registrationProvider);

            return configurator;
        }

        /// <summary>
        /// Configure the Job Service saga state machines to use MongoDB
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="database">Ready to use MongoDB database</param>
        /// <param name="configure"></param>
        /// <returns></returns>
        public static IJobSagaRegistrationConfigurator MongoDbRepository(this IJobSagaRegistrationConfigurator configurator, IMongoDatabase database,
            Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            var registrationProvider = new MongoDbSagaRepositoryRegistrationProvider(r =>
            {
                r.Database(database);

                configure?.Invoke(r);
            });

            configurator.UseRepositoryRegistrationProvider(registrationProvider);

            return configurator;
        }

        /// <summary>
        /// Use the MongoDB saga repository for sagas configured by type (without a specific generic call to AddSaga/AddSagaStateMachine)
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="configure"></param>
        public static void SetMongoDbSagaRepositoryProvider(this IRegistrationConfigurator configurator, Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            configurator.SetSagaRepositoryProvider(new MongoDbSagaRepositoryRegistrationProvider(configure));
        }

        /// <summary>
        /// Use the MongoDB saga repository for sagas configured by type (without a specific generic call to AddSaga/AddSagaStateMachine)
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="connectionString">The connection string for the MongoDB database</param>
        /// <param name="configure"></param>
        public static void SetMongoDbSagaRepositoryProvider(this IRegistrationConfigurator configurator, string connectionString,
            Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            configurator.SetSagaRepositoryProvider(new MongoDbSagaRepositoryRegistrationProvider(r =>
            {
                r.Connection = connectionString;

                configure?.Invoke(r);
            }));
        }

        /// <summary>
        /// Use the MongoDB saga repository for sagas configured by type (without a specific generic call to AddSaga/AddSagaStateMachine)
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="database">Ready to use MongoDB database</param>
        /// <param name="configure"></param>
        public static void SetMongoDbSagaRepositoryProvider(this IRegistrationConfigurator configurator, IMongoDatabase database,
            Action<IMongoDbSagaRepositoryConfigurator> configure)
        {
            configurator.SetSagaRepositoryProvider(new MongoDbSagaRepositoryRegistrationProvider(r =>
            {
                r.Database(database);

                configure?.Invoke(r);
            }));
        }
    }
}