X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FApplicationRecordHandler.java;h=93e12976b9e665d5c359f39fd55ba289588ca8ce;hb=7c0368363c3e5dbb7eb2a08f343187a93f050617;hp=ce340a76a96eaec73853fb42d67fd60e7a62e665;hpb=d576eea9bc9208d9e5003bd8c8c132bed96b5c40;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java b/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java
index ce340a7..93e1297 100644
--- a/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java
+++ b/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java
@@ -22,7 +22,7 @@ public class ApplicationRecordHandler implements RecordHandler<String, String>
 
     if (message.equals("CALCULATE"))
     {
-      Long result = state.get(partition).calculate(user);
+      AdderResult result = state.get(partition).calculate(user);
       log.info("New result for {}: {}", user, result);
       return;
     }
@@ -30,12 +30,12 @@ public class ApplicationRecordHandler implements RecordHandler<String, String>
     state.get(partition).addToSum(user, Integer.parseInt(message));
   }
 
-  protected void addPartition(Integer partition, Map<String, Long> state)
+  protected void addPartition(Integer partition, Map<String, AdderResult> state)
   {
     this.state.put(partition, new AdderBusinessLogic(state));
   }
 
-  protected Map<String, Long> removePartition(Integer partition)
+  protected Map<String, AdderResult> removePartition(Integer partition)
   {
     return this.state.remove(partition).getState();
   }