日常开发中,经常会碰到一些自引用的实体,比如系统菜单、目录实体,这类实体往往自己引用自己,所以我们必须学会使用Code First来建立这一类的模型.
以下是自引用表的数据库关系图:
ok,下面开始介绍从零创建一个Code First版的自引用模型.
1、往目标项目中添加EF包,通过NuGet程序包添加
导入相关的程序集.
2、创建自引用实体类
public class Category
{
[Key]
[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
public int CategoryId { get; private set; }
public string Name { get; set; }
public int? ParentCategoryId { get; private set; }
[ForeignKey("ParentCategoryId")]
public virtual Category ParentCategory { get; set; }
public virtual List<Category> Subcategories { get; set; }
public Category()
{
Subcategories = new List<Category>();
}
}
3、创建一个数据库上下文,该上下文必须继承DbContext,代码如下:
public class EF6RecipesContext : DbContext
{
public DbSet<Category> Categories { get; set; }
public EF6RecipesContext()
: base("name=EF6RecipeEntities")
{
}
protected override void OnModelCreating(DbModelBuilder modelBuilder)
{
base.OnModelCreating(modelBuilder);
modelBuilder.Entity<Category>().HasMany(cat => cat.Subcategories).WithOptional(cat => cat.ParentCategory);
}
}
4、截至这一步,分析下代码,典型的目录实体,从实体类可以看出该实体拥有单个父类型、子类型集合,这里比较特殊的是,这里的父类型和子类型都是自己,也就是自引用.注意:一个没有付类型的实体,该实体就是整个继承类型的最顶端.
5、编写测试代码:
static void Main(string[] args)
{
Example();
}
static void Example()
{
using (var context = new EF6RecipesContext())
{
var first = new Category { Name = "第一级菜单" };
var second = new Category { Name = "第二级菜单" };
first.Subcategories.Add(second);
second = new Category { Name = "第二级菜单" };
first.Subcategories.Add(second);
second = new Category { Name = "第二级菜单" };
first.Subcategories.Add(second);
var top = new Category { Name = "*菜单" };
top.Subcategories.Add(first);
context.Categories.Add(top);
context.SaveChanges();
}
using (var context = new EF6RecipesContext())
{
var roots = context.Categories.Where(c => c.ParentCategory == null);
roots.ToList().ForEach(root => Print(root, ));
}
Console.ReadKey();
}
static void Print(Category cat, int level)
{
StringBuilder sb = new StringBuilder();
Console.WriteLine("{0}{1}", sb.Append(' ', level).ToString(), cat.Name);
cat.Subcategories.ForEach(child => Print(child, level + ));//递归,直到最后遍历的节点没有子节点集合,则跳出递归循环
}
简单解释下测试代码的逻辑:
(1)、从所有的节点中获取没有父节点的节点,该节点为*节点
(2)、然后通过递归将该*节点下面的所有的子节点全部遍历出来,每当递归到的节点含有子节点集合,则递归的深度加1.当一个继承链遍历完毕,继续遍历第二个继承链.