X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FDriverController.java;h=5a09c1b0c26f483f911386362485c0e0de071cd5;hb=c808810e9e33afe33b29f7fd3921023ecd15483d;hp=f6ff47f8ac49c458a593770fadafc2aab34bd5c4;hpb=b3777fba0ae679d9e2c9d36626fa208a952f83e8;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 f6ff47f..5a09c1b 100644
--- a/src/main/java/de/juplo/kafka/DriverController.java
+++ b/src/main/java/de/juplo/kafka/DriverController.java
@@ -2,6 +2,7 @@ package de.juplo.kafka;
 
 import lombok.RequiredArgsConstructor;
 import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.Map;
@@ -13,7 +14,7 @@ import java.util.concurrent.ExecutionException;
 public class DriverController
 {
   private final EndlessConsumer consumer;
-  private final KeyCountingRecordHandler keyCountingRecordHandler;
+  private final SumRecordHandler wordcount;
 
 
   @PostMapping("start")
@@ -30,9 +31,22 @@ public class DriverController
 
 
   @GetMapping("seen")
-  public Map<Integer, Map<String, Long>> seen()
+  public Map<Integer, Map<String, Map<String, Long>>> seen()
   {
-    return keyCountingRecordHandler.getSeen();
+    return wordcount.getSeen();
+  }
+
+  @GetMapping("seen/{user}")
+  public ResponseEntity<Map<String, Long>> seen(@PathVariable String user)
+  {
+    for (Map<String, Map<String, Long>> users : wordcount.getSeen().values())
+    {
+      Map<String, Long> words = users.get(user);
+      if (words != null)
+        return ResponseEntity.ok(words);
+    }
+
+    return ResponseEntity.notFound().build();
   }