我已经扩展了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