private bool WriteEntity(AuditEntryState state)
{
if (state.EntityType == null)
{
return(false);
}
WriteProperties(state);
WriteRelationships(state);
return(true);
}
private static bool IsLoaded(AuditEntryState state, NavigationProperty navigationProperty, IMemberAccessor accessor)
{
var relationshipManager = state.ObjectStateEntry.RelationshipManager;
var getEntityReference = _relatedAccessor.Value.MakeGenericMethod(accessor.MemberType);
var parameters = new[]
{
navigationProperty.RelationshipType.FullName,
navigationProperty.ToEndMember.Name
};
var entityReference = getEntityReference.Invoke(relationshipManager, parameters) as EntityReference;
return(entityReference != null && entityReference.IsLoaded);
}
public AuditLog UpdateLog(AuditLog auditLog)
{
if (auditLog == null)
{
throw new ArgumentNullException("auditLog");
}
// must call to make sure changes are detected
ObjectContext.DetectChanges();
var entityState = EntityState.Modified;
entityState = entityState | EntityState.Added;
entityState = entityState | EntityState.Deleted;
IEnumerable <ObjectStateEntry> changes = ObjectContext
.ObjectStateManager
.GetObjectStateEntries(entityState);
foreach (ObjectStateEntry objectStateEntry in changes)
{
if (objectStateEntry.Entity == null)
{
continue;
}
Type entityType = objectStateEntry.Entity.GetType();
entityType = ObjectContext.GetObjectType(entityType);
if (!configuration.IsAuditable(entityType))
{
continue;
}
var state = new AuditEntryState(objectStateEntry)
{
AuditLog = auditLog,
ObjectContext = ObjectContext,
};
if (WriteEntity(state))
{
auditLog.Entities.Add(state.AuditEntity);
}
}
return(auditLog);
}
//private static string GetForeignKey(NavigationProperty navigationProperty)
//{
// var association = navigationProperty.RelationshipType as AssociationType;
// if (association == null)
// return null;
// // only support first constraint
// var referentialConstraint = association.ReferentialConstraints.FirstOrDefault();
// if (referentialConstraint == null)
// return null;
// var toProperties = referentialConstraint
// .ToProperties
// .Select(p => p.Name)
// .ToArray();
// return string.Join(",", toProperties);
//}
private static object GetDisplayValue(AuditEntryState state, NavigationProperty navigationProperty, IMemberAccessor displayMember, DbDataRecord values)
{
if (values == null)
{
return(null);
}
var association = navigationProperty.RelationshipType as AssociationType;
if (association == null)
{
return(null);
}
// only support first constraint
var referentialConstraint = association.ReferentialConstraints.FirstOrDefault();
if (referentialConstraint == null)
{
return(null);
}
var toProperties = referentialConstraint
.ToProperties
.Select(p => p.Name)
.ToList();
var fromProperties = referentialConstraint
.FromProperties
.Select(p => p.Name)
.ToList();
// make sure key columns match
if (fromProperties.Count != toProperties.Count)
{
return(null);
}
var edmType = referentialConstraint
.FromProperties
.Select(p => p.DeclaringType)
.FirstOrDefault();
if (edmType == null)
{
return(null);
}
var entitySet = GetEntitySet(state.ObjectContext, edmType.FullName);
var sql = new StringBuilder();
sql.Append("SELECT VALUE t.")
.Append(displayMember.Name)
.Append(" FROM ")
.Append(entitySet.Name)
.Append(" as t")
.Append(" WHERE ");
var parameters = new List <ObjectParameter>();
for (int index = 0; index < fromProperties.Count; index++)
{
if (index > 0)
{
sql.Append(" AND ");
}
string fromProperty = fromProperties[index];
string toProperty = toProperties[index];
var value = GetValue(values, toProperty);
var name = "@" + fromProperty;
sql.Append(" t.").Append(fromProperty);
if (value != null)
{
sql.Append(" == ").Append(name);
parameters.Add(new ObjectParameter(fromProperty, value));
}
else
{
sql.Append(" is null");
}
}
var q = state.ObjectContext.CreateQuery <object>(
sql.ToString(),
parameters.ToArray());
return(q.FirstOrDefault());
}
private void WriteRelationships(AuditEntryState state)
{
var properties = state.EntityType.NavigationProperties;
if (properties.Count == 0)
{
return;
}
var modifiedMembers = state.ObjectStateEntry
.GetModifiedProperties()
.ToList();
var type = state.ObjectType;
var currentValues = state.IsDeleted
? state.ObjectStateEntry.OriginalValues
: state.ObjectStateEntry.CurrentValues;
var originalValues = state.IsModified
? state.ObjectStateEntry.OriginalValues
: null;
foreach (NavigationProperty navigationProperty in properties)
{
if (navigationProperty.ToEndMember.RelationshipMultiplicity == RelationshipMultiplicity.Many ||
navigationProperty.FromEndMember.RelationshipMultiplicity != RelationshipMultiplicity.Many)
{
continue;
}
string name = navigationProperty.Name;
if (configuration.IsNotAudited(type, name))
{
continue;
}
var accessor = state.EntityAccessor.Find(name);
IMemberAccessor displayMember = configuration.GetDisplayMember(accessor.MemberType);
if (displayMember == null)
{
continue; // no display property, skip
}
bool isModified = IsModifed(navigationProperty, modifiedMembers);
if (state.IsModified && !isModified)
{
continue; // this means the property was not changed, skip it
}
bool isLoaded = IsLoaded(state, navigationProperty, accessor);
if (!isLoaded)
{
continue;
}
var auditProperty = new AuditProperty();
try
{
auditProperty.Name = name;
auditProperty.Type = accessor.MemberType.FullName;
auditProperty.IsRelationship = true;
//auditProperty.ForeignKey = GetForeignKey(navigationProperty);
object currentValue = null;
if (isLoaded)
{
// get value directly from instance to save db call
object valueInstance = accessor.GetValue(state.Entity);
if (valueInstance != null)
{
currentValue = displayMember.GetValue(valueInstance);
}
}
else
{
// get value from db
currentValue = GetDisplayValue(state, navigationProperty, displayMember, currentValues);
}
// format
//currentValue = FormatValue(state, name, currentValue);
if (!state.IsModified && currentValue == null)
{
continue; // skip null value
}
switch (state.AuditEntity.Action)
{
case AduitAction.Added:
auditProperty.Current = currentValue;
break;
case AduitAction.Modified:
auditProperty.Current = currentValue ?? _nullText;
object originalValue = GetDisplayValue(state, navigationProperty, displayMember, originalValues);
//originalValue = FormatValue(state, name, originalValue);
auditProperty.Original = originalValue;
break;
case AduitAction.Deleted:
auditProperty.Original = currentValue;
break;
}
}
catch (Exception ex)
{
Trace.TraceError(ex.Message);
if (state.IsDeleted)
{
auditProperty.Original = _errorText;
}
else
{
auditProperty.Current = _errorText;
}
}
state.AuditEntity.Properties.Add(auditProperty);
}
}
private void WriteProperties(AuditEntryState state)
{
var properties = state.EntityType.Properties;
if (properties == null)
{
return;
}
var modifiedMembers = state.ObjectStateEntry
.GetModifiedProperties()
.ToList();
var type = state.ObjectType;
var currentValues = state.IsDeleted
? state.ObjectStateEntry.OriginalValues
: state.ObjectStateEntry.CurrentValues;
var originalValues = state.IsModified
? state.ObjectStateEntry.OriginalValues
: null;
foreach (EdmProperty edmProperty in properties)
{
string name = edmProperty.Name;
if (configuration.IsNotAudited(type, name))
{
continue;
}
bool isModified = modifiedMembers.Any(m => m == name);
if (state.IsModified && !isModified)
{
continue; // this means the property was not changed, skip it
}
var auditProperty = new AuditProperty();
try
{
auditProperty.Name = name;
auditProperty.Type = GetType(edmProperty);
var currentValue = GetValue(currentValues, name);
//currentValue = FormatValue(state, name, currentValue);
if (!state.IsModified && currentValue == null)
{
continue; // ignore null properties?
}
switch (state.AuditEntity.Action)
{
case AduitAction.Added:
auditProperty.Current = currentValue;
break;
case AduitAction.Modified:
auditProperty.Current = currentValue;
if (originalValues != null)
{
object originalValue = GetValue(originalValues, edmProperty.Name);
//originalValue = FormatValue(state, name, originalValue);
auditProperty.Original = originalValue;
}
break;
case AduitAction.Deleted:
auditProperty.Original = currentValue;
break;
}
}
catch (Exception ex)
{
Trace.TraceError(ex.Message);
if (state.IsDeleted)
{
auditProperty.Original = _errorText;
}
else
{
auditProperty.Current = _errorText;
}
}
state.AuditEntity.Properties.Add(auditProperty);
} // foreach property
}
