[EF Core] EF Core Code-First 移除外键 —— 重写SQL生成器

[EF Core] EF Core Code-First 移除外键 —— 重写SQL生成器

使用EF Core时最烦的就是生成的某些 SQL 其实并不是你想要的结果,例如外键约束等等。

一个最简单的例子就是,因为EF Core会根据导航属性生成外键约束等原因,导致很多开发者抛弃了更易维护的Code First模式,而转为Db First以获取更自由的数据库结构。

其实我们可以通过重写EF CoreMigrationsSqlGenerator来解决:

public class CustomMigrationsSqlGenerator : MigrationsSqlGenerator
{
    public CustomMigrationsSqlGenerator(MigrationsSqlGeneratorDependencies dependencies, IMigrationsAnnotationProvider migrationsAnnotations) : base(dependencies)
    {
    }

    protected override void Generate(Microsoft.EntityFrameworkCore.Migrations.Operations.CreateTableOperation operation, IModel? model, MigrationCommandListBuilder builder, bool terminate = true)
    {
        operation.ForeignKeys.Clear();
        base.Generate(operation, model, builder, terminate);
    }
}
public class CustomDbContext : DbContext
{
    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        // ...
        optionsBuilder.ReplaceService<IMigrationsSqlGenerator, CustomMigrationsSqlGenerator>();
        // ...
    }
}

上述代码便提供了移除ForeignKeys即表结构中的外键约束。当然,需要在DbContext中重写OnConfiguring方法,使用optionsBuilder.ReplaceService将默认的SQL生成器,替换为我们自己实现的部分,就可以在EF Core生成数据库结构时,直接去除外键约束了。

初次之外,根据MigrationsSqlGenerator暴露出来的 API 来看,我们还能做很多事情,例如重写字段的类型或者在实体上做文章,具体需要大家根据场景进行探寻了。