数据库迁移分表

examples/Demo.DateSharding/Program.cs

@@ -11,7 +11,7 @@ class Program
         static async Task Main(string[] args)
         {
             DateTime startTime = DateTime.Now.AddMinutes(-5);
-            await Host.CreateDefaultBuilder()
+            var host = Host.CreateDefaultBuilder()
                 .ConfigureServices(services =>
                 {
                     //配置初始化
@@ -23,7 +23,7 @@ await Host.CreateDefaultBuilder()
                         //按分钟分表
                         config.SetDateSharding<Base_UnitTest>(nameof(Base_UnitTest.CreateTime), ExpandByDateMode.PerMinute, startTime);
                     });
-                }).RunConsoleAsync();
+                }).Build();
             //host.Start();
 
             //var serviceProvider = host.Services;

examples/Demo.DbMigrator/CustomContext.cs

@@ -0,0 +1,13 @@
+using EFCore.Sharding;
+
+namespace DbMigrator
+{
+    public class CustomContext : GenericDbContext
+    {
+        public CustomContext(GenericDbContext dbContext) 
+            : base(dbContext)
+        {
+
+        }
+    }
+}

examples/Demo.DbMigrator/Demo.DbMigrator.csproj

@@ -14,6 +14,8 @@
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
+    <PackageReference Include="Microsoft.Extensions.Hosting" Version="3.1.8" />
+    <PackageReference Include="Microsoft.Extensions.Hosting.Abstractions" Version="3.1.8" />
   </ItemGroup>
 
   <ItemGroup>

examples/Demo.DbMigrator/DesignTimeDbContextFactory.cs

@@ -1,18 +1,43 @@
-using Microsoft.EntityFrameworkCore;
+using EFCore.Sharding;
 using Microsoft.EntityFrameworkCore.Design;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
 
 namespace DbMigrator
 {
-    public class DesignTimeDbContextFactory : IDesignTimeDbContextFactory<DbContext>
+    public class DesignTimeDbContextFactory : IDesignTimeDbContextFactory<CustomContext>, IDesignTimeServices
     {
+        public void ConfigureDesignTimeServices(IServiceCollection serviceCollection)
+        {
+            throw new System.NotImplementedException();
+        }
+
         /// <summary>
         /// 创建数据库上下文
         /// </summary>
         /// <param name="args"></param>
         /// <returns></returns>
-        public DbContext CreateDbContext(string[] args)
+        public CustomContext CreateDbContext(string[] args)
         {
-            return new DbContext1();
+            using var host = Host.CreateDefaultBuilder()
+                .ConfigureServices(services =>
+                {
+                    services.AddEFCoreSharding(x =>
+                    {
+                        x.MigrationsWithoutForeignKey();
+                    });
+                }).Build();
+            host.Start();
+
+            var db = host.Services
+                .GetService<IDbFactory>()
+                .GetDbContext(new DbContextParamters
+                {
+                    ConnectionString = "Data Source=localhost;Initial Catalog=DbMigrator;Integrated Security=True",
+                    DbType = DatabaseType.SqlServer,
+                });
+
+            return new CustomContext(db);
         }
     }
 }

examples/Demo.DbMigrator/Entities/Order.cs

@@ -1,9 +1,11 @@
 using System;
 using System.Collections.Generic;
 using System.ComponentModel.DataAnnotations;
+using System.ComponentModel.DataAnnotations.Schema;
 
 namespace Demo.DbMigrator
 {
+    [Table(nameof(Order))]
     public class Order
     {
         [Key]

src/EFCore.Sharding/Config/Bootstrapper.cs

@@ -1,4 +1,5 @@
-using Microsoft.Extensions.Hosting;
+using EFCore.Sharding.Config;
+using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Options;
 using System;
 using System.Threading;
@@ -14,6 +15,7 @@ public Bootstrapper(IServiceProvider serviceProvider, IOptions<EFCoreShardingOpt
         {
             _serviceProvider = serviceProvider;
             _shardingOptions = shardingOptions.Value;
+            Cache.ServiceProvider = serviceProvider;
         }
         protected override Task ExecuteAsync(CancellationToken stoppingToken)
         {

src/EFCore.Sharding/Config/Cache.cs

@@ -0,0 +1,9 @@
+using System;
+
+namespace EFCore.Sharding.Config
+{
+    internal static class Cache
+    {
+        public static IServiceProvider ServiceProvider { get; set; }
+    }
+}

src/EFCore.Sharding/Config/EFCoreShardingExtensions.cs

@@ -14,13 +14,13 @@ public static class EFCoreShardingExtensions
         /// <param name="services">服务集合</param>
         /// <param name="shardingBuilder">配置项</param>
         /// <returns></returns>
-        public static IServiceCollection AddEFCoreSharding(this IServiceCollection services, Action<IShardingBuilder> shardingBuilder)
+        public static IServiceCollection AddEFCoreSharding(this IServiceCollection services, Action<IShardingBuilder> shardingBuilder = null)
         {
             services.AddOptions<EFCoreShardingOptions>();
             services.AddLogging();
 
             ShardingContainer container = new ShardingContainer(services);
-            shardingBuilder(container);
+            shardingBuilder?.Invoke(container);
 
             services.AddSingleton<IShardingBuilder>(container);
             services.AddSingleton<IShardingConfig>(container);