添加efcore6 分片脚本

src/ShardingCore/EFCores/EFCore6x/ScriptMigrationGenerator.cs

@@ -0,0 +1,131 @@
+#if EFCORE6
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.EntityFrameworkCore.Migrations;
+using ShardingCore.Core.DbContextCreator;
+using ShardingCore.Core.RuntimeContexts;
+using ShardingCore.Exceptions;
+using ShardingCore.Extensions;
+using ShardingCore.Helpers;
+using ShardingCore.Sharding.ShardingDbContextExecutors;
+
+namespace ShardingCore.EFCores
+{
+    public sealed class ScriptMigrationGenerator
+    {
+        private readonly IShardingRuntimeContext _shardingRuntimeContext;
+        private readonly string _fromMigration;
+        private readonly string _toMigration;
+        private readonly MigrationsSqlGenerationOptions _options;
+
+        public ScriptMigrationGenerator(IShardingRuntimeContext shardingRuntimeContext, string fromMigration = null,
+            string toMigration = null,
+            MigrationsSqlGenerationOptions options = MigrationsSqlGenerationOptions.Default)
+        {
+            _shardingRuntimeContext = shardingRuntimeContext;
+            _fromMigration = fromMigration;
+            _toMigration = toMigration;
+            _options = options;
+        }
+
+        public string GenerateScript()
+        {
+            var virtualDataSource = _shardingRuntimeContext.GetVirtualDataSource();
+            var allDataSourceNames = virtualDataSource.GetAllDataSourceNames();
+            var dbContextCreator = _shardingRuntimeContext.GetDbContextCreator();
+            var shardingProvider = _shardingRuntimeContext.GetShardingProvider();
+            var shardingConfigOptions = _shardingRuntimeContext.GetShardingConfigOptions();
+            var defaultDataSourceName = virtualDataSource.DefaultDataSourceName;
+
+            using (var scope = shardingProvider.CreateScope())
+            {
+                using (var shellDbContext = dbContextCreator.GetShellDbContext(scope.ServiceProvider))
+                {
+                    var parallelCount = shardingConfigOptions.MigrationParallelCount;
+                    if (parallelCount <= 0)
+                    {
+                        throw new ShardingCoreInvalidOperationException($"migration parallel count must >0");
+                    }
+
+                    //默认数据源需要最后执行 否则可能会导致异常的情况下GetPendingMigrations为空
+                    var partitionMigrationUnits = allDataSourceNames.Where(o => o != defaultDataSourceName)
+                        .Partition(parallelCount);
+                    var scriptStringBuilder = new StringBuilder();
+                    foreach (var migrationUnits in partitionMigrationUnits)
+                    {
+                        var migrateUnits = migrationUnits.Select(o => new MigrateUnit(shellDbContext, o)).ToList();
+                        var scriptSql = ExecuteMigrateUnits(_shardingRuntimeContext, migrateUnits);
+                        scriptStringBuilder.AppendLine(scriptSql);
+                    }
+
+                    //包含默认默认的单独最后一次处理
+                    if (allDataSourceNames.Contains(defaultDataSourceName))
+                    {
+                        var scriptSql = ExecuteMigrateUnits(_shardingRuntimeContext,
+                            new List<MigrateUnit>() { new MigrateUnit(shellDbContext, defaultDataSourceName) });
+                        scriptStringBuilder.AppendLine(scriptSql);
+                    }
+
+                    return scriptStringBuilder.ToString();
+                }
+            }
+        }
+
+        private string ExecuteMigrateUnits(IShardingRuntimeContext shardingRuntimeContext,
+            List<MigrateUnit> migrateUnits)
+        {
+            var shardingMigrationManager = shardingRuntimeContext.GetShardingMigrationManager();
+            var dbContextCreator = shardingRuntimeContext.GetDbContextCreator();
+            var routeTailFactory = shardingRuntimeContext.GetRouteTailFactory();
+            var migrateTasks = migrateUnits.Select(migrateUnit =>
+            {
+                return Task.Run(() =>
+                {
+                    using (shardingMigrationManager.CreateScope())
+                    {
+                        shardingMigrationManager.Current.CurrentDataSourceName = migrateUnit.DataSourceName;
+
+                        var dbContextOptions = CreateDbContextOptions(shardingRuntimeContext,
+                            migrateUnit.ShellDbContext.GetType(),
+                            migrateUnit.DataSourceName);
+
+                        using (var dbContext = dbContextCreator.CreateDbContext(migrateUnit.ShellDbContext,
+                                   new ShardingDbContextOptions(dbContextOptions,
+                                       routeTailFactory.Create(string.Empty, false))))
+                        {
+                            var migrator = dbContext.GetService<IMigrator>();
+                            return $"-- DataSource:{migrateUnit.DataSourceName}" + Environment.NewLine +
+                                   migrator.GenerateScript(_fromMigration, _toMigration, _options) +
+                                   Environment.NewLine;
+                        }
+                    }
+                });
+            }).ToArray();
+            var scripts = TaskHelper.WhenAllFastFail(migrateTasks).WaitAndUnwrapException();
+            return string.Join(Environment.NewLine, scripts);
+        }
+
+        private DbContextOptions CreateDbContextOptions(IShardingRuntimeContext shardingRuntimeContext,
+            Type dbContextType, string dataSourceName)
+        {
+            var virtualDataSource = shardingRuntimeContext.GetVirtualDataSource();
+            var shardingConfigOptions = shardingRuntimeContext.GetShardingConfigOptions();
+            var dbContextOptionBuilder = DataSourceDbContext.CreateDbContextOptionBuilder(dbContextType);
+            var connectionString = virtualDataSource.GetConnectionString(dataSourceName);
+            virtualDataSource.UseDbContextOptionsBuilder(connectionString, dbContextOptionBuilder);
+            shardingConfigOptions.ShardingMigrationConfigure?.Invoke(dbContextOptionBuilder);
+            //迁移
+            dbContextOptionBuilder.UseShardingOptions(shardingRuntimeContext);
+            return dbContextOptionBuilder.Options;
+        }
+    }
+}
+
+
+#endif

src/ShardingCore/EFCores/ShardingMigrator.cs

@@ -2,6 +2,7 @@ using System;
 using System.Collections.Generic;
 using System.Diagnostics;
 using System.Linq;
+using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.EntityFrameworkCore;
@@ -77,6 +78,14 @@ namespace ShardingCore.EFCores
            await DynamicShardingHelper.DynamicMigrateWithDataSourcesAsync(_shardingRuntimeContext, allDataSourceNames, null,cancellationToken);
 
         }
+#if EFCORE6
+
+        public override string GenerateScript(string fromMigration = null, string toMigration = null,
+            MigrationsSqlGenerationOptions options = MigrationsSqlGenerationOptions.Default)
+        {
+          return new ScriptMigrationGenerator(_shardingRuntimeContext, fromMigration, toMigration, options).GenerateScript();
+        }
+#endif
     }
 
 }