如何无需修改DbContext即可拦截EF Core中的ModelBuilder实例

EF Core 5+ 提供了 IModelCustomizer 扩展点，可以完全满足你的非侵入式配置需求，无需修改用户的 DbContext 实现代码，仅通过 DbContextOptionsBuilder 扩展即可完成配置注入。

实现步骤 #

1. 自定义ModelCustomizer #

继承EF Core默认的 ModelCustomizer 类，重写配置逻辑，在触发用户的 OnModelCreating 方法前执行你的扩展配置：

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;

// 类库内部实现，不对外暴露
internal class AwesomeExtensionsModelCustomizer : ModelCustomizer
{
    public AwesomeExtensionsModelCustomizer(ModelCustomizerDependencies dependencies) 
        : base(dependencies)
    {
    }

    public override void Customize(ModelBuilder modelBuilder, DbContext context)
    {
        // 先执行你的扩展配置
        modelBuilder.AddAwesomeExtensionsSupportingEntities();
        // 再执行默认逻辑，触发用户重写的OnModelCreating方法
        base.Customize(modelBuilder, context);
    }
}

如果你开发的是关系型数据库专属扩展，也可以选择继承 RelationalModelCustomizer 适配关系库的默认配置逻辑。

2. 实现UseAwesomeExtensions扩展方法 #

扩展 DbContextOptionsBuilder，替换EF Core默认的 IModelCustomizer 服务为你自定义的实现：

public static class AwesomeExtensionsDbContextOptionsBuilderExtensions
{
    public static DbContextOptionsBuilder UseAwesomeExtensions(this DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.ReplaceService<IModelCustomizer, AwesomeExtensionsModelCustomizer>();
        return optionsBuilder;
    }

    // 适配泛型DbContextOptionsBuilder的重载
    public static DbContextOptionsBuilder<TContext> UseAwesomeExtensions<TContext>(this DbContextOptionsBuilder<TContext> optionsBuilder)
        where TContext : DbContext
    {
        optionsBuilder.ReplaceService<IModelCustomizer, AwesomeExtensionsModelCustomizer>();
        return optionsBuilder;
    }
}

最终使用效果 #

完全符合你预期的接入方式，用户不需要修改自己的DbContext代码：

public void AddServices(IServiceCollection services)
{
    services.AddDbContext<MyBusinessDbContext>(options =>
    {
        options.UseAwesomeExtensions();
        // 其他原有配置（如UseSqlServer等）照常编写
    });
}

注意事项 #

• 配置执行顺序可自由调整：如果你需要你的扩展配置在用户的OnModelCreating之后执行，仅需把AddAwesomeExtensionsSupportingEntities的调用放到base.Customize之后即可。
• 该方案兼容EF Core 5/6/7/8所有主流版本，无额外依赖。
• 如果存在多个扩展同时替换IModelCustomizer的场景，最后调用的扩展会生效，如需兼容多扩展场景可额外搭配装饰器模式实现。

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