X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FDriverController.java;h=cc3abedacf9c5b06924eb5ea91497191c861b537;hb=eba11d4859d1e2b936bcd9e8075986b5179b32ea;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..cc3abed 100644
--- a/src/main/java/de/juplo/kafka/DriverController.java
+++ b/src/main/java/de/juplo/kafka/DriverController.java
@@ -2,37 +2,66 @@ 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.List;
 import java.util.Map;
-import java.util.concurrent.ExecutionException;
+import java.util.Optional;
+import java.util.stream.Collectors;
 
 
 @RestController
 @RequiredArgsConstructor
 public class DriverController
 {
-  private final EndlessConsumer consumer;
-  private final KeyCountingRecordHandler keyCountingRecordHandler;
+  private final ApplicationRecordHandler recordHandler;
+  private final AdderResults results;
 
 
-  @PostMapping("start")
-  public void start()
+  @GetMapping("state")
+  public Map<Integer, Map<String, AdderResult>> state()
   {
-    consumer.start();
+    return
+        recordHandler
+            .getState()
+            .entrySet()
+            .stream()
+            .collect(Collectors.toMap(
+                entry -> entry.getKey(),
+                entry -> entry.getValue().getState()));
   }
 
-  @PostMapping("stop")
-  public void stop() throws ExecutionException, InterruptedException
+  @GetMapping("state/{user}")
+  public ResponseEntity<Long> state(@PathVariable String user)
   {
-    consumer.stop();
+    for (AdderBusinessLogic adder : recordHandler.getState().values())
+    {
+      Optional<Long> sum = adder.getSum(user);
+      if (sum.isPresent())
+        return ResponseEntity.ok(sum.get());
+    }
+
+    return ResponseEntity.notFound().build();
   }
 
+  @GetMapping("results")
+  public Map<Integer, Map<String, List<AdderResult>>> results()
+  {
+    return results.getState();
+  }
 
-  @GetMapping("seen")
-  public Map<Integer, Map<String, Long>> seen()
+  @GetMapping("results/{user}")
+  public ResponseEntity<List<AdderResult>> results(@PathVariable String user)
   {
-    return keyCountingRecordHandler.getSeen();
+    for (Map<String, List<AdderResult>> resultsByUser : this.results.getState().values())
+    {
+      List<AdderResult> results = resultsByUser.get(user);
+      if (results != null)
+        return ResponseEntity.ok(results);
+    }
+
+    return ResponseEntity.notFound().build();
   }