向所有查询实体框架添加过滤器

Question

我想将 CompanyID 过滤器添加到我的所有实体框架请求中。因为每个用户必须只看到他们的记录。我不想在业务层中添加过滤器 (x=>x.CompanyID == cID) 的所有方法。如何自动添加过滤器要求。

我在 DAL 中的 GetList 方法

     public List<TEntity> GetList(Expression<Func<TEntity, bool>> filter)
    {
        using (var context = new TContext())
        {

            return filter == null
                ? context.Set<TEntity>().ToList()
                : context.Set<TEntity>().Where(filter).ToList();
        }
    }

商业

   public List<FinanceData> GetAll()
        {
            return _financeDal.GetList(filter:x=>x.CompanyID==_cID);
        }

Answer 1

在Entity Framework Core 2.0 中，您可以使用全局查询过滤器。

1. 仅向一个实体添加过滤器：

    public interface IDelete { bool IsDeleted { get; set; } } public class Blog : IDelete { public int BlogId { get; set; } public string Name { get; set; } public string Url { get; set; } public List<Post> Posts { get; set; } } public class Post : IDelete { public int PostId { get; set; } public string Title { get; set; } public string Content { get; set; } public bool IsDeleted { get; set; } public int BlogId { get; set; } public Blog Blog { get; set; } } // Default method inside the DbContext or your default Context protected override void OnModelCreating(ModelBuilder modelBuilder) { base.OnModelCreating(builder); modelBuilder.Entity<Blog>() // Add Global filter to the Blog entity .HasQueryFilter(p => p.IsDeleted == false); modelBuilder.Entity<Post>() // Add Global filter to the Post entity .HasQueryFilter(p => p.IsDeleted == false); }

2. 如果你有很多实体，第一种方法不好，使用下面的代码将全局过滤器应用于所有实体（魔术方式）：

    public static class ModelBuilderExtension { public static void ApplyGlobalFilters<TInterface>(this ModelBuilder modelBuilder, Expression<Func<TInterface, bool>> expression) { var entities = modelBuilder.Model .GetEntityTypes() .Where(e => e.ClrType.GetInterface(typeof(TInterface).Name) != null) .Select(e => e.ClrType); foreach (var entity in entities) { var newParam = Expression.Parameter(entity); var newbody = ReplacingExpressionVisitor.Replace(expression.Parameters.Single(), newParam, expression.Body); modelBuilder.Entity(entity).HasQueryFilter(Expression.Lambda(newbody, newParam)); } } } // Default method inside the DbContext or your default Context protected override void OnModelCreating(ModelBuilder builder) { base.OnModelCreating(builder); modelBuilder.Entity<Blog>(); modelBuilder.Entity<Post>(); builder.ApplyGlobalFilters<IDelete>(e => e.IsDeleted == false); }

查询将是：

    exec sp_executesql N'SELECT [x].[BlogId], [x].[Name], [x].[Url]
    FROM [dbo].[Blog] AS [x]
    WHERE [x].[IsDeleted] = 0'

Answer 2

您建议的所有内容都不适用于以下情况： HasQueryFilter功能，但HasQueryFilter每个 HTTP 请求。 ApplyGlobalFilters / OnModelCreating应用于模型创建一次。 但是，如果您使用不同的请求参数重新加载页面 - 它们将不会被考虑以过滤掉DbSet 。 如果您将过滤器添加到GetAll调用 - 另一个调用“ Include ” - 实体将没有此过滤器。 我们需要真正的全局机制来根据特定条件过滤掉DbSet s - 每个请求（页面刷新）可能会改变。

Answer 3

您可以在此类实体中实现IHasCompanyId接口。 然后将存储库模式实现为：

public class MyRepository<T>
{
    public MyRepository(DbContext dbContext, int companyID)
    {
        if (dbContext == null) 
            throw new ArgumentNullException("Null DbContext");
        DbContext = dbContext;
        DbSet = DbContext.Set<T>();

        CompanyID = companyID;
    }

    protected DbContext DbContext { get; set; }
    protected int CompanyID  { get; set; }

    protected DbSet<T> DbSet { get; set; }

    // Add filter here
    public virtual IQueryable<T> GetAll()
    {
        if(typeof(IHasCompanyID).IsAssignableFrom(typeof(T)))
            return DbSet.Where(x => x.CompanyID == CompanyID);
        else 
            return DbSet;
    }
}

并将_financeDal初始化为：

var _financeDal = new MyRepository<TEntity>(dbContext, companyID);

Answer 4

您还可以扩展object context并添加返回 IQueryable 的扩展方法

喜欢

public class CustomdbContext : DbContext
{       
    public IQueryable<TEntity> ApplyCustomerFilter<TEntity>(IQueryable<TEntity> query) where TEntity : Customer 
    {
         return query.Where(x => x.CustomerId == customerctxId);
    }
}