X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FApplicationRecordHandler.java;h=bc18d59b6596507d9ebc03678ceb2b3f487512e4;hb=refs%2Ftags%2Fsumup-adder--drop-duplicates---lvm-2-tage;hp=ef38357ede2d8a334d2dfea6942c6e7ffee6c3af;hpb=9ac94aed684ce23a186792a3275a574d5adfa836;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 ef38357..bc18d59 100644
--- a/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java
+++ b/src/main/java/de/juplo/kafka/ApplicationRecordHandler.java
@@ -73,13 +73,9 @@ public class ApplicationRecordHandler implements RecordHandler<String, String>
 
   protected ApplicationState removePartition(Integer partition)
   {
-    ApplicationState state =
-      new ApplicationState(
-        this.next.get(partition),
-        this.state.remove(partition).getState());
-
+    ApplicationState state = getState(partition);
     this.next.remove(partition);
-
+    this.state.remove(partition);
     return state;
   }
 
@@ -89,8 +85,11 @@ public class ApplicationRecordHandler implements RecordHandler<String, String>
     return state;
   }
 
-  public AdderBusinessLogic getState(Integer partition)
+  public ApplicationState getState(Integer partition)
   {
-    return state.get(partition);
+    return
+      new ApplicationState(
+        this.next.get(partition),
+        this.state.get(partition).getState());
   }
 }