add state to cancellable events

KevinJump · Nov 27, 2024 · d535133 · d535133
1 parent 00f7c59
commit d535133
Showing 1 changed file with 25 additions and 11 deletions.
diff --git a/uSync.BackOffice/Notifications/SyncScopedNotificationPublisher.cs b/uSync.BackOffice/Notifications/SyncScopedNotificationPublisher.cs
@@ -3,20 +3,18 @@
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
-
 using Microsoft.Extensions.Logging;
-
 using Umbraco.Cms.Core.Events;
 using Umbraco.Cms.Core.Notifications;
 using Umbraco.Cms.Infrastructure.HostedServices;
-
 using uSync.BackOffice.Configuration;
 using uSync.BackOffice.Services;
 using uSync.BackOffice.SyncHandlers;
 
 namespace uSync.BackOffice.Notifications;
+
 internal class SyncScopedNotificationPublisher
-    : ScopedNotificationPublisher<INotificationHandler>
+    : ScopedNotificationPublisher<INotificationHandler>, IScopedNotificationPublisher
 {
     private readonly ILogger<SyncScopedNotificationPublisher> _logger;
     private readonly IEventAggregator _eventAggregator;
@@ -43,6 +41,18 @@ public SyncScopedNotificationPublisher(
         _uSyncEventService = uSyncEventService;
     }
 
+    bool IScopedNotificationPublisher.PublishCancelable(ICancelableNotification notification)
+    {
+        SetSingleNotificationState(notification);
+        return base.PublishCancelable(notification);
+    }
+
+    Task<bool> IScopedNotificationPublisher.PublishCancelableAsync(ICancelableNotification notification)
+    {
+        SetSingleNotificationState(notification);
+        return base.PublishCancelableAsync(notification);
+    }
+
     protected override void PublishScopedNotifications(IList<INotification> notifications)
     {
         if (notifications.Count == 0) return;
@@ -71,7 +81,6 @@ protected override void PublishScopedNotifications(IList<INotification> notifica
                             return Task.CompletedTask;
                         }
                     });
-
             }
             else
             {
@@ -91,11 +100,16 @@ private void SetNotificationStates(IList<INotification> notifications)
     {
         foreach (var notification in notifications)
         {
-            if (notification is StatefulNotification stateful)
-            {
-                stateful.State[uSync.EventStateKey] = true;
-                stateful.State[uSync.EventPausedKey] = _uSyncEventService.IsPaused;
-            }
+            SetSingleNotificationState(notification);
+        }
+    }
+
+    private void SetSingleNotificationState(INotification notification)
+    {
+        if (notification is StatefulNotification stateful)
+        {
+            stateful.State[uSync.EventStateKey] = true;
+            stateful.State[uSync.EventPausedKey] = _uSyncEventService.IsPaused;
         }
     }
-}
+}