up

src/Zastava/StellaOps.Zastava.Observer/Worker/RuntimeEventDispatchService.cs

@@ -13,6 +13,7 @@ internal sealed class RuntimeEventDispatchService : BackgroundService
 {
     private readonly IRuntimeEventBuffer buffer;
     private readonly IRuntimeEventsClient eventsClient;
+    private readonly IRuntimeFactsClient runtimeFactsClient;
     private readonly IOptionsMonitor<ZastavaObserverOptions> observerOptions;
     private readonly TimeProvider timeProvider;
     private readonly ILogger<RuntimeEventDispatchService> logger;
@@ -20,12 +21,14 @@ internal sealed class RuntimeEventDispatchService : BackgroundService
     public RuntimeEventDispatchService(
         IRuntimeEventBuffer buffer,
         IRuntimeEventsClient eventsClient,
+        IRuntimeFactsClient runtimeFactsClient,
         IOptionsMonitor<ZastavaObserverOptions> observerOptions,
         TimeProvider timeProvider,
         ILogger<RuntimeEventDispatchService> logger)
     {
         this.buffer = buffer ?? throw new ArgumentNullException(nameof(buffer));
         this.eventsClient = eventsClient ?? throw new ArgumentNullException(nameof(eventsClient));
+        this.runtimeFactsClient = runtimeFactsClient ?? throw new ArgumentNullException(nameof(runtimeFactsClient));
         this.observerOptions = observerOptions ?? throw new ArgumentNullException(nameof(observerOptions));
         this.timeProvider = timeProvider ?? throw new ArgumentNullException(nameof(timeProvider));
         this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
@@ -116,14 +119,17 @@ internal sealed class RuntimeEventDispatchService : BackgroundService
             return;
         }
 
-        var request = new RuntimeEventsIngestRequest
-        {
-            BatchId = $"obs-{timeProvider.GetUtcNow():yyyyMMddTHHmmssfff}-{Guid.NewGuid():N}",
-            Events = batch.Select(item => item.Envelope).ToArray()
-        };
-
         try
         {
+            var envelopes = batch.Select(item => item.Envelope).ToArray();
+            var factsPublished = await TryPublishRuntimeFactsAsync(envelopes, cancellationToken).ConfigureAwait(false);
+
+            var request = new RuntimeEventsIngestRequest
+            {
+                BatchId = $"obs-{timeProvider.GetUtcNow():yyyyMMddTHHmmssfff}-{Guid.NewGuid():N}",
+                Events = envelopes
+            };
+
             var result = await eventsClient.PublishAsync(request, cancellationToken).ConfigureAwait(false);
             if (result.Success)
             {
@@ -132,10 +138,11 @@ internal sealed class RuntimeEventDispatchService : BackgroundService
                     await item.CompleteAsync().ConfigureAwait(false);
                 }
 
-                logger.LogInformation("Runtime events batch published (batchId={BatchId}, accepted={Accepted}, duplicates={Duplicates}).",
+                logger.LogInformation("Runtime events batch published (batchId={BatchId}, accepted={Accepted}, duplicates={Duplicates}, runtimeFacts={FactsPublished}).",
                     request.BatchId,
                     result.Accepted,
-                    result.Duplicates);
+                    result.Duplicates,
+                    factsPublished);
             }
             else if (result.RateLimited)
             {
@@ -166,6 +173,38 @@ internal sealed class RuntimeEventDispatchService : BackgroundService
         }
     }
 
+    private async Task<bool> TryPublishRuntimeFactsAsync(RuntimeEventEnvelope[] envelopes, CancellationToken cancellationToken)
+    {
+        if (envelopes.Length == 0)
+        {
+            return false;
+        }
+
+        var options = observerOptions.CurrentValue.Reachability;
+        var request = RuntimeFactsBuilder.Build(envelopes, options);
+        if (request is null)
+        {
+            return false;
+        }
+
+        try
+        {
+            await runtimeFactsClient.PublishAsync(request, cancellationToken).ConfigureAwait(false);
+            logger.LogDebug("Published {Count} runtime facts (callgraphId={CallgraphId}).", request.Events.Count, request.CallgraphId);
+            return true;
+        }
+        catch (RuntimeFactsException ex) when (!cancellationToken.IsCancellationRequested)
+        {
+            logger.LogWarning(ex, "Runtime facts publish failed; batch will be retried.");
+            throw;
+        }
+        catch (Exception ex) when (!cancellationToken.IsCancellationRequested)
+        {
+            logger.LogWarning(ex, "Runtime facts publish encountered an unexpected error; batch will be retried.");
+            throw;
+        }
+    }
+
     private async Task RequeueBatchAsync(IEnumerable<RuntimeEventBufferItem> batch, CancellationToken cancellationToken)
     {
         foreach (var item in batch)