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

.NET跨平台

于 2022-09-04 22:34:43 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

文章标签： c# sql .netcore EFCORE

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

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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);
}