skip count

Travigo · Nov 9, 2024 · 31c95a7 · 31c95a7
1 parent e2926aa
commit 31c95a7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/pkg/realtime/vehicletracker/vehicleupdateevent.go b/pkg/realtime/vehicletracker/vehicleupdateevent.go
@@ -15,6 +15,7 @@ type VehicleUpdateEvent struct {
 	SourceType             string
 
 	VehicleLocationUpdate *VehicleLocationUpdate
+	ServiceAlertUpdate    *ServiceAlertUpdate
 
 	DataSource *ctdf.DataSource
 	RecordedAt time.Time
@@ -48,3 +49,13 @@ type VehicleLocationEventStopUpdate struct {
 	ArrivalOffset   int
 	DepartureOffset int
 }
+
+type ServiceAlertUpdate struct {
+	Type ctdf.ServiceAlertType
+
+	Title       string
+	Description string
+
+	ValidFrom  time.Time
+	ValidUntil time.Time
+}
diff --git a/pkg/stats/calculator/util.go b/pkg/stats/calculator/util.go
@@ -40,6 +40,10 @@ func CountCountries(datasources map[string]int) map[string]int {
 		datasourceSplit := strings.Split(datasource, "-")
 		country := datasourceSplit[0]
 
+		if country == "travigo" {
+			continue
+		}
+
 		countries[country] += count
 	}