Merge branch 'main' into feat/notification-service

src/repositories/GetterRepository.ts

@@ -16,9 +16,26 @@ export interface GetterRepository {
 
   getEtasForShuttleId(shuttleId: string): Promise<IEta[]>;
   getEtasForStopId(stopId: string): Promise<IEta[]>;
-
   getEtaForShuttleAndStopId(shuttleId: string, stopId: string): Promise<IEta | null>;
 
+  /**
+   * Subscribe to all updates in ETA data.
+   * The subscriber persists even if the ETA data does not
+   * exist within the repository, and may fire again
+   * if ETA data is restored.
+   * @param listener
+   */
+  subscribeToEtaUpdates(
+    listener: (eta: IEta) => void,
+  ): void;
+
+  /**
+   * Unsubscribe from all ETA updates for the given callback.
+   * Callback must be passed by reference.
+   * @param listener
+   */
+  unsubscribeFromEtaUpdates(listener: (eta: IEta) => void): void;
+
   getOrderedStopByRouteAndStopId(routeId: string, stopId: string): Promise<IOrderedStop | null>;
 
   /**
@@ -34,4 +51,4 @@ export interface GetterRepository {
    * @param routeId
    */
   getOrderedStopsByRouteId(routeId: string): Promise<IOrderedStop[]>;
-}
+}

src/repositories/UnoptimizedInMemoryRepository.ts

@@ -14,6 +14,8 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
   private etas: IEta[] = [];
   private orderedStops: IOrderedStop[] = [];
 
+  private subscribers: ((eta: IEta) => void)[] = [];
+
   public async getSystems() {
     return this.systems;
   }
@@ -49,7 +51,7 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
   public async getShuttleById(shuttleId: string) {
     return this.findEntityById(shuttleId, this.shuttles);
   }
-  
+
   public async getEtasForShuttleId(shuttleId: string) {
     return this.etas.filter(eta => eta.shuttleId === shuttleId);
   }
@@ -58,6 +60,17 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
     return this.etas.filter(eta => eta.stopId === stopId);
   }
 
+  public subscribeToEtaUpdates(listener: (eta: IEta) => void) {
+    this.subscribers.push(listener);
+  }
+
+  public unsubscribeFromEtaUpdates(listener: (eta: IEta) => void) {
+    const index = this.subscribers.findIndex((existingListener) => existingListener == listener);
+    if (index >= 0) {
+      this.subscribers.splice(index, 1);
+    }
+  }
+
   public async getEtaForShuttleAndStopId(shuttleId: string, stopId: string) {
     return this.findEntityByMatcher<IEta>((value) => value.stopId === stopId && value.shuttleId === shuttleId, this.etas);
   }
@@ -138,6 +151,13 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
     } else {
       this.etas.push(eta);
     }
+    this.publishEtaUpdateToSubscribers(eta);
+  }
+
+  private publishEtaUpdateToSubscribers(eta: IEta) {
+    this.subscribers.forEach(subscriber => {
+      subscriber(eta);
+    });
   }
 
   private async removeEntityByMatcherIfExists<T>(callback: (value: T) => boolean, arrayToSearchIn: T[]) {
@@ -209,4 +229,4 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
     this.stops = [];
   }
 
-}
+}