学习实体框架6命令树拦截器

时间:2015-03-31 20:00:00

标签: c# entity-framework entity-framework-6 interceptor

这纯粹是一个学习实验(耶和华科学!)。这并不意味着实际上可以在任何地方使用。我想了解EF6的命令树拦截器是如何工作的。

我正在尝试修改截获的命令树,为所有查询添加“IsActive = 1”过滤器。我注意到这类文件的文档存在严重缺陷。两个问题:

如何有选择地拦截命令树,例如实现接口的实体(例如IHasAnActiveProperty)?现在我注意到拦截器正在拦截对历史上下文的查询,这与我的MyEntity无关。

如何将过滤器添加到所有查询中?

public class MyEntity
{
    [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
    public int Id { get; set; }

    public string Name { get; set; }

    public bool IsActive { get; set; }
}

public class MyDbContext : DbContext
{
    public DbSet<MyEntity> Entities { get; set; }

    public MyDbContext() : base("default") { }


    public MyDbContext(string nameOrConnectionString)
        : base(nameOrConnectionString) {}
}

public class MyConfiguration : DbConfiguration
{
    public MyConfiguration()
    {
        AddInterceptor(new MyInterceptor());
    }
}

public class MyInterceptor : IDbCommandTreeInterceptor
{
    public void TreeCreated(DbCommandTreeInterceptionContext interceptionContext)
    {
        var query = interceptionContext.Result as DbQueryCommandTree;
        if (query != null)
        {
            Debug.WriteLine("##################");
            Debug.WriteLine("DataSpace: {0}", interceptionContext.Result.DataSpace);
            Debug.WriteLine(query.ToString());

            query.Query.Accept(new MyVisitor());
        }
    }
}

这就是问题的关键所在。正如您所看到的,我已经确定了一些需要包含的内容,但我不知道如何将这些内容放在一起。

public class MyVisitor : BasicExpressionVisitor
{
    public override void Visit(DbFilterExpression expression)
    {
        // add WHERE [IsActive] = 1
        /* Building blocks:
         *      DbExpressionBuilder.Equal
         *      DbExpressionBuilder.True
         *      DbExpressionBuilder.And   --- only when a filter expression already exists
         */

        var isActiveProperty = expression.Property("IsActive");
        var equalExp = DbExpressionBuilder.Equal(isActiveProperty, DbExpressionBuilder.True);
    }
}

假设我们有这个:

class Program
{
    static void Main(string[] args)
    {
        using (var ctx = new MyDbContext())
        {
            var entities = ctx.Entities.FirstOrDefault(x => x.Name == "amy");
            Console.WriteLine(entities);
        }
    }
}

此查询将生成此命令树:

DbQueryCommandTree
|_Parameters
|_Query : Collection{Record['Id'=Edm.Int32, 'Name'=Edm.String, 'IsActive'=Edm.Boolean]}
  |_Project
    |_Input : 'Limit1'
    | |_Limit
    |   |_Filter
    |   | |_Input : 'Extent1'
    |   | | |_Scan : CodeFirstDatabase.MyEntity
    |   | |_Predicate
    |   |   |_
    |   |     |_'amy'
    |   |     |_=
    |   |     |_Var(Extent1).Name
    |   |_1
    |_Projection
      |_NewInstance : Record['Id'=Edm.Int32, 'Name'=Edm.String, 'IsActive'=Edm.Boolean]
        |_Column : 'Id'
        | |_Var(Limit1).Id
        |_Column : 'Name'
        | |_Var(Limit1).Name
        |_Column : 'IsActive'
          |_Var(Limit1).IsActive

但我不明白从哪里开始。

1 个答案:

答案 0 :(得分:1)

我使用以下引用来帮助我实现自定义“软删除”拦截机制:

希望这会有所帮助