Can you convince a DataContext to treat a column as always dirty?

后端 未结 5 737
花落未央
花落未央 2020-12-06 02:39

Is there a way to force LINQ-to-SQL to treat a column as dirty? Globally would suffice....

Basically, I\'ve got a problem with some audit code on a legacy system tha

5条回答
  •  北荒
    北荒 (楼主)
    2020-12-06 03:32

    Based on KristoferA's answer, I ended up with something like below; this is evil and brittle (reflection often is), but may have to suffice for now. The other side of the battle is to change the triggers to behave:

    partial class MyDataContext // or a base-class
    {
        public override void SubmitChanges(System.Data.Linq.ConflictMode failureMode)
        {
            this.MakeUpdatesDirty("UpdatedBy", "Updated_By");
            base.SubmitChanges(failureMode);
        }
    }
    public static class DataContextExtensions
    {
        public static void MakeUpdatesDirty(
            this DataContext dataContext,
            params string[] members)
        {
            if (dataContext == null) throw new ArgumentNullException("dataContext");
            if (members == null) throw new ArgumentNullException("members");
            if (members.Length == 0) return; // nothing to do
            foreach (object instance in dataContext.GetChangeSet().Updates)
            {
                MakeDirty(dataContext, instance, members);
            }
        }
        public static void MakeDirty(
            this DataContext dataContext, object instance ,
            params string[] members)
        {
            if (dataContext == null) throw new ArgumentNullException("dataContext");
            if (instance == null) throw new ArgumentNullException("instance");
            if (members == null) throw new ArgumentNullException("members");
            if (members.Length == 0) return; // nothing to do
            const BindingFlags AllInstance = BindingFlags.Instance | BindingFlags.NonPublic | BindingFlags.Public;
            object commonDataServices = typeof(DataContext)
                .GetField("services", AllInstance)
                .GetValue(dataContext);
            object changeTracker = commonDataServices.GetType()
                .GetProperty("ChangeTracker", AllInstance)
                .GetValue(commonDataServices, null);
            object trackedObject = changeTracker.GetType()
                .GetMethod("GetTrackedObject", AllInstance)
                .Invoke(changeTracker, new object[] { instance });
            var memberCache = trackedObject.GetType()
                .GetField("dirtyMemberCache", AllInstance)
                .GetValue(trackedObject) as BitArray;
    
            var entityType = instance.GetType();
            var metaType = dataContext.Mapping.GetMetaType(entityType);
            for(int i = 0 ; i < members.Length ; i++) {
                var member = entityType.GetMember(members[i], AllInstance);
                if(member != null && member.Length == 1) {
                    var metaMember = metaType.GetDataMember(member[0]);
                    if (metaMember != null)
                    {
                        memberCache.Set(metaMember.Ordinal, true);
                    }
                }
            }
        }
    }
    

提交回复
热议问题