`AddOrReplaceEvent` when unit of work `CompleteAsync`.

framework/src/Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWork.cs

@@ -34,7 +34,11 @@ public class UnitOfWork : IUnitOfWork, ITransientDependency
     public string? ReservationName { get; set; }
 
     protected List<Func<Task>> CompletedHandlers { get; } = new List<Func<Task>>();
+
+    protected List<KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>> DistributedEventWithPredicates { get; } = new List<KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>>();
     protected List<UnitOfWorkEventRecord> DistributedEvents { get; } = new List<UnitOfWorkEventRecord>();
+
+    protected List<KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>> LocalEventWithPredicates { get; } = new List<KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>>();
     protected List<UnitOfWorkEventRecord> LocalEvents { get; } = new List<UnitOfWorkEventRecord>();
 
     public event EventHandler<UnitOfWorkFailedEventArgs> Failed = default!;
@@ -135,24 +139,31 @@ public class UnitOfWork : IUnitOfWork, ITransientDependency
             _isCompleting = true;
             await SaveChangesAsync(cancellationToken);
 
+            DistributedEvents.AddRange(GetEventsRecordsByPredicate(DistributedEventWithPredicates));
+            LocalEvents.AddRange(GetEventsRecordsByPredicate(LocalEventWithPredicates));
+
             while (LocalEvents.Any() || DistributedEvents.Any())
             {
                 if (LocalEvents.Any())
                 {
                     var localEventsToBePublished = LocalEvents.OrderBy(e => e.EventOrder).ToArray();
+                    LocalEventWithPredicates.Clear();
                     LocalEvents.Clear();
                     await UnitOfWorkEventPublisher.PublishLocalEventsAsync(
                         localEventsToBePublished
                     );
+                    LocalEvents.AddRange(GetEventsRecordsByPredicate(LocalEventWithPredicates));
                 }
 
                 if (DistributedEvents.Any())
                 {
                     var distributedEventsToBePublished = DistributedEvents.OrderBy(e => e.EventOrder).ToArray();
+                    DistributedEventWithPredicates.Clear();
                     DistributedEvents.Clear();
                     await UnitOfWorkEventPublisher.PublishDistributedEventsAsync(
                         distributedEventsToBePublished
                     );
+                    DistributedEvents.AddRange(GetEventsRecordsByPredicate(DistributedEventWithPredicates));
                 }
 
                 await SaveChangesAsync(cancellationToken);
@@ -244,38 +255,44 @@ public class UnitOfWork : IUnitOfWork, ITransientDependency
         UnitOfWorkEventRecord eventRecord,
         Predicate<UnitOfWorkEventRecord>? replacementSelector = null)
     {
-        AddOrReplaceEvent(LocalEvents, eventRecord, replacementSelector);
+        LocalEventWithPredicates.Add(new KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>(eventRecord, replacementSelector));
     }
 
     public virtual void AddOrReplaceDistributedEvent(
         UnitOfWorkEventRecord eventRecord,
         Predicate<UnitOfWorkEventRecord>? replacementSelector = null)
     {
-        AddOrReplaceEvent(DistributedEvents, eventRecord, replacementSelector);
+        DistributedEventWithPredicates.Add(new KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>(eventRecord, replacementSelector));
     }
 
-    public virtual void AddOrReplaceEvent(
+    protected virtual List<UnitOfWorkEventRecord> GetEventsRecordsByPredicate(List<KeyValuePair<UnitOfWorkEventRecord, Predicate<UnitOfWorkEventRecord>?>> eventWithPredicates)
-        List<UnitOfWorkEventRecord> eventRecords,
-        UnitOfWorkEventRecord eventRecord,
-        Predicate<UnitOfWorkEventRecord>? replacementSelector = null)
     {
-        if (replacementSelector == null)
+        var eventRecords = new List<UnitOfWorkEventRecord>();
-        {
+        foreach (var eventWithPredicate in eventWithPredicates)
-            eventRecords.Add(eventRecord);
-        }
-        else
         {
-            var foundIndex = eventRecords.FindIndex(replacementSelector);
+            var eventRecord = eventWithPredicate.Key;
-            if (foundIndex < 0)
+            var replacementSelector = eventWithPredicate.Value;
+
+            if (replacementSelector == null)
             {
                 eventRecords.Add(eventRecord);
             }
             else
             {
-                eventRecord.SetOrder(eventRecords[foundIndex].EventOrder);
+                var foundIndex = eventRecords.FindIndex(replacementSelector);
-                eventRecords[foundIndex] = eventRecord;
+                if (foundIndex < 0)
+                {
+                    eventRecords.Add(eventRecord);
+                }
+                else
+                {
+                    eventRecord.SetOrder(eventRecords[foundIndex].EventOrder);
+                    eventRecords[foundIndex] = eventRecord;
+                }
             }
         }
+
+        return eventRecords;
     }
 
     protected virtual async Task OnCompletedAsync()