X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FDriverController.java;fp=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FDriverController.java;h=3aa931402d9d99c048d38bf2faf798fd22511c0a;hb=3495017ed2116f338c3342a313abdb7170683573;hp=fdae76f979c98aa6016af81b1b0c12ba0666c405;hpb=5f35354fd694f78599d66ee9e01fb4c0d89cc5bb;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/DriverController.java b/src/main/java/de/juplo/kafka/DriverController.java
index fdae76f..3aa9314 100644
--- a/src/main/java/de/juplo/kafka/DriverController.java
+++ b/src/main/java/de/juplo/kafka/DriverController.java
@@ -7,7 +7,9 @@ import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.concurrent.ExecutionException;
+import java.util.stream.Collectors;
 
 
 @RestController
@@ -31,20 +33,27 @@ public class DriverController
   }
 
 
-  @GetMapping("seen")
-  public Map<Integer, Map<String, List<Long>>> seen()
+  @GetMapping("state")
+  public Map<Integer, Map<String, Long>> state()
   {
-    return sumRecordHandler.getSeen();
+    return
+        sumRecordHandler
+            .getState()
+            .entrySet()
+            .stream()
+            .collect(Collectors.toMap(
+                entry -> entry.getKey(),
+                entry -> entry.getValue().getState()));
   }
 
-  @GetMapping("seen/{user}")
-  public ResponseEntity<List<Long>> seen(@PathVariable String user)
+  @GetMapping("state/{user}")
+  public ResponseEntity<Long> seen(@PathVariable String user)
   {
-    for (Map<String, List<Long>> users : sumRecordHandler.getSeen().values())
+    for (SumBusinessLogic sumBusinessLogic : sumRecordHandler.getState().values())
     {
-      List<Long> results = users.get(user);
-      if (results != null)
-        return ResponseEntity.ok(results);
+      Optional<Long> sum = sumBusinessLogic.getSum(user);
+      if (sum.isPresent())
+        return ResponseEntity.ok(sum.get());
     }
 
     return ResponseEntity.notFound().build();