EF Core中如何拦截架构创建并通过扩展方法添加实体

针对你的需求，推荐使用装饰器模式包装IModelCustomizer，既保留用户自定义的模型配置逻辑，又能自动添加你的实体类，无需担心覆盖原有功能。以下是具体实现步骤：

1. 实现IModelCustomizer装饰器 #

创建一个装饰器类，包装原有的IModelCustomizer实例，先执行用户的自定义逻辑，再添加你的实体配置：

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Metadata.Builders;

public class MyModelCustomizerDecorator : IModelCustomizer
{
    private readonly IModelCustomizer _originalCustomizer;

    // 注入原有的IModelCustomizer实例
    public MyModelCustomizerDecorator(IModelCustomizer originalCustomizer)
    {
        _originalCustomizer = originalCustomizer;
    }

    public void Customize(ModelBuilder modelBuilder, DbContext context)
    {
        // 先执行用户原有的模型自定义逻辑（包括DbContext的OnModelCreating）
        _originalCustomizer.Customize(modelBuilder, context);

        // 添加你的实体类配置，自动生成对应数据表
        modelBuilder.Entity<YourLibraryEntity>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.ToTable("YourLibraryEntities"); // 指定表名
            // 其他属性配置示例
            entity.Property(e => e.Name).IsRequired().HasMaxLength(100);
        });
    }
}

// 你的类库中的实体类示例
public class YourLibraryEntity
{
    public int Id { get; set; }
    public string Name { get; set; }
    // 其他业务属性...
}

2. 编写DbContextOptionsBuilder扩展方法 #

在扩展方法中替换IModelCustomizer服务为你的装饰器，确保EF Core使用增强后的逻辑：

public static class DbContextOptionsBuilderExtensions
{
    public static DbContextOptionsBuilder MyExtensionMethod(this DbContextOptionsBuilder options)
    {
        // 替换服务为装饰器，保留原有的IModelCustomizer实现
        options.ReplaceService<IModelCustomizer, MyModelCustomizerDecorator>();

        return options;
    }
}

3. 用户侧使用方式 #

用户只需在注册DbContext时调用你的扩展方法即可，无需修改自身DbContext或额外配置：

Services.AddDbContext<MyContext>(options => options
     .UseSqlServer("YourConnectionString")
     .MyExtensionMethod())

方案优势 #

• 无侵入性：用户无需修改现有DbContext代码，仅需调用扩展方法
• 兼容用户自定义配置：装饰器会先执行原有的IModelCustomizer逻辑（包括用户重写的OnModelCreating或自定义的IModelCustomizer），再添加你的实体配置，不会覆盖用户功能
• 逻辑集中：所有自动添加实体的逻辑封装在类库中，用户无需关注细节

内容的提问来源于stack exchange，提问作者sindiofala