继上篇文章（EF Core懒人小技巧之拒绝DbSet）之后，最近笔者把这个小功能单独封装成一个扩展方法并开源，欢迎交流和Star~
GitHub： EntityFrameworkCore.Extension.AutoMapping
Nuget：EntityFrameworkCore.Extension.AutoMapping
EntityFrameworkCore.Extension.AutoMapping.Abp
EntityFrameworkCore.Extension.AutoMapping.AbpVNext
如何使用
在DbContext.cs中重写OnModelCreating方法：
using EntityFrameworkCore.Extension;
... //此处省略其它代码
public class XmateDbContext:DbContext
{
... //此处省略其它代码
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
var modelAssemblyName = "XMate.Models";//实体类所在类库的名称，不包含扩展名(.dll)
modelBuilder.AutoMappingEntityTypes<IEntity>(modelAssemblyName);//泛型IEntity为所有实体类的规约类型
base.OnModelCreating(modelBuilder);//这个必须加，否则报错
...//此处省略其它代码
}
}
这样，我们就可以不用写满屏的DbSet了。
但是，在有的第三方框架中可能就会诞生新的问题。。。
比如在ABP或者VNext框架中，用过ABP框架的都应该知道，ABP是通过扫描DbContext中的DbSet来实现将实体类的仓储自动注册到IOC容器中的，下面我们就需要自己动手来实现：
public static class AutoRegisterEntityRepositoryExtensions
{
/// <summary>
/// 将数据表实体类型对应的仓储注入到IOC容器
/// </summary>
/// <param name="iocManager"></param>
public static void RegisterDbEntityRepositories<TDbContext>(this IIocManager iocManager, string modelAssemblyName) where TDbContext : DbContext
{
foreach (var entityType in GetDbEntityType(typeof(IEntity<>), modelAssemblyName))
{
var keyType = entityType.GetInterfaces().Where(t => t.IsGenericType && t.GetGenericTypeDefinition() == typeof(IEntity<>)).SelectMany(t => t.GetGenericArguments()).First();
var genericRepositoryType = typeof(IRepository<,>).MakeGenericType(entityType, keyType);
var impType = typeof(EfCoreRepositoryBase<,,>).MakeGenericType(typeof(TDbContext), entityType, keyType);
iocManager.RegisterIfNot(genericRepositoryType, impType, lifeStyle: DependencyLifeStyle.Transient);
}
}
/// <summary>
/// 获取数据表实体类型列表
/// </summary>
/// <param name="constraintType">实体定义约束类型</param>
/// <param name="modelAssemblyName">实体类所在dll名称，不包含后缀名(.dll)</param>
/// <returns></returns>
private static List<Type> GetDbEntityType(Type constraintType, string modelAssemblyName)
{
var all = AppDomain.CurrentDomain.GetAssemblies();
var types = all.WhereIf(!modelAssemblyName.IsNullOrWhiteSpace(), a => a.FullName.Contains(modelAssemblyName))
.SelectMany(m => m.GetTypes().Where(t => t.IsClass && !t.IsAbstract && (t.IsImplement(constraintType) || t.IsSubclass(constraintType))).ToList())
.Distinct()
.ToList();
return types.Where(t => !t.GetCustomAttributes<NotMappedAttribute>().Any()).ToList();
}
}
注：以上代码摘自：AutoRegisterEntityRepositoryExtensions.cs
在ABP VNext中的实现思路也是如此，这里就不贴代码了，感兴趣的可以查阅源代码
在Abp中实现自动注入实体类对应的Repository
using EntityFrameworkCore.Extension.AutoMapping.Abp;
... //此处省略其它代码
public class XmateModule:AbpModule
{
... //此处省略其它代码
//重写Initialize方法
public override void Initialize()
{
... //此处省略其它代码
var modelAssemblyName = "XMate.Models";//实体类所在类库的名称，不包含扩展名(.dll)
IocManager.RegisterDbEntityRepositories(modelAssemblyName);
}
}
在Abp VNext中实现自动注入实体类对应的Repository
using EntityFrameworkCore.Extension.AutoMapping.AbpVNext;
... //此处省略其它代码
public class XmateModule:AbpModule
{
... //此处省略其它代码
//重写ConfigureServices方法
public override void ConfigureServices(ServiceConfigurationContext context)
{
... //此处省略其它代码
var modelAssemblyName = "XMate.Models";//实体类所在类库的名称，不包含扩展名(.dll)
context.Services.RegisterDbEntityRepositories(modelAssemblyName);
}
}
