Added Session Factories for Identity and DirtyTracked sessions

docs/configuration/hostbuilder.md

@@ -246,7 +246,7 @@ public interface IConfigureMarten
     void Configure(IServiceProvider services, StoreOptions options);
 }
 ```
-<sup><a href='https://github.com/JasperFx/marten/blob/master/src/Marten/MartenServiceCollectionExtensions.cs#L719-L730' title='Snippet source file'>snippet source</a> | <a href='#snippet-sample_iconfiguremarten' title='Start of snippet'>anchor</a></sup>
+<sup><a href='https://github.com/JasperFx/marten/blob/master/src/Marten/MartenServiceCollectionExtensions.cs#L728-L739' title='Snippet source file'>snippet source</a> | <a href='#snippet-sample_iconfiguremarten' title='Start of snippet'>anchor</a></sup>
 <!-- endSnippet -->
 
 You could alternatively implement a custom `IConfigureMarten` class like so:
@@ -580,7 +580,7 @@ public class Startup
             // See https://martendb.io/configuration/optimized_artifact_workflow.html
             .OptimizeArtifactWorkflow()
             // Spin up the DocumentStore right this second!
-            .InitializeStore();
+            .InitializeWith();
     }
 
     // And other methods we don't care about here...

docs/schema/migrations.md

@@ -154,7 +154,7 @@ services.AddMarten(opts =>
     // database changes on application startup
     .ApplyAllDatabaseChangesOnStartup();
 ```
-<sup><a href='https://github.com/JasperFx/marten/blob/master/src/CoreTests/MartenServiceCollectionExtensionsTests.cs#L164-L177' title='Snippet source file'>snippet source</a> | <a href='#snippet-sample_using_applyalldatabasechangesonstartup' title='Start of snippet'>anchor</a></sup>
+<sup><a href='https://github.com/JasperFx/marten/blob/master/src/CoreTests/MartenServiceCollectionExtensionsTests.cs#L165-L178' title='Snippet source file'>snippet source</a> | <a href='#snippet-sample_using_applyalldatabasechangesonstartup' title='Start of snippet'>anchor</a></sup>
 <!-- endSnippet -->
 
 In the option above, Marten is calling the same functionality within an `IHostedService` background task.

src/AspNetCoreWithMarten/Samples/EagerInitialization/Startup.cs

@@ -26,9 +26,9 @@ public void ConfigureServices(IServiceCollection services)
             // See https://martendb.io/configuration/optimized_artifact_workflow.html
             .OptimizeArtifactWorkflow()
             // Spin up the DocumentStore right this second!
-            .InitializeStore();
+            .InitializeWith();
     }
 
     // And other methods we don't care about here...
 }
-#endregion
+#endregion

src/CoreTests/MartenServiceCollectionExtensionsTests.cs

@@ -10,6 +10,7 @@
 using Marten.Internal;
 using Marten.Internal.Sessions;
 using Marten.Services;
+using Marten.Sessions;
 using Marten.Testing.Documents;
 using Marten.Testing.Harness;
 using Microsoft.Extensions.DependencyInjection;

src/Marten/MartenServiceCollectionExtensions.cs

@@ -12,6 +12,7 @@
 using Marten.Internal;
 using Marten.Schema;
 using Marten.Services;
+using Marten.Sessions;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging;
@@ -595,11 +596,24 @@ public MartenConfigurationExpression AddAsyncDaemon(DaemonMode mode)
         ///     IDocumentStore.LightweightSession();
         /// </summary>
         /// <returns></returns>
-        public MartenConfigurationExpression UseLightweightSessions()
-        {
+        public MartenConfigurationExpression UseLightweightSessions() =>
             BuildSessionsWith<LightweightSessionFactory>();
-            return this;
-        }
+
+        /// <summary>
+        ///     Use identity sessions by default for the injected IDocumentSession objects. Equivalent to
+        ///     IDocumentStore.IdentitySession();
+        /// </summary>
+        /// <returns></returns>
+        public MartenConfigurationExpression UseIdentitySession() =>
+            BuildSessionsWith<IdentitySessionFactory>();
+
+        /// <summary>
+        ///     Use dirty-tracked sessions by default for the injected IDocumentSession objects. Equivalent to
+        ///     IDocumentStore.DirtyTrackedSession();
+        /// </summary>
+        /// <returns></returns>
+        public MartenConfigurationExpression UseDirtyTrackedSession() =>
+            BuildSessionsWith<DirtyTrackedSessionFactory>();
 
         /// <summary>
         ///     Eagerly build the application's DocumentStore during application
@@ -627,10 +641,8 @@ public IDocumentStore InitializeStore()
         ///     Adds the optimized artifact workflow to this store. TODO -- LINK TO DOCS
         /// </summary>
         /// <returns></returns>
-        public MartenConfigurationExpression OptimizeArtifactWorkflow()
-        {
-            return OptimizeArtifactWorkflow(TypeLoadMode.Auto);
-        }
+        public MartenConfigurationExpression OptimizeArtifactWorkflow() =>
+            OptimizeArtifactWorkflow(TypeLoadMode.Auto);
 
         /// <summary>
         ///     Adds the optimized artifact workflow to this store with ability to override the TypeLoadMode in "Production" mode.
@@ -651,10 +663,8 @@ public MartenConfigurationExpression OptimizeArtifactWorkflow(TypeLoadMode typeL
         /// </summary>
         /// <param name="developmentEnvironment"></param>
         /// <returns></returns>
-        public MartenConfigurationExpression OptimizeArtifactWorkflow(string developmentEnvironment)
-        {
-            return OptimizeArtifactWorkflow(TypeLoadMode.Auto, developmentEnvironment);
-        }
+        public MartenConfigurationExpression OptimizeArtifactWorkflow(string developmentEnvironment) =>
+            OptimizeArtifactWorkflow(TypeLoadMode.Auto, developmentEnvironment);
 
         /// <summary>
         ///     Adds the optimized artifact workflow to this store with ability to override the TypeLoadMode in "Production" mode.
@@ -672,7 +682,6 @@ public MartenConfigurationExpression OptimizeArtifactWorkflow(TypeLoadMode typeL
             return this;
         }
 
-
         /// <summary>
         ///     Adds initial data sets to the Marten store and ensures that they will be
         ///     executed upon IHost initialization

src/Marten/Sessions/DirtyTrackedSessionFactory.cs

@@ -0,0 +1,15 @@
+namespace Marten.Sessions;
+
+internal class DirtyTrackedSessionFactory: ISessionFactory
+{
+    private readonly IDocumentStore _store;
+
+    public DirtyTrackedSessionFactory(IDocumentStore store) =>
+        _store = store;
+
+    public IQuerySession QuerySession() =>
+        _store.QuerySession();
+
+    public IDocumentSession OpenSession() =>
+        _store.DirtyTrackedSession();
+}

src/Marten/Sessions/IdentitySessionFactory.cs

@@ -0,0 +1,15 @@
+namespace Marten.Sessions;
+
+internal class IdentitySessionFactory: ISessionFactory
+{
+    private readonly IDocumentStore _store;
+
+    public IdentitySessionFactory(IDocumentStore store) =>
+        _store = store;
+
+    public IQuerySession QuerySession() =>
+        _store.QuerySession();
+
+    public IDocumentSession OpenSession() =>
+        _store.LightweightSession();
+}

src/Marten/Sessions/LightweightSessionFactory.cs

@@ -0,0 +1,15 @@
+namespace Marten.Sessions;
+
+internal class LightweightSessionFactory: ISessionFactory
+{
+    private readonly IDocumentStore _store;
+
+    public LightweightSessionFactory(IDocumentStore store) =>
+        _store = store;
+
+    public IQuerySession QuerySession() =>
+        _store.QuerySession();
+
+    public IDocumentSession OpenSession() =>
+        _store.LightweightSession();
+}