diff options
Diffstat (limited to 'src/Enmarcha.Backend/Services/Processors')
| -rw-r--r-- | src/Enmarcha.Backend/Services/Processors/RenfeRealTimeProcessor.cs | 72 | ||||
| -rw-r--r-- | src/Enmarcha.Backend/Services/Processors/TussaRealTimeProcessor.cs (renamed from src/Enmarcha.Backend/Services/Processors/SantiagoRealTimeProcessor.cs) | 8 | ||||
| -rw-r--r-- | src/Enmarcha.Backend/Services/Processors/VitrasaUsageProcessor.cs (renamed from src/Enmarcha.Backend/Services/Processors/VigoUsageProcessor.cs) | 8 |
3 files changed, 80 insertions, 8 deletions
diff --git a/src/Enmarcha.Backend/Services/Processors/RenfeRealTimeProcessor.cs b/src/Enmarcha.Backend/Services/Processors/RenfeRealTimeProcessor.cs new file mode 100644 index 0000000..dcddd5d --- /dev/null +++ b/src/Enmarcha.Backend/Services/Processors/RenfeRealTimeProcessor.cs @@ -0,0 +1,72 @@ +using Enmarcha.Backend.Types; +using Enmarcha.Backend.Types.Arrivals; +using Enmarcha.Sources.GtfsRealtime; +using Arrival = Enmarcha.Backend.Types.Arrivals.Arrival; + +namespace Enmarcha.Backend.Services.Processors; + +public class RenfeRealTimeProcessor : AbstractRealTimeProcessor +{ + private readonly GtfsRealtimeEstimatesProvider _realtime; + private readonly ILogger<RenfeRealTimeProcessor> _logger; + + public RenfeRealTimeProcessor( + GtfsRealtimeEstimatesProvider realtime, + ILogger<RenfeRealTimeProcessor> logger + ) + { + _realtime = realtime; + _logger = logger; + } + + public override async Task ProcessAsync(ArrivalsContext context) + { + if (!context.StopId.StartsWith("renfe:")) return; + + try + { + var delays = await _realtime.GetRenfeDelays(); + var positions = await _realtime.GetRenfePositions(); + System.Diagnostics.Activity.Current?.SetTag("realtime.count", delays.Count); + + foreach (Arrival contextArrival in context.Arrivals) + { + var trainNumber = contextArrival.TripId.Split(":")[1][..5]; + + contextArrival.Headsign.Destination = trainNumber + " - " + contextArrival.Headsign.Destination; + + if (delays.TryGetValue(trainNumber, out var delay)) + { + if (delay is null) + { + // TODO: Indicate train got cancelled + continue; + } + + var delayMinutes = delay.Value / 60; + contextArrival.Delay = new DelayBadge() + { + Minutes = delayMinutes + }; + + contextArrival.Estimate.Minutes += delayMinutes; + contextArrival.Estimate.Precision = ArrivalPrecision.Confident; + } + + if (positions.TryGetValue(trainNumber, out var position)) + { + contextArrival.CurrentPosition = new Position + { + Latitude = position.Latitude, + Longitude = position.Longitude, + OrientationDegrees = 0 // TODO: Set the proper degrees + }; + } + } + } + catch (Exception ex) + { + _logger.LogError(ex, "Error fetching Renfe real-time data"); + } + } +} diff --git a/src/Enmarcha.Backend/Services/Processors/SantiagoRealTimeProcessor.cs b/src/Enmarcha.Backend/Services/Processors/TussaRealTimeProcessor.cs index a4f7d5b..7808a3f 100644 --- a/src/Enmarcha.Backend/Services/Processors/SantiagoRealTimeProcessor.cs +++ b/src/Enmarcha.Backend/Services/Processors/TussaRealTimeProcessor.cs @@ -5,16 +5,16 @@ using Arrival = Enmarcha.Backend.Types.Arrivals.Arrival; namespace Enmarcha.Backend.Services.Processors; -public class SantiagoRealTimeProcessor : AbstractRealTimeProcessor +public class TussaRealTimeProcessor : AbstractRealTimeProcessor { private readonly SantiagoRealtimeEstimatesProvider _realtime; private readonly FeedService _feedService; - private readonly ILogger<SantiagoRealTimeProcessor> _logger; + private readonly ILogger<TussaRealTimeProcessor> _logger; - public SantiagoRealTimeProcessor( + public TussaRealTimeProcessor( SantiagoRealtimeEstimatesProvider realtime, FeedService feedService, - ILogger<SantiagoRealTimeProcessor> logger) + ILogger<TussaRealTimeProcessor> logger) { _realtime = realtime; _feedService = feedService; diff --git a/src/Enmarcha.Backend/Services/Processors/VigoUsageProcessor.cs b/src/Enmarcha.Backend/Services/Processors/VitrasaUsageProcessor.cs index 52218d9..a2f90d3 100644 --- a/src/Enmarcha.Backend/Services/Processors/VigoUsageProcessor.cs +++ b/src/Enmarcha.Backend/Services/Processors/VitrasaUsageProcessor.cs @@ -4,17 +4,17 @@ using Microsoft.Extensions.Caching.Memory; namespace Enmarcha.Backend.Services.Processors; -public class VigoUsageProcessor : IArrivalsProcessor +public class VitrasaUsageProcessor : IArrivalsProcessor { private readonly HttpClient _httpClient; private readonly IMemoryCache _cache; - private readonly ILogger<VigoUsageProcessor> _logger; + private readonly ILogger<VitrasaUsageProcessor> _logger; private readonly FeedService _feedService; - public VigoUsageProcessor( + public VitrasaUsageProcessor( HttpClient httpClient, IMemoryCache cache, - ILogger<VigoUsageProcessor> logger, + ILogger<VitrasaUsageProcessor> logger, FeedService feedService) { _httpClient = httpClient; |
