traffic-a22-forecast: add validity check

noi-techpark · Oct 3, 2023 · 8c1e589 · 8c1e589
1 parent 05d154f
commit 8c1e589
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/...ffic-a22-forecast/src/main/java/com/opendatahub/traffic/a22/forecast/dto/ForecastDto.java b/...ffic-a22-forecast/src/main/java/com/opendatahub/traffic/a22/forecast/dto/ForecastDto.java
@@ -46,6 +46,14 @@ public static class ForecastData {
     public static class TrafficDataLine {
         @JsonProperty("PrevisioniTraffico")
         public List<TrafficData> data;
+
+        // if every day of whole month has value = 0; forecast is not valid
+        public boolean isValid() {
+            for (TrafficData trafficData : data)
+                if (trafficData.isValid())s
+                    return true;
+            return false;
+        }
     }
 
     @ToString
@@ -72,6 +80,14 @@ public static class TrafficData {
         @JsonProperty("Sud")
         @JsonIgnoreProperties("Tipo")
         public Map<String, TrafficValues> south;
+
+        public boolean isValid() {
+            for (TrafficValues values : south.values())
+                if ((values.value0to6.type + values.value6to12.type + values.value12to18.type
+                        + values.value18to24.type) == 0)
+                    return false;
+            return true;
+        }
     }
 
     @ToString

diff --git a/...-a22-forecast/src/main/java/com/opendatahub/traffic/a22/forecast/mapping/ForecastMap.java b/...-a22-forecast/src/main/java/com/opendatahub/traffic/a22/forecast/mapping/ForecastMap.java
@@ -19,10 +19,11 @@ public class ForecastMap extends HashMap<String, Map<Long, TrafficValues>> {
 
     public void add(ForecastDto dto) {
         for (TrafficDataLine line : dto.data.trafficDataLines) {
-            for (TrafficData data : line.data) {
-                putValues(data.south, data.date, "Sud");
-                putValues(data.north, data.date, "Nord");
-            }
+            if (line.isValid())
+                for (TrafficData data : line.data) {
+                    putValues(data.south, data.date, "Sud");
+                    putValues(data.north, data.date, "Nord");
+                }
         }
     }