Little fixes on maintenance

Author: HaikAsatryan

src/SharedKernel/Maintenance/MaintenanceCachePoller.cs

@@ -3,6 +3,8 @@
 
 namespace SharedKernel.Maintenance;
 
+//This is for local cache entity to poll the maintenance mode from distributed cache
+//This should be removed then L1 + L2 cache is implemented in hybrid cache
 internal class MaintenanceCachePoller(HybridCache hybridCache, MaintenanceState state) : BackgroundService
 {
    protected override async Task ExecuteAsync(CancellationToken stoppingToken)

src/SharedKernel/Maintenance/MaintenanceMode.cs

@@ -5,11 +5,4 @@ public enum MaintenanceMode
    Disabled = 0,
    EnabledForClients = 1,
    EnabledForAll = 2
-}
-
-//This is for local cache entity to poll the maintenance mode from distributed cache
-//This should be removed then L1 + L2 cache is implemented in hybrid cache
-
-// This is a local cache entity to hold the maintenance mode in memory
-// This should be removed then L1 + L2 cache is implemented in hybrid cache
-// thread-safe local snapshot
+}

src/SharedKernel/Maintenance/MaintenanceState.cs

@@ -2,6 +2,9 @@
 
 namespace SharedKernel.Maintenance;
 
+// This is a local cache entity to hold the maintenance mode in memory
+// This should be removed then L1 + L2 cache is implemented in hybrid cache
+// thread-safe local snapshot
 public sealed class MaintenanceState(HybridCache cache)
 {
    private const string Key = "maintenance-mode";
@@ -16,7 +19,6 @@ public MaintenanceMode Mode
    // for admin/API to change mode (updates local immediately, then L2)
    public async Task SetModeAsync(MaintenanceMode mode, CancellationToken ct = default)
    {
-      Mode = mode;
       await cache.SetAsync(
          Key,
          new MaintenanceCacheEntity
@@ -30,6 +32,8 @@ await cache.SetAsync(
             Flags = null
          },
          cancellationToken: ct);
+      
+      Mode = mode;
    }
 
    // used by the poller only