酷应用

Entity Framework Core 实现全局查询过滤

百家作者：CSDN 2019-11-12 10:37:40

作者 | 喵叔

责编 | 刘静

出品 | CSDN（ID：CSDNnews）

微软在 Entity Framework Core 2+ 中引入了全局查询过滤器，简化了构建多租户应用程序和实体软删除的复杂度。这篇文章我将通过代码的形式对全局过滤查询进行详细的讲解。在讲解前我们先来简单说一下什么是多租户，所谓多租户简单来说是指一个单独的实例可以为多个组织服务。多租户技术为共用的数据中心内如何以单一系统架构与服务提供多数客户端相同甚至可定制化的服务，并且仍然可以保障客户的数据隔离。

接下来我们先来看一个例子，我们假定多个租户使用同一个数据库，同一个Schema，区分租户是根据表中的 tId 区分。我们新建一个项目，在项目中重写 DbContext 上下文里的 OnModelCreating 方法，在这个方法中我们使用 HasQueryFilter 方法进行软删除。

public?class?EFContext?:?DbContext
{
????protected?override?void?OnModelCreating(ModelBuilder?modelBuilder)
????{
????????modelBuilder.Entity().HasQueryFilter(p?=>?!p.IsDelete);
????????modelBuilder.Entity().HasQueryFilter(p?=>?!p.IsDelete);
????????base.OnModelCreating(modelBuilder);
????}
}

上面这段代码大部分人看完一定觉得没有问题，但是我要告诉各位的是这段代码有一个很大的问题，就是每次去操作 Employee 或 Department 实体时都会应用 HasQueryFilter , 那么如果由几十个甚至上百个 Model 的时候，如果按照上述的方式来编写代码的话那么我们将会配置 HasQueryFilter 几十次上百次。那么问题来了，着这种情况下我们应该怎样支持多租户，应该怎样实现软删除，以及应该怎样实现模型查询过滤的自动检测。下面我们就来一个问题一个问题的讲解。

准备基础代码

我们首先准备 Model 代码，代码很简单：

///?
///?实体基类
///?
public?class?BaseModel
{
????public?int?Id?{?get;?set;?}
????//租户Id
????public?int?TId?{?get;?set;?}
????//是否删除（软删除）
????public?bool?IsDelete?{?get;?set;?}
}
///?
///?员工类
///?
public?class?Employee:BaseModel
{
????public?string?Name?{?get;?set;?}
????public?int?Age?{?get;?set;?}
????public?Department?Department?{?get;?set;?}
}
///?
///?部门类
///?
public?class?Department:BaseModel
{
????public?string?DepName?{?get;?set;?}
????public?int?EmployeeId?{?get;?set;?}
}

下面我们实现 Employee 和 Department 的模型映射：

public?class?EmployeeConfiguration?:?IEntityTypeConfiguration< Employee>
{
????public?void?Configure(EntityTypeBuilder?builder)
????{
????????builder.ToTable("Employee");
????????builder.HasKey(p?=>?p.Id);
????????builder.Property(p?=>?p.Name);
????????builder.Property(p?=>?p.IsDelete);
????????builder.Property(p?=>?p.Age);
????????builder.HasQueryFilter(p?=>?!p.IsDelete);
????}
}
public?class?DepartmentConfiguration?:?IEntityTypeConfiguration< Department>
{
????public?void?Configure(EntityTypeBuilder?builder)
????{
????????builder.ToTable("Department");
????????builder.HasKey(p?=>?p.Id);
????????builder.Property(p?=>?p.DepName);
????????builder.Property(p?=>?p.IsDelete);
????????builder.HasQueryFilter(p?=>?!p.IsDelete);
????????builder.HasMany(p?=>?p.Employees).WithOne(e?=>?e.Department).HasForeignKey(k?=>?k.DepartmentId);
????}
}

接下来我们再定义一个多租户Id接口和实现，实现里的返回值是定死的，用来模拟检测与当前请求相关的租户。

public?interface?ITenantProvider
{
????int?GetTId();
}
public?class?TenantProvider?:?ITenantProvider
{
????public?int?GetTId()
????{
????????return?1;
????}
}

最后我们定义查询 Department 的接口和实现：

public?interface?IDepartmentDb
{
????IQueryable?GetDepartments();
}
public?class?DepartmentDb?:?IDepartmentDb
{
????private?EFContext?ef;
????public?DepartmentDb(EFContext?_ef)
????{
????????ef?=?_ef;
????}
????public?IQueryable?GetDepartments()
????{
????????var?departments?=?ef.Departments;
????????return?departments;
????}
}

前提条件

如果要为所有实体配置全局查询过滤器，就必须能够自动检测出实体类型，同时类型检测时必须具有缓存支持。基于这两条我们动手创建获取实体类型的接口和实现。首先利用 DependencyContext 获取运行时程序集，将获得的程序集添加到集合中，然后查找出继承自基类 BaseModel 的程序集，如果查找到了就返回，如果没有查找到就实现全局过滤缓存，代码如下：

public?interface?IEntityTypeProvider
{
?????IEnumerable?GetTypes();
}

public?class?EntityTypeProvider?:?IEntityTypeProvider
{
????IList?entityTypeCache;
????public?IEnumerable?GetTypes()
????{
????????if(entityTypeCache!=null)
????????{
????????????return?entityTypeCache.ToList();
????????}
????????entityTypeCache?=?(from?a?in?GetReferencingAssemblies()
???????????????????????????from?t?in?a.DefinedTypes
???????????????????????????where?t.BaseType?==?typeof(BaseModel)
???????????????????????????select?t.AsType()).ToList();
????????return?entityTypeCache;
????}

????private?IEnumerable?GetReferencingAssemblies()
????{
????????var?assemblies?=?new?List();
????????var?dependencies?=?DependencyContext.Default.RuntimeLibraries;
????????foreach?(var?library?in?dependencies)
????????{
????????????var?assembly?=?Assembly.Load(new?AssemblyName(library.Name));
????????????assemblies.Add(assembly);
????????}
????????return?assemblies;
????}

}

应用

上一小节我们查找到了继承基类的所有实体，那么现在我们就将全局过滤器应用到实体。

第一步首先，获取租户 id 和前面对应的实现，并注入到上下文构造函数中：

public?class?EFContext?:?DbContext
{
????public?DbSet?Employees;
????public?DbSet?Departments;
????protected?override?void?OnModelCreating(ModelBuilder?modelBuilder)
????{
????????modelBuilder.ApplyConfiguration(new?EmployeeConfiguration());
????????modelBuilder.ApplyConfiguration(new?DepartmentConfiguration());
????????base.OnModelCreating(modelBuilder);
????}
????private?int?tId;
????private?IEntityTypeProvider?entityTypeProvider;
????public?EFContext(DbContextOptions?options,ITenantProvider?tenantProvider,IEntityTypeProvider?entityTypeProvider):base(options)
????{
????????tId?=?tenantProvider.GetTId();
????????this.entityTypeProvider?=?entityTypeProvider;
????}
}

第二步在上下文中定义全局查询过滤的泛型方法：

public?void?GlobalQuery?(ModelBuilder?builder)?where?T?:BaseModel
{
????builder.Entity().HasQueryFilter(e?=>?e.TId?==?tId?&&?!e.IsDelete);
}

然后在上下文中获取设置全局查询过滤方法的 MethodInfo 类：

static?readonly?MethodInfo?GlobalQueryMethod?=?typeof(EFContext).GetMethods(BindingFlags.Public?|?BindingFlags.Instance).Single(p=>p.IsGenericMethod?&&?p.Name==?"GlobalQuery");

最后，在OnModelCreating方法中，通过注入的 entityTypeProvider 获取需要全局查询过滤的类型集合，并进行遍历，调用得到进行查询过滤的方法传入 modelBuilder 参数，从而实现多租户查询过滤。

protected?override?void?OnModelCreating(ModelBuilder?modelBuilder)
{
????modelBuilder.ApplyConfiguration(new?EmployeeConfiguration());
????modelBuilder.ApplyConfiguration(new?DepartmentConfiguration());
????foreach?(var?item?in?entityTypeProvider.GetTypes())
????{
????????var?method?=?GlobalQueryMethod.MakeGenericMethod(item);
????????method.Invoke(this,?new?object[]?{?modelBuilder?});
????}
????base.OnModelCreating(modelBuilder);
}