fix(collector/warsaw): processing free_places_total of warsaw

gbaranski · Nov 30, 2023 · 45d863d · 45d863d
1 parent 5e2161b
commit 45d863d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/collector/warsaw/source.go b/collector/warsaw/source.go
@@ -93,12 +93,17 @@ func (s Source) ParkingLots(ctx context.Context) (<-chan map[wheretopark.ID]wher
 		}
 
 		state := wheretopark.State{
-			LastUpdated: lastUpdate.In(defaultTimezone),
-			AvailableSpots: map[string]uint{
-				wheretopark.SpotTypeCarElectric: vendor.FreePlacesTotal.Electric,
-				wheretopark.SpotTypeCar:         vendor.FreePlacesTotal.Public,
-				wheretopark.SpotTypeCarDisabled: vendor.FreePlacesTotal.Disabled,
-			},
+			LastUpdated:    lastUpdate.In(defaultTimezone),
+			AvailableSpots: map[string]uint{},
+		}
+		if vendor.FreePlacesTotal.Public > 0 {
+			state.AvailableSpots[wheretopark.SpotTypeCar] = uint(vendor.FreePlacesTotal.Public)
+		}
+		if vendor.FreePlacesTotal.Disabled > 0 {
+			state.AvailableSpots[wheretopark.SpotTypeCarDisabled] = uint(vendor.FreePlacesTotal.Disabled)
+		}
+		if vendor.FreePlacesTotal.Electric > 0 {
+			state.AvailableSpots[wheretopark.SpotTypeCarElectric] = uint(vendor.FreePlacesTotal.Electric)
 		}
 		parkingLots[id] = wheretopark.ParkingLot{
 			Metadata: metadata,

diff --git a/collector/warsaw/vendor.go b/collector/warsaw/vendor.go
@@ -1,9 +1,9 @@
 package warsaw
 
 type FreePlacesTotal struct {
-	Disabled uint `json:"disabled"`
-	Public   uint `json:"public"`
-	Electric uint `json:"electric"`
+	Disabled int `json:"disabled"`
+	Public   int `json:"public"`
+	Electric int `json:"electric"`
 }
 
 type TotalPlaces struct {