X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FAdderBusinessLogic.java;h=1f3d9aa2e67db2934b836d4093cb7485d354f881;hb=a2e8fc924e5b472d6b90c42d311514f91ea452f1;hp=1f8ebc2583245df37a9908673811ec9d1c48581e;hpb=d8215e876b3caae0352dc111856a41d549b4da20;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 1f8ebc2..1f3d9aa 100644
--- a/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
+++ b/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
@@ -35,11 +35,27 @@ public class AdderBusinessLogic
     return Optional.ofNullable(state.get(user));
   }
 
+  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)
   {
     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;
   }
 }