X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FAdderBusinessLogic.java;h=ecd9590de696b14501145682be4dbd58a0b6c955;hb=d576eea9bc9208d9e5003bd8c8c132bed96b5c40;hp=bd89871be92a9d41c0240e98a73126dd3eecfdd2;hpb=3aa0d314b04adea02cb8c7735687ac46e44601b3;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 bd89871..ecd9590 100644
--- a/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
+++ b/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
@@ -22,14 +22,6 @@ public class AdderBusinessLogic
   }
 
 
-  public synchronized void startSum(String user)
-  {
-    if (state.containsKey(user))
-      throw new IllegalStateException("Sumation for " + user + " already in progress, state: " + state.get(user));
-
-    state.put(user, 0l);
-  }
-
   public synchronized Optional<Long> getSum(String user)
   {
     return Optional.ofNullable(state.get(user));
@@ -37,13 +29,25 @@ public class AdderBusinessLogic
 
   public synchronized void addToSum(String user, Integer value)
   {
+    if (!state.containsKey(user))
+      throw new IllegalStateException("No sumation for " + user + " in progress");
+    if (value == null || value < 1)
+      throw new IllegalArgumentException("Not a positive number: " + value);
+
+    long result = state.get(user) + value;
+    state.put(user, result);
   }
 
-  public synchronized Long endSum(String user)
+  public synchronized Long calculate(String user)
   {
     if (!state.containsKey(user))
       throw new IllegalStateException("No sumation for " + user + " in progress");
 
-    return state.get(user);
+    return state.remove(user);
+  }
+
+  protected Map<String, Long> getState()
+  {
+    return state;
   }
 }