add opt.shouldQueryDeparturesAt(stopId, hops) 📝; 5.0.0

examples/simple.js

@@ -8,7 +8,10 @@ import {createWalkAndDiscoverStations as createWalk} from '../index.js'
 const hafas = createHafas(dbProfile, 'hafas-discover-stations example')
 const walk = createWalk(hafas)
 const berlinFriedrichstr = '8011306'
-const stations = walk(berlinFriedrichstr, {concurrency: 8})
+const stations = walk(berlinFriedrichstr, {
+	concurrency: 8,
+	shouldQueryDeparturesAt: (stopId, hops) => hops <= 1,
+})
 
 const DEV = process.env.NODE_ENV === 'dev'
 stations.on('hafas-error', (err) => {

index.js

@@ -39,6 +39,7 @@ const createWalkAndDiscoverStations = (hafas) => {
 			parseStationId: id => id,
 			subStops: true,
 			linesOfStops: false,
+			shouldQueryDeparturesAt: (stopId, hops) => true,
 			...opt,
 		}
 		if (!('when' in opt) || opt.when === null) {
@@ -91,8 +92,11 @@ const createWalkAndDiscoverStations = (hafas) => {
 			visitedStopsAndStations[sId] = true
 			nrOfStopsAndStations++
 			out.push(station)
-			queue.push(queryDepartures(sId, hops + 1))
 			stopsAndStationsPerSecond = stopsAndStationsSpeed(1)
+
+			if (opt.shouldQueryDeparturesAt(sId, hops + 1)) {
+				queue.push(queryDepartures(sId, hops + 1))
+			}
 		}
 
 		const onStop = (stop, hops) => {
@@ -104,8 +108,11 @@ const createWalkAndDiscoverStations = (hafas) => {
 
 			nrOfStopsAndStations++
 			out.push(stop)
-			queue.push(queryDepartures(sId, hops + 1))
 			stopsAndStationsPerSecond = stopsAndStationsSpeed(1)
+
+			if (opt.shouldQueryDeparturesAt(sId, hops + 1)) {
+				queue.push(queryDepartures(sId, hops + 1))
+			}
 		}
 
 		const onStopsAndStations = (stopsAndStations, hops) => {

package.json

@@ -1,7 +1,7 @@
 {
 	"name": "hafas-discover-stations",
 	"description": "Pass in a HAFAS client, discover stations by querying departures.",
-	"version": "4.2.2",
+	"version": "5.0.0",
 	"type": "module",
 	"main": "index.js",
 	"files": [

readme.md

@@ -60,6 +60,7 @@ walk(stationId, [opt])
 - `timeout`: timeout for a single job in milliseconds – default: `10000`
 - `parseStationId`: an optional function to process station IDs – default: `id => id`
 - `stationLines`: Query lines of stops/stations? – default: `false`
+- `shouldQueryDeparturesAt(stopId, nrOfHops)`: Should departures be queries from this `stopId`, effectively continuing the graph search? – default: `() => true`
 
 
 ## Contributing