X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FAdderBusinessLogic.java;h=cbd500dcdab4de973960aa6ca8bf52fa3be21f46;hb=b9a482c8228e0b9197d7cb0c3f6b8849e5cd66da;hp=a3848c466fe1dde9d393a4b7718fc541b0b8b330;hpb=47657918b484fad60ac4062177cfeb569275d3d3;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/AdderBusinessLogic.java b/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
index a3848c4..cbd500d 100644
--- a/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
+++ b/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
@@ -3,6 +3,7 @@ package de.juplo.kafka;
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Optional;
 
 
 public class AdderBusinessLogic
@@ -21,11 +22,33 @@ public class AdderBusinessLogic
   }
 
 
-  public synchronized void startSum(String user)
+  public synchronized Optional<Long> getSum(String user)
   {
-    if (state.containsKey(user))
-      throw new IllegalStateException("Sumation for " + user + " already in progress, state: " + state.get(user));
+    return Optional.ofNullable(state.get(user));
+  }
 
-    state.put(user, 0l);
+  public synchronized void addToSum(String user, Integer value)
+  {
+    if (value == null || value < 1)
+      throw new IllegalArgumentException("Not a positive number: " + value);
+
+    long sum =
+        Optional
+            .ofNullable(state.get(user))
+            .orElse(0l);
+    state.put(user, sum + value);
+  }
+
+  public synchronized Long calculate(String user)
+  {
+    if (!state.containsKey(user))
+      throw new IllegalStateException("No sumation for " + user + " in progress");
+
+    return state.remove(user);
+  }
+
+  protected Map<String, Long> getState()
+  {
+    return state;
   }
 }