add additional method to get shuttles by route

This commit is contained in:
2024-12-22 19:56:46 -08:00
parent 2ca9551bfa
commit 0221bda314
2 changed files with 5 additions and 0 deletions

View File

@@ -67,6 +67,7 @@ export interface Repository {
getShuttlesBySystemId(systemId: string): Promise<IShuttle[]>;
getShuttleById(shuttleId: string): Promise<IShuttle | null>;
getShuttlesByRouteId(routeId: string): Promise<IShuttle[]>;
getEtasForShuttleId(shuttleId: string): Promise<IEta[]>;
getEtasForStopId(stopId: string): Promise<IEta[]>;

View File

@@ -41,6 +41,10 @@ export class UnoptimizedInMemoryRepository implements Repository {
return this.shuttles.filter(shuttle => shuttle.systemId === systemId);
}
public async getShuttlesByRouteId(routeId: string) {
return this.shuttles.filter(shuttle => shuttle.routeId === routeId);
}
public async getShuttleById(shuttleId: string) {
return this.findEntityById(shuttleId, this.shuttles);
}