Add get quote sse endpouint

Author: PurpleTheBest

TrainStation/Endpoints/StationEndpoints.cs

@@ -1,4 +1,5 @@
 using Microsoft.AspNetCore.Mvc;
+using System.Text;
 using Train.Station.API.Models;
 using Train.Station.API.Services;
 using Train.Station.Client;
@@ -17,8 +18,8 @@ public static RouteGroupBuilder MapEndpoints(this RouteGroupBuilder group)
         group.MapGet("/routes", GetAllRoutesAsync)
             .Produces<IEnumerable<RouteDto>>();
 
-        //group.MapGet("/quote", GetQuoteAsync)
-        //    .Produces<ApiResponseQuoteDto>();
+        group.MapGet("/quote-sse", GetQuoteAsync)
+            .Produces(statusCode: 200, contentType: "text/event-stream");
 
         //group.MapGet("/{solver}/swaps", GetAllSwapsAsync)
         //    .Produces<ApiResponseSwapDto>();
@@ -103,34 +104,68 @@ private static async Task<IResult> GetAllRoutesAsync(
         return Results.Ok(routes);
     }
 
-    //private static async Task<IResult> GetQuoteAsync(
-    //    IRouteService routeService,
-    //    HttpContext httpContext,
-    //    [AsParameters] GetQuoteQueryParams queryParams)
-    //{
-    //    var quoteRequest = new QuoteRequest
-    //    {
-    //        SourceNetwork = queryParams.SourceNetwork!,
-    //        SourceToken = queryParams.SourceToken!,
-    //        DestinationNetwork = queryParams.DestinationNetwork!,
-    //        DestinationToken = queryParams.DestinationToken!,
-    //        Amount = queryParams.Amount!.Value,
-    //    };
-
-    //    var quote = await routeService.GetValidatedQuoteAsync(quoteRequest);
-
-    //    if (quote == null)
-    //    {
-    //        return Results.NotFound(new ApiResponse()
-    //        {
-    //            Error = new ApiError()
-    //            {
-    //                Code = "QUOTE_NOT_FOUND",
-    //                Message = "Quote not found",
-    //            }
-    //        });
-    //    }
-
-    //    return Results.Ok(new ApiResponseQuoteDto { Data = quote });
-    //}
+    private static async Task GetQuoteAsync(
+        HttpContext httpContext,
+        RouteCache routeCache,
+        SolverCache solverCache,
+        IHttpClientFactory httpClientFactory,
+        [FromQuery] string sourceNetwork,
+        [FromQuery] string sourceToken,
+        [FromQuery] string destinationNetwork,
+        [FromQuery] string destinationToken,
+        [FromQuery] double amount)
+    {
+        httpContext.Response.Headers.ContentType = "text/event-stream";
+
+        var lps = routeCache.GetLpsByRoute(
+            sourceNetwork,
+            sourceToken,
+            destinationNetwork,
+            destinationToken);
+
+        var cancellationToken = httpContext.RequestAborted;
+
+            foreach (var lpName in lps)
+            {
+                if (cancellationToken.IsCancellationRequested)
+                {
+                    break;
+                }
+
+                try
+                {
+                    var solverInfo = solverCache.GetAll()[lpName];
+                    var httpClient = httpClientFactory.CreateClient(lpName);
+
+                    var trainSilverClient = new TrainSolverApiClient(
+                        solverInfo.Url.ToString(), httpClient);
+
+                    var quote = await trainSilverClient.QuoteAsync(
+                        amount,
+                        sourceNetwork,
+                        sourceToken,
+                        destinationNetwork,
+                        destinationToken);
+
+                    var message = new
+                    {
+                        Provider = lpName,
+                        Quote = quote.Data
+                    };
+
+                    var json = System.Text.Json.JsonSerializer.Serialize(message);
+
+                    await httpContext.Response.WriteAsync($"data: {json}\n\n");
+                    await httpContext.Response.Body.FlushAsync();
+
+                }
+                catch
+                {
+                }
+            }
+
+         
+
+        await httpContext.Response.Body.FlushAsync();
+    }
 }

TrainStation/Program.cs

@@ -67,6 +67,9 @@
 
 var app = builder.Build();
 
+app.UseRateLimiter();
+app.UseCors();
+
 app.MapGroup("/api")
     .MapGet("/health", () => Results.Ok())
     .WithTags("System")
@@ -85,7 +88,5 @@
     c.DisplayRequestDuration();
 });
 
-app.UseRateLimiter();
-app.UseCors();
 
 await app.RunAsync();

TrainStation/Services/RoutePollingService.cs

@@ -15,8 +15,8 @@ protected override async Task ExecuteAsync(CancellationToken stoppingToken)
             {
                 try
                 {
-                    var name = lp.Name;
-                    var baseAddress = lp.Url.ToString();
+                    var name = lp.Key;
+                    var baseAddress = lp.Value.Url.ToString();
 
                     var trainSilverClient = new TrainSolverApiClient(
                         baseAddress, 

TrainStation/Services/SolverCache.cs

@@ -1,11 +1,12 @@
-using System.Text.Json;
+using System.Collections.Concurrent;
+using System.Text.Json;
 using Train.Station.API.Models;
 
 namespace Train.Station.API.Services;
 
 public class SolverCache
 {
-    private readonly List<Solver> _solvers;
+    private readonly Dictionary<string, Solver> _solvers;
 
     public SolverCache(IWebHostEnvironment env)
     {
@@ -19,8 +20,8 @@ public SolverCache(IWebHostEnvironment env)
         _solvers = JsonSerializer.Deserialize<List<Solver>>(json, new JsonSerializerOptions
         {
             PropertyNameCaseInsensitive = true
-        }) ?? new List<Solver>();
+        }).ToDictionary(x=> x.Name) ?? new ();
     }
 
-    public IReadOnlyList<Solver> GetAll() => _solvers;
+    public IReadOnlyDictionary<string, Solver> GetAll() => _solvers;
 }