From c2fcb3f036d738f3eb1f94a0ec177023b8b99567 Mon Sep 17 00:00:00 2001
From: Bertrik Sikken <bertrik@sikken.nl>
Date: Tue, 23 Jan 2024 08:37:44 +0100
Subject: [PATCH] Decode xml gz payload in TrafficHandler

---
 .../traffic/TrafficHandler.java               | 39 +++++++++++++++++--
 1 file changed, 36 insertions(+), 3 deletions(-)

diff --git a/verkeersdrukte/src/main/java/nl/bertriksikken/verkeersdrukte/traffic/TrafficHandler.java b/verkeersdrukte/src/main/java/nl/bertriksikken/verkeersdrukte/traffic/TrafficHandler.java
index c1ebdbb..0796e71 100644
--- a/verkeersdrukte/src/main/java/nl/bertriksikken/verkeersdrukte/traffic/TrafficHandler.java
+++ b/verkeersdrukte/src/main/java/nl/bertriksikken/verkeersdrukte/traffic/TrafficHandler.java
@@ -1,13 +1,20 @@
 package nl.bertriksikken.verkeersdrukte.traffic;
 
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.dataformat.xml.XmlMapper;
 import com.google.common.util.concurrent.Runnables;
 import io.dropwizard.lifecycle.Managed;
+import nl.bertriksikken.datex2.D2LogicalModel;
+import nl.bertriksikken.datex2.MeasuredValue;
+import nl.bertriksikken.datex2.SiteMeasurements;
+import nl.bertriksikken.verkeersdrukte.ndw.FileResponse;
 import nl.bertriksikken.verkeersdrukte.ndw.NdwClient;
 import nl.bertriksikken.verkeersdrukte.ndw.NdwConfig;
-import nl.bertriksikken.verkeersdrukte.ndw.FileResponse;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.time.Duration;
 import java.time.Instant;
@@ -15,6 +22,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.*;
+import java.util.zip.GZIPInputStream;
 
 public final class TrafficHandler implements ITrafficHandler, Managed {
 
@@ -24,10 +32,11 @@ public final class TrafficHandler implements ITrafficHandler, Managed {
 
     private final ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
     private final NdwClient ndwClient;
+    private final ObjectMapper mapper;
 
     public TrafficHandler(NdwConfig config) {
         this.ndwClient = NdwClient.create(config);
-
+        this.mapper = new XmlMapper();
     }
 
     @Override
@@ -54,21 +63,45 @@ private void downloadTrafficSpeed() {
             Duration age = Duration.between(response.getLastModified(), Instant.now());
             next = response.getLastModified().plusSeconds(65);
             LOG.info("Got data, {} bytes, age {}", response.getContents().length, age);
+            decode(new ByteArrayInputStream(response.getContents()));
         } catch (IOException e) {
             LOG.warn("Download failed", e);
             next = Instant.now().plusSeconds(60);
         }
 
+        // schedule next
         Duration interval = Duration.between(Instant.now(), next);
         while (interval.isNegative()) {
             interval = interval.plusSeconds(60);
         }
-        LOG.info("Scheduling next download in {}, at {}", interval, next);
+        LOG.info("Scheduling next download in {}", interval);
         executor.schedule(this::downloadTrafficSpeed, interval.toMillis(), TimeUnit.MILLISECONDS);
 
         notifyClients();
     }
 
+    private void decode(ByteArrayInputStream inputStream) throws IOException {
+        LOG.info("Parsing");
+        try (GZIPInputStream gzis = new GZIPInputStream(inputStream)) {
+            JsonNode node = mapper.readValue(gzis, JsonNode.class);
+            JsonNode d2LogicalModel = node.at("/Body/d2LogicalModel");
+            D2LogicalModel model = mapper.treeToValue(d2LogicalModel, D2LogicalModel.class);
+            D2LogicalModel.PayloadPublication payloadPublication = model.payloadPublication;
+            LOG.info("Payload publication: type {}, time {}", payloadPublication.type, payloadPublication.publicationTime);
+
+            D2LogicalModel.MeasuredDataPublication measuredDataPublication = (D2LogicalModel.MeasuredDataPublication) payloadPublication;
+            int numMeasurements = 0;
+            int numSites = 0;
+            for (SiteMeasurements measurements : measuredDataPublication.siteMeasurementsList) {
+                for (MeasuredValue value : measurements.measuredValueList) {
+                    numMeasurements++;
+                }
+                numSites++;
+            }
+            LOG.info("Parsed {} measurements from {} sites", numMeasurements, numSites);
+        }
+    }
+
     @Override
     public void subscribe(String clientId, INotifyData callback) {
         Subscription subscription = new Subscription(clientId, callback);