Provide a way to override YesSql configuration (#11713)

src/OrchardCore/OrchardCore.Data.YesSql.Abstractions/IScopedIndexProvider.cs

@@ -1,10 +1,11 @@
+using YesSql;
 using YesSql.Indexes;
 
 namespace OrchardCore.Data
 {
     /// <summary>
     /// Represents a contract that used to denote an <see cref="IIndexProvider"/> that needs to be resolved by the DI and registered
-    /// at the <see cref="YesSql.ISession"/> level.
+    /// at the <see cref="ISession"/> level.
     /// </summary>
     public interface IScopedIndexProvider : IIndexProvider
     {

src/OrchardCore/OrchardCore.Data.YesSql.Abstractions/YesSqlOptions.cs

@@ -0,0 +1,20 @@
+using YesSql;
+
+namespace OrchardCore.Data.YesSql.Abstractions;
+
+public class YesSqlOptions
+{
+    public int CommandsPageSize { get; set; } = 500;
+
+    public bool QueryGatingEnabled { get; set; } = true;
+
+    public IIdGenerator IdGenerator { get; set; }
+
+    public ITableNameConvention TableNameConvention { get; set; }
+
+    public IAccessorFactory IdentifierAccessorFactory { get; set; }
+
+    public IAccessorFactory VersionAccessorFactory { get; set; }
+
+    public IContentSerializer ContentSerializer { get; set; }
+}

src/OrchardCore/OrchardCore.Data.YesSql/OrchardCoreBuilderExtensions.cs

@@ -8,6 +8,7 @@
 using OrchardCore.Data;
 using OrchardCore.Data.Documents;
 using OrchardCore.Data.Migration;
+using OrchardCore.Data.YesSql.Abstractions;
 using OrchardCore.Environment.Shell;
 using OrchardCore.Environment.Shell.Models;
 using OrchardCore.Environment.Shell.Scope;
@@ -47,21 +48,17 @@ public static OrchardCoreBuilder AddDataAccess(this OrchardCoreBuilder builder)
                 services.TryAddDataProvider(name: "Postgres", value: "Postgres", hasConnectionString: true, sampleConnectionString: "Server=localhost;Port=5432;Database=Orchard;User Id=username;Password=password", hasTablePrefix: true, isDefault: false);
 
                 // Configuring data access
-
                 services.AddSingleton(sp =>
                 {
                     var shellSettings = sp.GetService<ShellSettings>();
 
-                    // Before the setup a 'DatabaseProvider' may be configured without a required 'ConnectionString'.
+                    // Before the setup, a 'DatabaseProvider' may be configured without a required 'ConnectionString'.
                     if (shellSettings.State == TenantState.Uninitialized || shellSettings["DatabaseProvider"] == null)
                     {
                         return null;
                     }
 
-                    IConfiguration storeConfiguration = new YesSql.Configuration
-                    {
-                        ContentSerializer = new PoolingJsonContentSerializer(sp.GetService<ArrayPool<char>>()),
-                    };
+                    var storeConfiguration = GetStoreConfiguration(sp);
 
                     switch (shellSettings["DatabaseProvider"])
                     {
@@ -102,7 +99,7 @@ public static OrchardCoreBuilder AddDataAccess(this OrchardCoreBuilder builder)
                             throw new ArgumentException("Unknown database provider: " + shellSettings["DatabaseProvider"]);
                     }
 
-                    if (!string.IsNullOrWhiteSpace(shellSettings["TablePrefix"]))
+                    if (!String.IsNullOrWhiteSpace(shellSettings["TablePrefix"]))
                     {
                         storeConfiguration = storeConfiguration.SetTablePrefix(shellSettings["TablePrefix"] + "_");
                     }
@@ -161,5 +158,44 @@ public static OrchardCoreBuilder AddDataAccess(this OrchardCoreBuilder builder)
 
             return builder;
         }
+
+        private static IConfiguration GetStoreConfiguration(IServiceProvider sp)
+        {
+            var yesSqlOptions = sp.GetService<IOptions<YesSqlOptions>>().Value;
+
+            var storeConfiguration = new YesSql.Configuration
+            {
+                CommandsPageSize = yesSqlOptions.CommandsPageSize,
+                QueryGatingEnabled = yesSqlOptions.QueryGatingEnabled,
+                ContentSerializer = new PoolingJsonContentSerializer(sp.GetService<ArrayPool<char>>()),
+            };
+
+            if (yesSqlOptions.IdGenerator != null)
+            {
+                storeConfiguration.IdGenerator = yesSqlOptions.IdGenerator;
+            }
+
+            if (yesSqlOptions.TableNameConvention != null)
+            {
+                storeConfiguration.TableNameConvention = yesSqlOptions.TableNameConvention;
+            }
+
+            if (yesSqlOptions.IdentifierAccessorFactory != null)
+            {
+                storeConfiguration.IdentifierAccessorFactory = yesSqlOptions.IdentifierAccessorFactory;
+            }
+
+            if (yesSqlOptions.VersionAccessorFactory != null)
+            {
+                storeConfiguration.VersionAccessorFactory = yesSqlOptions.VersionAccessorFactory;
+            }
+
+            if (yesSqlOptions.ContentSerializer != null)
+            {
+                storeConfiguration.ContentSerializer = yesSqlOptions.ContentSerializer;
+            }
+
+            return storeConfiguration;
+        }
     }
 }