在Code|在Code First模式中自动创建Entity模型

之前我在博客文章中介绍过如何使用Code First来创建数据库,通过CodeFirst的方式,可以大幅的减少开发人员的工作量,比传统的先创建库再ORM的方式简单了不少。但是,很多时候,特别是一些MIS系统,我们会涉及到大量的实体类型,类似如下所示:

public class DbContext : System.Data.Entity.DbContext{public DbContext() : base("name=DefaultConnection"){}public DbSet Penlils { get; set; }public DbSet Pens { get; set; }public DbSet Inks { get; set; }public DbSet Erasers { get; set; }//....}

在常用的CodeFirst方式下,每增加一种类型就需要在DbContext中增加一个属性。虽然并不算麻烦,但这种手动维护的方式存在如下两个问题:
  • 当实体类型较多且变更比较频繁的的时候,靠手动的方式维护实体列表很容易出错。
  • 有的时候,客户并不会买整套产品,只需要里面的部分模块,手动维护的方式不方便模块的裁剪。
此时,就需要我们来实现动态创建实体模型了,Entity Framework中本身是支持动态创建实体模型的,上面的实体模型就可以通过如下方式动态创建:
public class DbContext : System.Data.Entity.DbContext{public DbContext() : base("name=DefaultConnection"){}protected override void OnModelCreating(DbModelBuilder modelBuilder){base.OnModelCreating(modelBuilder); modelBuilder.Entity(); modelBuilder.Entity(); modelBuilder.Entity(); modelBuilder.Entity(); //....}}

PS:修改成这样的方式后,原来的代码可能出现如下问题:DbContext中没有 Inks属性了。此时只需要将原来的对db.Inks的访问换成 db.Set即可
结果上述操作后,虽然我们实现了动态创建,但实体类型还是手动添加的。因此我们还缺少一种实体类型的发现机制,这种发现机制在.net中实现还是比较简单的,这里我采用的是Attribute的方式。
首先写一个Attribute,
[AttributeUsage(AttributeTargets.Class, AllowMultiple = false, Inherited = false)]public class PersistentAttribute : Attribute{}

然后将需要自动创建的实体用该Attribute标记,
[Persistent]public class Pen[Persistent]public class Ink

最后,根据标记的实体添加实体模型。
public class DbContext : System.Data.Entity.DbContext{public DbContext() : base("name=DefaultConnection"){}protected override void OnModelCreating(DbModelBuilder modelBuilder){base.OnModelCreating(modelBuilder); var assembly = this.GetType().Assembly; var entityTypes = from type in assembly.GetTypes()where type.GetCustomAttribute() != nullselect type; var entityMethod = typeof(DbModelBuilder).GetMethod("Entity"); foreach (var type in entityTypes){entityMethod.MakeGenericMethod(type).Invoke(modelBuilder, new object[] { }); }}}

通过上述方法,就可以实现实体模型的自动创建了。
【在Code|在Code First模式中自动创建Entity模型】以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

    推荐阅读