save progress

src/VexLens/StellaOps.VexLens/Orchestration/IConsensusJobService.cs

@@ -160,6 +160,7 @@ public sealed class ConsensusJobService : IConsensusJobService
     private readonly IVexConsensusEngine _consensusEngine;
     private readonly IConsensusProjectionStore _projectionStore;
     private readonly IConsensusExportService _exportService;
+    private readonly TimeProvider _timeProvider;
 
     private const string SchemaVersion = "1.0.0";
 
@@ -172,11 +173,13 @@ public sealed class ConsensusJobService : IConsensusJobService
     public ConsensusJobService(
         IVexConsensusEngine consensusEngine,
         IConsensusProjectionStore projectionStore,
-        IConsensusExportService exportService)
+        IConsensusExportService exportService,
+        TimeProvider? timeProvider = null)
     {
         _consensusEngine = consensusEngine;
         _projectionStore = projectionStore;
         _exportService = exportService;
+        _timeProvider = timeProvider ?? TimeProvider.System;
     }
 
     public ConsensusJobRequest CreateComputeJob(
@@ -299,7 +302,7 @@ public sealed class ConsensusJobService : IConsensusJobService
             JobType: ConsensusJobTypes.SnapshotCreate,
             TenantId: request.TenantId,
             Priority: ConsensusJobTypes.GetDefaultPriority(ConsensusJobTypes.SnapshotCreate),
-            IdempotencyKey: $"snapshot:{requestHash}:{DateTimeOffset.UtcNow:yyyyMMddHHmm}",
+            IdempotencyKey: $"snapshot:{requestHash}:{_timeProvider.GetUtcNow():yyyyMMddHHmm}",
             Payload: JsonSerializer.Serialize(payload, JsonOptions));
     }
 
@@ -307,7 +310,7 @@ public sealed class ConsensusJobService : IConsensusJobService
         ConsensusJobRequest request,
         CancellationToken cancellationToken = default)
     {
-        var startTime = DateTimeOffset.UtcNow;
+        var startTime = _timeProvider.GetUtcNow();
 
         try
         {
@@ -350,7 +353,7 @@ public sealed class ConsensusJobService : IConsensusJobService
         ConsensusJobRequest request,
         CancellationToken cancellationToken)
     {
-        var startTime = DateTimeOffset.UtcNow;
+        var startTime = _timeProvider.GetUtcNow();
         var payload = JsonSerializer.Deserialize<ComputePayload>(request.Payload, JsonOptions)
             ?? throw new InvalidOperationException("Invalid compute payload");
 
@@ -363,7 +366,7 @@ public sealed class ConsensusJobService : IConsensusJobService
             JobType: request.JobType,
             ItemsProcessed: 1,
             ItemsFailed: 0,
-            Duration: DateTimeOffset.UtcNow - startTime,
+            Duration: _timeProvider.GetUtcNow() - startTime,
             ResultPayload: JsonSerializer.Serialize(new
             {
                 vulnerabilityId = payload.VulnerabilityId,
@@ -377,7 +380,7 @@ public sealed class ConsensusJobService : IConsensusJobService
         ConsensusJobRequest request,
         CancellationToken cancellationToken)
     {
-        var startTime = DateTimeOffset.UtcNow;
+        var startTime = _timeProvider.GetUtcNow();
         var payload = JsonSerializer.Deserialize<BatchComputePayload>(request.Payload, JsonOptions)
             ?? throw new InvalidOperationException("Invalid batch compute payload");
 
@@ -389,7 +392,7 @@ public sealed class ConsensusJobService : IConsensusJobService
             JobType: request.JobType,
             ItemsProcessed: itemCount,
             ItemsFailed: 0,
-            Duration: DateTimeOffset.UtcNow - startTime,
+            Duration: _timeProvider.GetUtcNow() - startTime,
             ResultPayload: JsonSerializer.Serialize(new { processedCount = itemCount }, JsonOptions),
             ErrorMessage: null);
     }
@@ -398,7 +401,7 @@ public sealed class ConsensusJobService : IConsensusJobService
         ConsensusJobRequest request,
         CancellationToken cancellationToken)
     {
-        var startTime = DateTimeOffset.UtcNow;
+        var startTime = _timeProvider.GetUtcNow();
 
         // Create snapshot using export service
         var snapshotRequest = ConsensusExportExtensions.FullExportRequest(request.TenantId);
@@ -409,7 +412,7 @@ public sealed class ConsensusJobService : IConsensusJobService
             JobType: request.JobType,
             ItemsProcessed: snapshot.Projections.Count,
             ItemsFailed: 0,
-            Duration: DateTimeOffset.UtcNow - startTime,
+            Duration: _timeProvider.GetUtcNow() - startTime,
             ResultPayload: JsonSerializer.Serialize(new
             {
                 snapshotId = snapshot.SnapshotId,
@@ -419,14 +422,14 @@ public sealed class ConsensusJobService : IConsensusJobService
             ErrorMessage: null);
     }
 
-    private static ConsensusJobResult CreateFailedResult(string jobType, DateTimeOffset startTime, string error)
+    private ConsensusJobResult CreateFailedResult(string jobType, DateTimeOffset startTime, string error)
     {
         return new ConsensusJobResult(
             Success: false,
             JobType: jobType,
             ItemsProcessed: 0,
             ItemsFailed: 1,
-            Duration: DateTimeOffset.UtcNow - startTime,
+            Duration: _timeProvider.GetUtcNow() - startTime,
             ResultPayload: null,
             ErrorMessage: error);
     }