自定义审计EF6.2与DbContext不检测修改

时间:2018-03-20 03:00:46

标签: c# entity-framework-6 audit-trail

我已经扩展了DbContext以启用审核,当我进行修改时我遇到了一个问题,因为我用来跟踪报告没有修改的集合对象,而且对于新记录,我没有得到主键:

public partial class FinanceConnection
{
    public int SaveChanges(int userId)
    {
        RecordAudit(userId);
        // AddTimestamps(userId);

        return base.SaveChanges();
    }

    public Task<int> SaveChangesAsync(int userId)
    {
        RecordAudit(userId);
        // AddTimestamps(userId);

        return base.SaveChangesAsync();
    }

    private void RecordAudit(int userId)
    {
        try
        {
            var listLogs = GetLogEntries(((IObjectContextAdapter)this).ObjectContext.ObjectStateManager, userId);

            foreach (var log in listLogs)
            {
                var auditLog = new AUDIT_LOG
                {
                    ID = log.Id,
                    AUDIT_TYPE_ID = log.Action,
                    TABLE_NAME = log.TableName,
                    COLUMN_NAME = log.ColumnName,
                    RECORD_PK = log.PrimaryKey,
                    OLD_VALUE = log.OldValue,
                    NEW_VALUE = log.NewValue,
                    RECORD_STAMP = log.Date,
                    USER_ID = log.UserId
                };

                AUDIT_LOGS.Add(auditLog);
            }
        }
        catch (Exception e)
        {
            Console.WriteLine(e);
        }
    }

    private IEnumerable<CustomLog> GetLogEntries(ObjectStateManager entities, int userId)
    {
        var listLogs = new List<CustomLog>();
        var entries = entities.GetObjectStateEntries(EntityState.Added | EntityState.Modified | EntityState.Deleted);
        var modifiedEntities = this.ChangeTracker.Entries().Where(p => p.State == EntityState.Added || p.State == EntityState.Modified || p.State == EntityState.Deleted).ToList();

        foreach (var entry in entries)
        {
            var tableName = entry.Entity.GetType().Name;
            var pk = GetPrimaryKeys(entry);

            switch (entry.State)
            {
                case EntityState.Added:
                {
                    ...
                    break;
                }
                case EntityState.Modified:
                {
                    ...
                    break;
                }
                case EntityState.Deleted:
                {
                    ...
                    break;
                }
            }
        }

        return listLogs;
    }

    private static string GetPrimaryKeys(ObjectStateEntry entry)
    {
        var pk = string.Empty;
        if (entry.EntityKey == null || entry.EntityKey.EntityKeyValues == null || entry.EntityKey.EntityKeyValues.Length == 0) return "N/A";
        return entry.EntityKey.EntityKeyValues.Aggregate(pk, (current, keyValue) => current + $"{keyValue.Key}={keyValue.Value};");
    }

}

问题:   - 我无法获得新记录的主键。新记录案例有效。   - 我无法得到修改过的案例,因为entities.GetObjectStateEntries(EntityState.Added | EntityState.Modified | EntityState.Deleted)为修改后的记录返回0

0 个答案:

没有答案