X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FDriverController.java;h=3aa931402d9d99c048d38bf2faf798fd22511c0a;hb=3495017ed2116f338c3342a313abdb7170683573;hp=a02fd2c7cfc7f45966ce488c584f0159212ac8c0;hpb=af66d390cd0a7f26e653cc129a89a63d11d8a7b6;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 a02fd2c..3aa9314 100644
--- a/src/main/java/de/juplo/kafka/DriverController.java
+++ b/src/main/java/de/juplo/kafka/DriverController.java
@@ -2,12 +2,14 @@ package de.juplo.kafka;
 
 import lombok.RequiredArgsConstructor;
 import org.springframework.http.HttpStatus;
-import org.springframework.web.bind.annotation.ExceptionHandler;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.ResponseStatus;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.http.ResponseEntity;
+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
@@ -15,6 +17,7 @@ import java.util.concurrent.ExecutionException;
 public class DriverController
 {
   private final EndlessConsumer consumer;
+  private final SumRecordHandler sumRecordHandler;
 
 
   @PostMapping("start")
@@ -29,6 +32,34 @@ public class DriverController
     consumer.stop();
   }
 
+
+  @GetMapping("state")
+  public Map<Integer, Map<String, Long>> state()
+  {
+    return
+        sumRecordHandler
+            .getState()
+            .entrySet()
+            .stream()
+            .collect(Collectors.toMap(
+                entry -> entry.getKey(),
+                entry -> entry.getValue().getState()));
+  }
+
+  @GetMapping("state/{user}")
+  public ResponseEntity<Long> seen(@PathVariable String user)
+  {
+    for (SumBusinessLogic sumBusinessLogic : sumRecordHandler.getState().values())
+    {
+      Optional<Long> sum = sumBusinessLogic.getSum(user);
+      if (sum.isPresent())
+        return ResponseEntity.ok(sum.get());
+    }
+
+    return ResponseEntity.notFound().build();
+  }
+
+
   @ExceptionHandler
   @ResponseStatus(HttpStatus.BAD_REQUEST)
   public ErrorResponse illegalStateException(IllegalStateException e)