@transportme/load-ptv-gtfs 1.4.18 → 1.4.20

package/lib/gtfs-parser/GTFSTrip.mjs

@@ -2,7 +2,7 @@ import { TRANSIT_MODES } from '../constants.mjs'
 import GTFSCalendar from './GTFSCalendar.mjs'
 
 export default class GTFSTrip {
-  
+
   #routeGTFSID
   #calendarName
   #operationDays
@@ -11,6 +11,7 @@ export default class GTFSTrip {
   #headsign
   #direction
   #block
+  #stopData
 
   /**
    * 
@@ -102,6 +103,14 @@ export default class GTFSTrip {
   getGTFSDirection() {
     return parseInt(this.#direction)
   }
+
+  setStopData(stopData) {
+    this.#stopData = stopData
+  }
+
+  getStopData() {
+    return this.#stopData
+  }
 }
 
 /**

package/lib/loader/TripLoader.mjs

@@ -151,6 +151,7 @@ export default class TripLoader {
       return data
     })
 
+    tripData.setStopData(stopTimings)
 
     let dbTrip = {
       mode: this.#mode,
@@ -200,6 +201,7 @@ export default class TripLoader {
     this.updateStopServices(processedData)
     
     this.#tripsToLoad.push(processedData)
+    this.#tripDetails[tripID] = null // Clear it from the cache once loaded to free up some memory
   }
 
   updateStopServices(trip) {
@@ -225,7 +227,7 @@ export default class TripLoader {
    * @param {object} options.routeIDMap A mapping of full route IDs to their simplified IDs.
    * @param {function} [options.processTrip] A function to modify trip data on a per-trip basis. It should return null to discard the trip.
    */
-  async loadTrips({ routeIDMap, processTrip }) {
+  async loadTrips({ routeIDMap, processTrip, onWrite = () => {} }) {
     this.#routeIDMap = routeIDMap
 
     await this.loadCalendarData()
@@ -241,11 +243,13 @@ export default class TripLoader {
       if (this.#tripsToLoad.length === 5000) { // Flush the cache every 5000 trips
         tripsLoaded += this.#tripsToLoad.length
         await this.writeTrips()
+        onWrite(tripsLoaded)
       }
     }
 
     tripsLoaded += this.#tripsToLoad.length
     await this.writeTrips()
+    onWrite(tripsLoaded)
   }
 
   getTripsToLoad() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@transportme/load-ptv-gtfs",
-  "version": "1.4.18",
+  "version": "1.4.20",
   "description": "",
   "main": "index.mjs",
   "scripts": {