EF Core根据实体自动创建对应 DbSet

最新推荐文章于 2025-09-10 21:20:57 发布

转载最新推荐文章于 2025-09-10 21:20:57 发布 · 1k 阅读

2 ·

CC 4.0 BY-SA版权

原文链接：https://blog.youkuaiyun.com/WuLex

文章标签：

#c# #sql #.netcore #EFCORE

该博客主要介绍了在EF Core中如何通过OnModelCreating方法自定义实体的映射过程，包括从指定程序集加载实体类型，设置主键和表名。作者使用了反射和动态实例化来配置模型，并对每个实体的主键和表名进行了设定，确保数据库表结构与实体类的对应。

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    Assembly entityAssembly = Assembly.Load(new AssemblyName("XXX.Entity"));
    IEnumerable < Type > typesToRegister = entityAssembly.GetTypes().Where(p => !string.IsNullOrEmpty(p.Namespace)).Where(p => !string.IsNullOrEmpty(p.GetCustomAttribute < TableAttribute > () ? .Name));
    foreach(Type type in typesToRegister)
    {
        dynamic configurationInstance = Activator.CreateInstance(type);
        modelBuilder.Model.AddEntityType(type);
    }
    foreach(var entity in modelBuilder.Model.GetEntityTypes())
    {
        PrimaryKeyConvention.SetPrimaryKey(modelBuilder, entity.Name);
        string currentTableName = modelBuilder.Entity(entity.Name).Metadata.GetTableName();
        modelBuilder.Entity(entity.Name).ToTable(currentTableName);
        //var properties = entity.GetProperties();
        //foreach (var property in properties)
        //{
        //    ColumnConvention.SetColumnName(modelBuilder, entity.Name, property.Name);
        //}
    }
    base.OnModelCreating(modelBuilder);
}