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或者支持下自家土蜂蜜吧，也诚挚欢迎各路大佬提出建议以及批评指正。