EF Core懒人小技巧之拒绝DbSet

前言

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

正经事

1. 约定规则

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

 using Microsoft.EntityFrameworkCore.Infrastructure;
 using System;
 using System.ComponentModel.DataAnnotations;
 using System.ComponentModel.DataAnnotations.Schema;
 using Zxw.Framework.NetCore.DbContextCore;
 using Zxw.Framework.NetCore.Models;

 namespace Zxw.Framework.Website.Models
 {
     [DbContext(typeof(SqlServerDbContext))]
     [Table("SysMenu")]
     public class SysMenu:BaseModel<string>
     {
         [Key]
         [Column("SysMenuId")]
         public override string Id { get; set; }

         public string ParentId { get; set; } = String.Empty;

         [MaxLength()]
         public string MenuPath { get; set; }

         [Required]
         [MaxLength()]
         public string MenuName { get; set; }

         [MaxLength()]
         public string MenuIcon { get; set; }

         [Required]
         [MaxLength()]
         public string Identity { get; set; }

         [Required]
         [MaxLength()]
         public string RouteUrl { get; set; }

         public bool Visiable { get; set; } = true;

         public bool Activable { get; set; } = true;

         [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
         public int SortIndex { get; set; }
     }
 }

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

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

先上代码为敬：

         protected override void OnModelCreating(ModelBuilder modelBuilder)
         {
             MappingEntityTypes(modelBuilder);
             base.OnModelCreating(modelBuilder);
         }

         private void MappingEntityTypes(ModelBuilder modelBuilder)
         {
             if (string.IsNullOrEmpty(Option.ModelAssemblyName))
                 return;
             var assembly = Assembly.Load(Option.ModelAssemblyName);
             var types = assembly?.GetTypes().Where(c=>c.GetCustomAttributes<DbContextAttribute>().Any());
             var list = types?.Where(t =>
                 t.IsClass && !t.IsGenericType && !t.IsAbstract &&
                 t.GetInterfaces().Any(m =>
                     m.IsAssignableFrom(typeof(BaseModel<>)) || m.IsAssignableFrom(typeof(BaseViewModel)))).ToList();
             if (list != null && list.Any())
             {
                 list.ForEach(t =>
                 {
                     var dbContextType = t.GetCustomAttributes<DbContextAttribute>().FirstOrDefault(x=>x.ContextType==GetType());
                     if (modelBuilder.Model.FindEntityType(t) == null && null!=dbContextType)
                         modelBuilder.Model.AddEntityType(t);
                 });
             }
         }

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

结束语

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