2014-01-08 48 views
2

我想创建一个保存点使用实体框架的任何更改的审计日志。到目前为止,我有这还算运作良好,使用下面的代码保存到每个字段进行的所有更改:如何从DBEntityEntry获取外键值?

foreach (string propertyName in dbEntry.OriginalValues.PropertyNames) 
{ 
    // For updates, we only want to capture the columns that actually changed 
    if (!object.Equals(dbEntry.OriginalValues.GetValue<object>(propertyName), dbEntry.CurrentValues.GetValue<object>(propertyName))) 
    { 
     result.Add(new AuditLog() 
     { 
      UserID = UserId, 
      EventDateUTC = changeTime, 
      EventType = "M", // Modified 
      TableName = tableName, 
      RecordID = primaryKey.ToString(), 
      ColumnName = propertyName, 
      OriginalValue = dbEntry.OriginalValues.GetValue<object>(propertyName) == null ? null : dbEntry.OriginalValues.GetValue<object>(propertyName).ToString(), 
      NewValue = dbEntry.CurrentValues.GetValue<object>(propertyName) == null ? null : dbEntry.CurrentValues.GetValue<object>(propertyName).ToString() 
     }); 
    } 
} 

我现在面临的问题是如何获得值属于该对象的任何外键。例如:我有一个车辆对象与一系列查找表(如变速箱,模型等)有关系。如果这些值更改,审计表将存储更改的ID,但我想存储实际值。

有没有办法在这种情况下获得外键值?

回答

1

Alrighty ...这是一个古老的问题,但我已经花了最后一个工作,因为我有完全相同的要求。也许还有一个更简单的方法,但这里是我使用的代码:

你的原代码,稍加修改,我的目的(的recordId始终是一个int),并调用新的方法来计算新的价值

foreach (string propertyName in dbEntry.OriginalValues.PropertyNames) 
      { 
       // For updates, we only want to capture the columns that actually changed 
       if (!Equals(dbEntry.OriginalValues.GetValue<object>(propertyName), dbEntry.CurrentValues.GetValue<object>(propertyName))) 
       { 
        var newVal = getNewValueAsString(dbEntry, tableName, propertyName); 

        result.Add(new AuditLog 
           { 
            UserID = currentUser.ID, 
            Timestamp = changeTime, 
            EventType = EventType.Modified, 
            TableName = tableName, 
            RecordID = dbEntry.OriginalValues.GetValue<int>(keyName), 
            ColumnName = propertyName, 
            OriginalValue = dbEntry.OriginalValues.GetValue<object>(propertyName) == null ? null : dbEntry.OriginalValues.GetValue<object>(propertyName).ToString(), 
            NewValue = newVal 
           } 
         ); 
       } 
      } 

称为“IsName”

​​

标志着“名”与IsName属性(注意代码将默认为一个名为“名称”属性,如果它没有找到一个外键模型的属性的新属性)

[Required] 
[IsName] 
public string Name { get; set; } 

而繁重的代码

private string getNewValueAsString(DbEntityEntry dbEntry, string tableName, string propertyName) 
    { 
     var fkVal = getForeignKeyValue(tableName, propertyName, dbEntry.CurrentValues.GetValue<object>(propertyName)); 
     return fkVal != null ? fkVal.ToString() 
          : (dbEntry.CurrentValues.GetValue<object>(propertyName) == null ? null 
           : dbEntry.CurrentValues.GetValue<object>(propertyName).ToString()); 
    } 

    private object getForeignKeyValue(string tableName, string propertyName, object foreignKeyID) 
    { 
     // if this property is part of a foreign key, we need to instead look that up and store the value of the 
     // foreign key 

     // first get all the foreign keys in the system 
     var workspace = ((IObjectContextAdapter)this).ObjectContext.MetadataWorkspace; 
     var items = workspace.GetItems<AssociationType>(DataSpace.CSpace); 
     if (items == null) return null; 
     var fk = items.Where(a => a.IsForeignKey).ToList(); 
     // now we look into the FK attributes and find that the "To Role" is out current table, and the 
     // "To Property" is out current property. The underscore is a bit of an assumption that the foreign 
     // key name built by EF will be ENTITY_BLAH_BLAH 
     var thisFk = fk.Where(x => x.ReferentialConstraints[0].ToRole.Name.StartsWith(tableName + "_")) 
      .FirstOrDefault(x => x.ReferentialConstraints[0].ToProperties[0].Name == propertyName); 
     // if fkname has no results, this is not a foreign key and we are done 
     if (thisFk == null) return null; 

     // Now that we know the foriegn key, we need to lookup the Name value in the other table 

     // find the assembly 
     var assembly = Assembly.GetCallingAssembly(); 
     // build the type for the foreign key entity 
     // e.g. if the current entity is Task, and the property is StatusID, we are 
     // getting the "TaskStatus" type with reflection 
     // "User" class is an object in the Models namespace - you could just hardcode the string if you want 
     var foreignKeyType = assembly.GetType(typeof(User).Namespace + "." + 
        thisFk.ReferentialConstraints[0].FromRole.GetEntityType().Name); 

     // get the DbSet, same as: "(new DBContext()).EntityName" 
     var fkSet = Set(foreignKeyType); 
     // and find the row in that table 
     var fkItem = fkSet.Find(foreignKeyID); 

     // find the first column marked with the "IsName" attribute, otherwise default to "Name" 
     var nameColProperty = foreignKeyType.GetProperties() 
      .FirstOrDefault(p => p.GetCustomAttributes(typeof(IsNameAttribute), false).Any()); 
     string nameCol = "Name"; 
     if (nameColProperty != null) nameCol = nameColProperty.Name; 
     var nameColProperty2 = fkItem.GetType().GetProperty(nameCol); 
     if (nameColProperty2 == null) return null; 

     // get the value 
     var fkValue = nameColProperty2.GetValue(fkItem, null); 

     // and now, my brain hurts 
     return fkValue; 
    } 
0

该解决方案是基于@JamesR's answer

我的目标是让代码更通用,因此它可以用于连接到不同表的多个外键。

改进值得注意的是:

  • 我感动的是得到propertyName的foreach循环之外的外键列表中的代码。由于FK列表不会根据特定属性而改变,因此没有理由每次都检索新列表。如果系统中有许多FK,这可能需要一段时间,所以您不希望不必要地重复该过程。

  • 而不是硬编码的一类特殊类型的像GetType(typeof(User),我使用检索从FK外键表名:

    string lookUpTableName = thisFk.ReferentialConstraints[0].FromRole.Name;  
    

    然后,虽然引用的FK属性名称通常是ID,因为它可以改变,我检索到的FK属性的名称,以及:

    string lookUpPropertyName = thisFk.ReferentialConstraints[0].FromProperties[0].Name; 
    

    然后我用ObjectContext.ExecuteStoreQuery动态插上表和列名和检索外键文本值。

  • 如果某个属性是FK,我将为原始的新值获取FK文本值。


完整代码:

首先,获取系统中所有外键的列表。

IObjectContextAdapter contextAdapter = ((IObjectContextAdapter)this); 
MetadataWorkspace workspace = contextAdapter.ObjectContext.MetadataWorkspace; 
var items = workspace.GetItems<AssociationType>(DataSpace.CSpace); 

List<AssociationType> FKList = items == null ? null 
    : items.Where(a => a.IsForeignKey).ToList(); 

然后,循环访问属性列表,并在存在FK时使用外键值替换原始值和当前值。

foreach (string propertyName in entry.OriginalValues.PropertyNames) 
{ 
    var original = entry.OriginalValues.GetValue<object>(propertyName); 
    var current = entry.CurrentValues.GetValue<object>(propertyName); 

    if (FKList != null) 
    { 
     GetPossibleForeignKeyValues(tableName, propertyName, ref original, ref current, 
      FKList, contextAdapter); 
    } 

    if ((original == null && current != null) || 
     (original != null && !original.Equals(current))) 
    { 
     result.Add(new AuditLog() 
     { 
      UserID = UserId, 
      EventDateUTC = changeTime, 
      EventType = "M", // Modified 
      TableName = tableName, 
      RecordID = primaryKey.ToString(), 
      ColumnName = propertyName, 
      OriginalValue = original != null ? original.ToString() : "NULL", 
      NewValue = current != null ? current.ToString() : "NULL" 
     }); 
    } 
} 

这里是真正的外键发现代码:

private void GetPossibleForeignKeyValues(string tableName, string propertyName, 
    ref object originalFKValue, ref object newFKValue, 
    List<AssociationType> FKList, IObjectContextAdapter contextAdapter) 
{ 
    // If this property is part of a foreign key, look up and set the FKValue to the text 
    // value of the foreign key. Otherwise, just leave the FKValue alone. 

    // Look into the FK attributes and find that the "To Role" is out current table, 
    // and the "To Property" is out current property. 
    AssociationType thisFk = FKList.FirstOrDefault(x => 
     tableName.Contains(x.ReferentialConstraints[0].ToRole.Name) 
     && propertyName.Contains(x.ReferentialConstraints[0].ToProperties[0].Name)); 

    // If fkname has no results, this is not a foreign key and we are done. 
    if (thisFk != null) 
    { 
     // Now that we know the foriegn key, look up the Name value in the other table. 
     string lookUpTableName = thisFk.ReferentialConstraints[0].FromRole.Name; 
     string lookUpPropertyName = thisFk.ReferentialConstraints[0].FromProperties[0].Name; 

     //Assuming the FK column name is "Name". 
     //Use the idea in @JamesR's solution or some sort of LookUp table if it is not. 
     string commandText = BuildCommandText("Name", lookUpTableName, lookUpPropertyName); 

     originalFKValue = contextAdapter.ObjectContext 
      .ExecuteStoreQuery<string>(commandText, new SqlParameter("FKID", originalFKValue)) 
      .FirstOrDefault() ?? originalFKValue; 

     newFKValue = contextAdapter.ObjectContext 
      .ExecuteStoreQuery<string>(commandText, new SqlParameter("FKID", newFKValue)) 
      .FirstOrDefault() ?? originalFKValue; 

    } 
} 

这是我用来建立SQL的CommandText的方法:

private string BuildCommandText(string columnName, string lookUpTableName, 
    string lookUpPropertyName)  
{ 
    StringBuilder builder = new StringBuilder(); 

    builder.Append("SELECT "); 
    builder.Append(columnName); 
    builder.Append(" FROM "); 
    builder.Append(lookUpTableName); 
    builder.Append(" WHERE "); 
    builder.Append(lookUpPropertyName); 
    builder.Append(" = @FKID"); 

    //The result query will look something like: 
    //SELECT ColumnName FROM TableName WHERE PropertyName = @FKID 

    return builder.ToString(); 
}