fix GTFS merging 🐛✅

follow-up to d4e7705

Author: derhuerst

lib/gtfs-matching.js

@@ -71,9 +71,8 @@ const runGtfsMatching = async (cfg, opt = {}) => {
 		help: 'number of successfully matched VDV Fahrts',
 		registers: [register],
 		labelNames: [
-			'with_sollfahrt', // merged IstFahrt contains a REF-AUS SollFahrt
-			'with_komplett_istfahrt', // merged IstFahrt contains a Komplettfahrt=true AUS IstFahrt
-			'with_partial_istfahrts', // merged IstFahrt contains >0 partial AUS IstFahrts
+			'prognose_moeglich', // merged VDV Fahrt contains PrognoseMoeglich=true
+			'komplettfahrt', // merged VDV Fahrt contains Komplettfahrt=true
 			'cached',
 		],
 	})
@@ -82,9 +81,8 @@ const runGtfsMatching = async (cfg, opt = {}) => {
 		help: 'number of matching failures',
 		registers: [register],
 		labelNames: [
-			'with_sollfahrt', // merged IstFahrt contains a REF-AUS SollFahrt
-			'with_komplett_istfahrt', // merged IstFahrt contains a Komplettfahrt=true AUS IstFahrt
-			'with_partial_istfahrts', // merged IstFahrt contains >0 partial AUS IstFahrts
+			'prognose_moeglich', // merged VDV Fahrt contains PrognoseMoeglich=true
+			'komplettfahrt', // merged VDV Fahrt contains Komplettfahrt=true
 			'cached',
 		],
 	})
@@ -93,19 +91,17 @@ const runGtfsMatching = async (cfg, opt = {}) => {
 		help: 'number of errors that have occured while matching',
 		registers: [register],
 		labelNames: [
-			'with_sollfahrt', // merged IstFahrt contains a REF-AUS SollFahrt
-			'with_komplett_istfahrt', // merged IstFahrt contains a Komplettfahrt=true AUS IstFahrt
-			'with_partial_istfahrts', // merged IstFahrt contains >0 partial AUS IstFahrts
+			'prognose_moeglich', // merged VDV Fahrt contains PrognoseMoeglich=true
+			'komplettfahrt', // merged VDV Fahrt contains Komplettfahrt=true
 		],
 	})
 	const matchingTimeSeconds = new Summary({
 		name: 'matching_time_seconds',
 		help: 'seconds trips need to be matched',
 		registers: [register],
 		labelNames: [
-			'with_sollfahrt', // merged IstFahrt contains a REF-AUS SollFahrt
-			'with_komplett_istfahrt', // merged IstFahrt contains a Komplettfahrt=true AUS IstFahrt
-			'with_partial_istfahrts', // merged IstFahrt contains >0 partial AUS IstFahrts
+			'prognose_moeglich', // merged VDV Fahrt contains PrognoseMoeglich=true
+			'komplettfahrt', // merged VDV Fahrt contains Komplettfahrt=true
 			'matched',
 			'cached',
 			// todo: schedule_feed_digest (slice)
@@ -168,10 +164,12 @@ const runGtfsMatching = async (cfg, opt = {}) => {
 
 		updateNatsMetricsForIncomingMsg(msg, tReceived)
 
+		const isPrognoseMoeglich = vdvFahrt.PrognoseMoeglich === 'true'
+		const isKomplettfahrt = vdvFahrt.Komplettfahrt === 'true'
+
 		const matchingMetricsLabels = {
-			with_sollfahrt: hasRefAusSollFahrt ? '1' : '0',
-			with_komplett_istfahrt: hasKomplettfahrtAusIstFahrt ? '1' : '0',
-			with_partial_istfahrts: hasPartialAusIstFahrts ? '1' : '0',
+			prognose_moeglich: isPrognoseMoeglich ? '1' : '0',
+			komplettfahrt: isKomplettfahrt ? '1' : '0',
 		}
 		let vdvFahrt = null
 		try {
@@ -188,8 +186,8 @@ const runGtfsMatching = async (cfg, opt = {}) => {
 
 		try {
 			// todo: validate against schema, error-log and abort if invalid
-			if (!Array.isArray(vdvFahrt[haltsField])) {
-				const err = new Error(`VDV Fahrt doesn't have an array ${haltsField}[]`)
+			if (!Array.isArray(vdvFahrt.IstHalts)) {
+				const err = new Error(`VDV Fahrt doesn't have an array .IstHalts[]`)
 				err.vdvFahrt = vdvFahrt
 				throw err
 			}

lib/merge-vdv-sollfahrts-istfahrts.js

@@ -389,13 +389,14 @@ const createMergeVdvFahrtWithRefAusSollFahrtAndAusIstFahrts = async (cfg) => {
 				return  [ausIstHaltWithIstFahrt.IstFahrt.Zst ?? null, ausIstHaltWithIstFahrt.IstFahrt.Zst ?? null]
 			}),
 		])
-		// todo: set PrognoseMoeglich=true as soon as there are >0 IstFahrts?
+		if (komplettfahrtAusIstFahrt || partialAusIstFahrts.length > 0) {
+			mergedIstFahrt.PrognoseMoeglich = 'true'
+		}
 
 		// merge {Soll,Ist}Halts
 		// todo: refactor this, the code is quite verbose for what it achieves
 		if (komplettfahrtAusIstFahrt) {
 			mergedIstFahrt.Komplettfahrt = 'true'
-			mergedIstFahrt.PrognoseMoeglich = 'true'
 			mergedIstFahrt.IstHalts = komplettfahrtAusIstFahrt.IstHalts.map((komplettfahrtAusIstHalt) => {
 				const isEquivalentVdvHalt = vdvHaltsAreEquivalent(komplettfahrtAusIstHalt)
 
@@ -429,7 +430,6 @@ const createMergeVdvFahrtWithRefAusSollFahrtAndAusIstFahrts = async (cfg) => {
 				)
 			})
 		} else if (partialAusIstFahrts.length > 0) {
-			mergedIstFahrt.PrognoseMoeglich = 'true'
 			mergedIstFahrt.IstHalts = partialAusIstFahrts
 				.map(ausIstHaltWithIstFahrt => omit(ausIstHaltWithIstFahrt, ['IstFahrt']))
 		} else {

test/fixtures/merged-aus-istfahrt-2025-04-11-76528-00066-1_VIP-no-komplettfahrt.js

@@ -8,7 +8,7 @@ const mergedAusIstFahrtNoKomplettfahrt = {
 	RichtungsText: 'Kirschallee',
 	VonRichtungsText: 'Bisamkiez',
 	VerkehrsmittelText: null,
-	PrognoseMoeglich: null,
+	PrognoseMoeglich: 'true',
 	Fahrradmitnahme: null,
 	FahrtID: {
 		FahrtBezeichner: '76528-00066-1#VIP',