You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

基于实体类型重写EF 6默认SQL生成逻辑的实现方法

在EF 6.2中为实现特定接口的实体自动追加扩展属性SQL

完全可以实现!EF 6.2提供了自定义迁移代码生成器的扩展点,能让你针对实现特定接口的实体,自动在迁移代码里追加那段扩展属性的SQL执行逻辑。下面是具体的实现步骤:

1. 先修正你的接口定义(小提示:你之前写的IMyEnterface打错字啦,应该是IMyInterface

public interface IMyInterface { }

2. 创建自定义迁移生成器

继承EF默认的CSharpMigrationGenerator,重写生成建表操作的逻辑,判断实体是否实现了IMyInterface,如果是就追加对应的SQL代码:

using System;
using System.Data.Entity.Migrations.Design;
using System.Data.Entity.Migrations.Model;

public class CustomMigrationGenerator : CSharpMigrationGenerator
{
    protected override void Generate(CreateTableOperation createTableOperation)
    {
        // 先让基类生成默认的建表代码
        base.Generate(createTableOperation);

        // 从迁移操作的注解中获取实体对应的CLR类型
        if (createTableOperation.Annotations.TryGetValue("ClrType", out var clrTypeObj) 
            && clrTypeObj is Type entityType)
        {
            // 检查实体是否实现了目标接口
            if (typeof(IMyInterface).IsAssignableFrom(entityType))
            {
                // 追加扩展属性的SQL执行代码
                WriteLine("            Sql(@\"EXEC sys.sp_addextendedproperty bla-bla-bla\");");
            }
        }
    }
}

3. 配置DbContext使用自定义生成器

在你的迁移配置类(继承自DbMigrationsConfiguration)里,指定使用我们刚才创建的生成器:

public class MyDbMigrationConfig : DbMigrationsConfiguration<YourDbContext>
{
    public MyDbMigrationConfig()
    {
        // 替换默认的代码生成器
        CodeGenerator = new CustomMigrationGenerator();
    }
}

4. 生成迁移时生效

之后你用Package Manager Console生成迁移的时候,只要指定这个配置类(如果不是项目默认的配置类的话):

Add-Migration YourMigrationName -ConfigurationTypeName MyDbMigrationConfig

生成后的迁移代码里,所有实现IMyInterface的实体对应的CreateTable方法后面,都会自动加上那段Sql(...)调用,完全符合你的需求。

原理说明

EF 6的迁移系统会把实体的元数据(比如CLR类型)存在迁移操作的Annotations集合里,我们通过读取ClrType注解就能拿到实体的类型,进而判断是否实现了目标接口。重写CSharpMigrationGenerator的方法,就能干预最终生成的迁移代码内容。

内容的提问来源于stack exchange,提问作者UserControl

火山引擎 最新活动