覆盖LINQ .Include(),可能吗?

时间:2014-01-24 08:57:01

标签: c# linq entity-framework linq-to-entities

我已经在这个问题上浏览互联网很长一段时间了,而且我在覆盖LINQ方法方面得到的结果要少得多。我不确定是否可以这样做,但我想知道是否有人可以确认这是否有效,或者建议替代方案。

情况如下(当然,这个问题简化了)

我们正在使用EF6 Code First来构建我们的数据库。我们添加了一个自定义(抽象)基类,所有实体都派生自该基类。这个基类实现了我们用于审计的一些字段(创建日期,创建日期,修改日期......),但我们还通过在 IsDeleted (bool)属性中添加了一个软删除系统。基类。

就我们的应用程序所知,绝不能返回IsDeleted == true的项目。

DataModel如下(再次,简化)

Company
    ---> 1 Company has many Departments
    Department
        ---> 1 Department has many Adresses
        Address

过去,我尝试创建一种通用的检索方法,通过在DataContext中创建表的“覆盖”来消除IsDeleted对象(也是一个自定义类,因为它会自动处理审计字段)。

对于在DataContext中找到的每个表:

public DbSet<Company> Companies { get; set; }

我们添加了第二个表,仅返回未删除的项目。

public IQueryable<Company> _Companies 
{
    get { return this.Companies .Where(co => !co.IsDeleted); }
}

所以我们拨打MyDataContext._Companies而不是MyDataContext.Companies。 这按预期工作。它可以很好地过滤掉已删除的项目。

但是,我们注意到后续.Include()语句的情况并非如此。如果我打电话:

var companies = MyDataContext._Companies.Include(x => x.Departments);

//...

还会退回公司中已删除的部门。

在我们目前的情况下,大多数核心业务逻辑已经实现,这些包含的语句都已到位。它们主要与安全有关。我可以更改所有语句,但我宁愿首先寻找一种方法来做到这一点,而不会过多地影响现有代码 这是第一个应用程序,其中查询的大小不允许我们分别调用每组实体(仅使用直接表而不是包含语句)。

所以我的问题有两个:

  • 我是否可以覆盖.Include(Func<x,y>)方法以自动包含对所选实体的IsDeleted标志的检查?
  • 如果可能,如何覆盖,如何将传递的lambda表达式与我想要执行的附加检查结合起来?

所以通过调用

someTable.Include(x => x.MyRelatedEntity);

它实际上会执行:

/* Not sure if this is proper syntax. I hope it explains what I'm trying to accomplish. */
someTable.Include(x => x.MyRelatedEntity.Where(y => !y.IsDeleted));

有人能指出我正确的方向吗?非常感谢!

注意:我知道我的问题代码不多。但我甚至不确定我能在多大程度上实现这一点。如果无法覆盖Include,还有其他方法吗?

更新

我实施了建议的解决方案,但是我遇到了所有数据库调用的问题。错误如下:

Problem in mapping fragments starting at line 245:Condition member 'Company.IsDeleted' with a condition other than 'IsNull=False' is mapped. Either remove the condition on Company.IsDeleted or remove it from the mapping.

阅读这个问题,似乎如果我使用IsDeleted作为条件(即建议的解决方案),我仍然不能将它用作属性。

然后问题变成:如何删除某些内容?删除后,永远不应该返回。但是应该可以删除未删除的项目。

我有什么方法可以通过IsDeleted过滤退回的商品,但我仍然允许将其设置为true并保存

3 个答案:

答案 0 :(得分:2)

您正在寻找的解决方案是要求实体的IsDeleted值为false

modelBuilder.Entity<Company>()
    .Map( emc => emc.Requires( "IsDeleted" ).HasValue( false ) );

现在只能从数据库中检索IsDeleted == false的公司

评论更新:

modelBuilder.Entity<Company>()
    .Map( emc => 
    {
        emc.MapInheritedProperties();
        emc.Requires( "IsDeleted" ).HasValue( false );
    } )
    .Ignore( c => c.IsDeleted );

更新:测试代码成功(找到辅助方法here):

[Table("EntityA")]
public partial class EntityA
{
    public int EntityAId { get; set; }
    public string Description { get; set; }


    public virtual EntityB PrimaryEntityB { get; set; }

    public virtual EntityB AlternativeEntityB { get; set; }

    public bool IsDeleted { get; set; }
}

[Table("EntityB")]
public partial class EntityB
{
    public int EntityBId { get; set; }
    public string Description { get; set; }

    [InverseProperty("PrimaryEntityB")]
    public virtual ICollection<EntityA> EntityAsViaPrimary { get; set; }
    [InverseProperty( "AlternativeEntityB" )]
    public virtual ICollection<EntityA> EntityAsViaAlternative { get; set; }
}

public partial class TestEntities : DbContext
{
    public TestEntities()
        : base("TestEntities")
    {
        Database.SetInitializer( new DatabaseInitializer() );
    }

    protected override void OnModelCreating(DbModelBuilder modelBuilder)
    {
        modelBuilder.Entity<EntityA>()
            .Map( emc =>
                {
                    emc.Requires( "IsDeleted" ).HasValue( false );
                } )
                .Ignore( a => a.IsDeleted );
    }

    public override int SaveChanges()
    {
        foreach( var entry in this.ChangeTracker.Entries<EntityA>() )
        {
            if( entry.State == EntityState.Deleted )
            {
                SoftDelete( entry );
            }
        }

        return base.SaveChanges();
    }

    private void SoftDelete( DbEntityEntry entry )
    {
        var entityType = entry.Entity.GetType();

        var tableName = GetTableName( entityType );
        var pkName = GetPrimaryKeyName( entityType );

        var deleteSql = string.Format( "update {0} set IsDeleted = 1 where {1} = @id",
            tableName,
            pkName );

        Database.ExecuteSqlCommand( deleteSql, new SqlParameter( "@id", entry.OriginalValues[ pkName ] ) );

        entry.State = EntityState.Detached;
    }

    private string GetPrimaryKeyName( Type type )
    {
        return GetEntitySet( type ).ElementType.KeyMembers[ 0 ].Name;
    }

    private string GetTableName( Type type )
    {
        EntitySetBase es = GetEntitySet( type );

        return string.Format( "[{0}].[{1}]",
            es.MetadataProperties[ "Schema" ].Value,
            es.MetadataProperties[ "Table" ].Value );
    }
    private EntitySetBase GetEntitySet( Type type )
    {
        ObjectContext octx = ( ( IObjectContextAdapter )this ).ObjectContext;

        string typeName = ObjectContext.GetObjectType( type ).Name;

        var es = octx.MetadataWorkspace
                        .GetItemCollection( DataSpace.SSpace )
                        .GetItems<EntityContainer>()
                        .SelectMany( c => c.BaseEntitySets
                                        .Where( e => e.Name == typeName ) )
                        .FirstOrDefault();

        if( es == null )
            throw new ArgumentException( "Entity type not found in GetTableName", typeName );

        return es;
    }

    public DbSet<EntityA> EntityAs { get; set; }
    public DbSet<EntityB> EntityBs { get; set; }
}

申请代码:

class Program
{
    static void Main(string[] args)
    {
        using( var db = new TestEntities() )
        {
            var a0 = new EntityA()
                {
                    EntityAId = 1,
                    Description = "hi"
                };

            var a1 = new EntityA()
                {
                    EntityAId = 2,
                    Description = "bye"
                };

            db.EntityAs.Add( a0 );
            db.EntityAs.Add( a1 );

            var b = new EntityB()
            {
                EntityBId = 1,
                Description = "Big B"
            };

            a1.PrimaryEntityB = b;

            db.SaveChanges();

            // this prints "1"
            Console.WriteLine( b.EntityAsViaPrimary.Count() );

            db.EntityAs.Remove( a1 );

            db.SaveChanges();

            // this prints "0"
            Console.WriteLine( b.EntityAsViaPrimary.Count() );
        }

        var input = Console.ReadLine();
    }
}

答案 1 :(得分:0)

您可以像这样进行软删除:

  1. 在OnModelCreating中为每个可以软删除的实体添加一个IsDeleted鉴别器
  2. 覆盖SaveChanges并查找要删除的所有条目
  3. 在这些条目上运行SQL以设置IsDeleted鉴别器,然后将其状态设置为“已分离”
  4. 更改任何唯一索引以忽略任何软删除记录
  5. 您可以在此答案中找到工作代码: How to soft delete using Entity Framework Code First

    这个代码被这个博客选中了: http://netpl.blogspot.com/2013/10/soft-delete-pattern-for-entity.html

答案 2 :(得分:0)

四年后,我终于偶然发现了一个完全符合我想要的功能的库。

EntityFramework.DynamicFilter

modelBuilder.Filter("IsDeleted", (ISoftDelete d) => d.IsDeleted, false);

以上代码段可确保不会检索到软删除的项目。这会影响两者直接查询(db.Set<Foo>().ToList())和间接加载的实体db.Set<Foo>().Include(e => e.Bars).ToList()),从而将软删除的实体完全隐藏在视图之外。


出于完整性考虑,我将其与对SaveChanges()的覆盖合并在一起,该覆盖将在提交数据库之前将硬删除转换为软删除。

这意味着开发人员能够安全地使用硬删除逻辑,他们甚至都不会意识到上下文正在使用软删除。他们不需要知道,不需要关心,他们不会忘记以正确的方式实现它,他们永远不需要为IsDeleted标志编写检查。