Merge pull request #813 from glucaci/transaction

Add transaction support

Author: danielgerlag

src/providers/WorkflowCore.Persistence.MongoDB/ServiceCollectionExtensions.cs

@@ -1,6 +1,5 @@
 using MongoDB.Driver;
 using System;
-using System.Linq;
 using WorkflowCore.Interface;
 using WorkflowCore.Models;
 using WorkflowCore.Persistence.MongoDB.Services;
@@ -33,5 +32,26 @@ public static WorkflowOptions UseMongoDB(
             });
             return options;
         }
+
+        public static WorkflowOptions UseMongoDB(
+            this WorkflowOptions options, 
+            Func<IServiceProvider, IMongoDatabase> createDatabase)
+        {
+            if (options == null) throw new ArgumentNullException(nameof(options));
+            if (createDatabase == null) throw new ArgumentNullException(nameof(createDatabase));
+
+            options.UsePersistence(sp =>
+            {
+                var db = createDatabase(sp);
+                return new MongoPersistenceProvider(db);
+            });
+            options.Services.AddTransient<IWorkflowPurger>(sp =>
+            {
+                var db = createDatabase(sp);
+                return new WorkflowPurger(db);
+            });
+
+            return options;
+        }
     }
 }