EF Core懒人小技巧之拒绝DbSet

前言

最近在项目中使用EF Core的频率越来越高，当项目比较大的时候，疯狂往DbContext中加各种DbSet，你会不会特难受？如果你是一键生成的大佬，那么请忽略本文。本文旨在不写 DbSet，那么就撸起柚(xiu)子干吧...

正经事

1. 约定规则

首先，为了达到偷懒的目的，我们得事先约定些规则，毕竟无规矩不成方圆。具体规则可以自定义，比如我目前的规则就是在数据表对应的实体类加上些特有的Attribute，比如现成的DbContextAttribute，这个Attribute的具体作用请参考MSDN官方文档。

 1 using Microsoft.EntityFrameworkCore.Infrastructure;
 2 using System;
 3 using System.ComponentModel.DataAnnotations;
 4 using System.ComponentModel.DataAnnotations.Schema;
 5 using Zxw.Framework.NetCore.DbContextCore;
 6 using Zxw.Framework.NetCore.Models;
 7 
 8 namespace Zxw.Framework.Website.Models
 9 {
10     [DbContext(typeof(SqlServerDbContext))]
11     [Table("SysMenu")]
12     public class SysMenu:BaseModel<string>
13     {
14         [Key]
15         [Column("SysMenuId")]
16         public override string Id { get; set; }
17 
18         public string ParentId { get; set; } = String.Empty;
19 
20         [MaxLength(2000)]
21         public string MenuPath { get; set; }
22 
23         [Required]
24         [MaxLength(20)]
25         public string MenuName { get; set; }
26 
27         [MaxLength(50)]
28         public string MenuIcon { get; set; }
29 
30         [Required]
31         [MaxLength(100)]
32         public string Identity { get; set; }
33 
34         [Required]
35         [MaxLength(200)]
36         public string RouteUrl { get; set; }
37 
38         public bool Visiable { get; set; } = true;
39 
40         public bool Activable { get; set; } = true;
41 
42         [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
43         public int SortIndex { get; set; }
44     }
45 }

View Code

OK，规则我们约定好了，下面进行下一步。

2. 重写数据库上下文的OnModelCreating方法

先上代码为敬：

 1         protected override void OnModelCreating(ModelBuilder modelBuilder)
 2         {
 3             MappingEntityTypes(modelBuilder);
 4             base.OnModelCreating(modelBuilder);
 5         }
 6 
 7         private void MappingEntityTypes(ModelBuilder modelBuilder)
 8         {
 9             if (string.IsNullOrEmpty(Option.ModelAssemblyName))
10                 return;
11             var assembly = Assembly.Load(Option.ModelAssemblyName);
12             var types = assembly?.GetTypes().Where(c=>c.GetCustomAttributes().Any());
13             var list = types?.Where(t =>
14                 t.IsClass && !t.IsGenericType && !t.IsAbstract &&
15                 t.GetInterfaces().Any(m =>
16                     m.IsAssignableFrom(typeof(BaseModel<>)) || m.IsAssignableFrom(typeof(BaseViewModel)))).ToList();
17             if (list != null && list.Any())
18             {
19                 list.ForEach(t =>
20                 {
21                     var dbContextType = t.GetCustomAttributes().FirstOrDefault(x=>x.ContextType==GetType());
22                     if (modelBuilder.Model.FindEntityType(t) == null && null!=dbContextType)
23                         modelBuilder.Model.AddEntityType(t);
24                 });
25             }
26         }

View Code

到此，或许已经明白，重要的只有一句：只要在 ModelBuilder.Model 中注册我们的数据表实体类就可以代替万千DbSet了。

结束语

本次分享就到此结束，不妨亲自试试。以上代码可以在我的个人开源项目 【Zxw.Framework.NetCore】中找到，如果本项目对您有帮助，请给个star或者支持下自家土蜂蜜吧，也诚挚欢迎各路大佬提出建议以及批评指正。