From c4fa56635fa9a40ea0915ed65f782beeac0bcea3 Mon Sep 17 00:00:00 2001
From: jmarchel <jmarchel@7bulls.com>
Date: Fri, 8 Mar 2024 13:44:51 +0100
Subject: [PATCH] Fix format of type 2 message

Change-Id: I17e85c5f49c050fcc2d4b2edc257b43237873b74
---
 .../ApplicationSpecificPredictionConsumer.java    | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/prediction-orchestrator/src/main/java/eu/nebulouscloud/predictionorchestrator/ApplicationSpecificPredictionConsumer.java b/prediction-orchestrator/src/main/java/eu/nebulouscloud/predictionorchestrator/ApplicationSpecificPredictionConsumer.java
index 2dcde79..76426f5 100644
--- a/prediction-orchestrator/src/main/java/eu/nebulouscloud/predictionorchestrator/ApplicationSpecificPredictionConsumer.java
+++ b/prediction-orchestrator/src/main/java/eu/nebulouscloud/predictionorchestrator/ApplicationSpecificPredictionConsumer.java
@@ -64,16 +64,17 @@ public class ApplicationSpecificPredictionConsumer extends Consumer {
 
         private Map<String, Object> transformToPredictedMetric(Map<String, Object> metric) {
             Map<String, Object> predictedMetric = new HashMap<>(metric);
-
-            // Set the prediction confidence to 0.60 as a naive prediction
-            predictedMetric.put("prediction_confidence", 0.60);
-
-            // Set the prediction interval with a length of zero
             Double metricValue = (Double) metric.get("metricValue");
-            predictedMetric.put("prediction_interval", Arrays.asList(metricValue, metricValue));
+            int level = (int) metric.get("level");
 
+            predictedMetric.put("timestamp", System.currentTimeMillis() / 1000);
+            // Set the prediction probability to 0.60 as a naive prediction
+            predictedMetric.put("probability", 0.60);
+            predictedMetric.put("level", level);
+            predictedMetric.put("metricValue", metricValue);
+            predictedMetric.put("confidence_interval", Arrays.asList(metricValue, metricValue));
             // Use the current system time as the prediction time
-            predictedMetric.put("prediction_time", System.currentTimeMillis() / 1000);
+            predictedMetric.put("predictionTime", System.currentTimeMillis() / 1000);
 
             return predictedMetric;
         }