Update ChangeCollector.cs

loresoft · Aug 23, 2024 · c8830ab · c8830ab
1 parent 7b9b544
commit c8830ab
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/MediatR.CommandQuery.Audit/ChangeCollector.cs b/src/MediatR.CommandQuery.Audit/ChangeCollector.cs
@@ -70,9 +70,11 @@ public IReadOnlyList<AuditRecord<TKey>> CollectGroupChanges<TProperty>(IEnumerab
 
         foreach (var group in entities.GroupBy(groupSelector))
         {
+            var key = group.Key?.ToString();
+
             try
             {
-                LogCollectingChanges(_logger, entityName, group.Key);
+                LogCollectingChanges(_logger, entityName, key);
 
                 var groupList = group
                     .OrderBy(p => p.PeriodEnd)
@@ -83,7 +85,7 @@ public IReadOnlyList<AuditRecord<TKey>> CollectGroupChanges<TProperty>(IEnumerab
             }
             catch (Exception ex)
             {
-                LogCollectingError(_logger, entityName, group.Key, ex.Message, ex);
+                LogCollectingError(_logger, entityName, key, ex.Message, ex);
 
                 throw;
             }
@@ -178,9 +180,9 @@ public IEnumerable<AuditRecord<TKey>> CollectChanges(IEnumerable<TEntity> entiti
 
 
     [LoggerMessage(1, LogLevel.Debug, "Collecting changes for {EntityName} with key {EntityKey} ...")]
-    static partial void LogCollectingChanges(ILogger logger, string entityName, object entityKey);
+    static partial void LogCollectingChanges(ILogger logger, string entityName, string? entityKey);
 
     [LoggerMessage(2, LogLevel.Error, "Error collecting changes for {EntityName} with key {EntityKey}: {ErrorMessage}")]
-    static partial void LogCollectingError(ILogger logger, string entityName, object entityKey, string errorMessage, Exception? exception);
+    static partial void LogCollectingError(ILogger logger, string entityName, string? entityKey, string errorMessage, Exception? exception);
 
 }