up

src/RiskEngine/StellaOps.RiskEngine/StellaOps.RiskEngine.WebService/Program.cs

@@ -1,3 +1,5 @@
+using Microsoft.AspNetCore.Mvc;
+using System.Linq;
 using StellaOps.RiskEngine.Core.Contracts;
 using StellaOps.RiskEngine.Core.Providers;
 using StellaOps.RiskEngine.Core.Services;
@@ -32,51 +34,92 @@ app.MapGet("/risk-scores/providers", (IRiskScoreProviderRegistry registry) =>
 
 app.MapPost("/risk-scores/jobs", async (
     ScoreRequest request,
-    RiskScoreQueue queue,
-    IRiskScoreProviderRegistry registry,
-    IRiskScoreResultStore store,
-    TimeProvider timeProvider,
+    [FromServices] RiskScoreQueue queue,
+    [FromServices] IRiskScoreProviderRegistry registry,
+    [FromServices] IRiskScoreResultStore store,
     CancellationToken ct) =>
 {
-    var job = new RiskScoreJob(Guid.NewGuid(), request);
-    await queue.EnqueueAsync(job.Request, ct).ConfigureAwait(false);
-    var worker = new RiskScoreWorker(queue, registry, store, timeProvider);
+    var normalized = new ScoreRequest(
+        request.Provider,
+        request.Subject,
+        request.Signals ?? new Dictionary<string, double>());
+
+    var jobId = await queue.EnqueueWithIdAsync(normalized, ct).ConfigureAwait(false);
+    var worker = new RiskScoreWorker(queue, registry, store, TimeProvider.System);
     var result = await worker.ProcessNextAsync(ct).ConfigureAwait(false);
-    return Results.Accepted($"/risk-scores/jobs/{job.JobId}", new { jobId = job.JobId, result });
+    return Results.Accepted($"/risk-scores/jobs/{jobId}", new { jobId, result });
 });
 
-app.MapGet("/risk-scores/jobs/{jobId:guid}", (Guid jobId, InMemoryRiskScoreResultStore store) =>
+app.MapGet("/risk-scores/jobs/{jobId:guid}", (
+    Guid jobId,
+    [FromServices] IRiskScoreResultStore store) =>
     store.TryGet(jobId, out var result)
         ? Results.Ok(result)
         : Results.NotFound());
 
 app.MapPost("/risk-scores/simulations", async (
     IReadOnlyCollection<ScoreRequest> requests,
-    IRiskScoreProviderRegistry registry,
-    TimeProvider timeProvider,
+    [FromServices] IRiskScoreProviderRegistry registry,
     CancellationToken ct) =>
+{
+    var results = await EvaluateAsync(requests, registry, ct).ConfigureAwait(false);
+    return Results.Ok(new { results });
+});
+
+app.MapPost("/risk-scores/simulations/summary", async (
+    IReadOnlyCollection<ScoreRequest> requests,
+    [FromServices] IRiskScoreProviderRegistry registry,
+    CancellationToken ct) =>
+{
+    var results = await EvaluateAsync(requests, registry, ct).ConfigureAwait(false);
+
+    var scores = results.Select(r => r.Score).ToArray();
+    var summary = new
+    {
+        averageScore = scores.Length == 0 ? 0d : scores.Average(),
+        minScore = scores.Length == 0 ? 0d : scores.Min(),
+        maxScore = scores.Length == 0 ? 0d : scores.Max(),
+        topMovers = results
+            .OrderByDescending(r => r.Score)
+            .ThenBy(r => r.Subject, StringComparer.Ordinal)
+            .Take(3)
+            .ToArray()
+    };
+
+    return Results.Ok(new { summary, results });
+});
+
+app.Run();
+
+static async Task<List<RiskScoreResult>> EvaluateAsync(
+    IReadOnlyCollection<ScoreRequest> requests,
+    IRiskScoreProviderRegistry registry,
+    CancellationToken ct)
 {
     var results = new List<RiskScoreResult>(requests.Count);
     foreach (var req in requests)
     {
-        if (!registry.TryGet(req.Provider, out var provider))
+        var normalized = new ScoreRequest(
+            req.Provider,
+            req.Subject,
+            req.Signals ?? new Dictionary<string, double>());
+
+        if (!registry.TryGet(normalized.Provider, out var provider))
         {
-            results.Add(new RiskScoreResult(Guid.NewGuid(), req.Provider, req.Subject, 0d, false, "Provider not registered", req.Signals, timeProvider.GetUtcNow()));
+            results.Add(new RiskScoreResult(Guid.NewGuid(), normalized.Provider, normalized.Subject, 0d, false, "Provider not registered", normalized.Signals, TimeProvider.System.GetUtcNow()));
             continue;
         }
 
         try
         {
-            var score = await provider.ScoreAsync(req, ct).ConfigureAwait(false);
-            results.Add(new RiskScoreResult(Guid.NewGuid(), req.Provider, req.Subject, score, true, null, req.Signals, timeProvider.GetUtcNow()));
+            var score = await provider.ScoreAsync(normalized, ct).ConfigureAwait(false);
+            results.Add(new RiskScoreResult(Guid.NewGuid(), normalized.Provider, normalized.Subject, score, true, null, normalized.Signals, TimeProvider.System.GetUtcNow()));
         }
         catch (Exception ex)
         {
-            results.Add(new RiskScoreResult(Guid.NewGuid(), req.Provider, req.Subject, 0d, false, ex.Message, req.Signals, timeProvider.GetUtcNow()));
+            results.Add(new RiskScoreResult(Guid.NewGuid(), normalized.Provider, normalized.Subject, 0d, false, ex.Message, normalized.Signals, TimeProvider.System.GetUtcNow()));
         }
     }
 
-    return Results.Ok(new { results });
-});
-
-app.Run();
+    return results;
+}