X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FAdderBusinessLogic.java;h=64fdb8ca3197e0d20861a8a0df5ccd0762460654;hb=4a8642d0c36413cbc283d35dda8977b0e9320372;hp=ecd9590de696b14501145682be4dbd58a0b6c955;hpb=d576eea9bc9208d9e5003bd8c8c132bed96b5c40;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 ecd9590..64fdb8c 100644
--- a/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
+++ b/src/main/java/de/juplo/kafka/AdderBusinessLogic.java
@@ -29,21 +29,22 @@ 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);
+    long sum =
+        Optional
+            .ofNullable(state.get(user))
+            .orElse(0l);
+    state.put(user, sum + value);
   }
 
-  public synchronized Long calculate(String user)
+  public synchronized AdderResult calculate(String user)
   {
     if (!state.containsKey(user))
       throw new IllegalStateException("No sumation for " + user + " in progress");
 
-    return state.remove(user);
+    return new AdderResult(66, state.remove(user));
   }
 
   protected Map<String, Long> getState()