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(2000)]
        public string MenuPath { get; set; }

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

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

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

        [Required]
        [MaxLength(200)]
        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或者支持下自家土蜂蜜吧，也誠摯歡迎各路大佬提出建議以及批評指正。