X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FKeyCountingRecordHandler.java;h=83b3ff2cd584907548bab7f137e43bf2ac610e4a;hb=487ca1ac0c73b1c07547d5316d145ed46490ba9d;hp=099dcf7c438ec851eb4de5ef150142f8ea884e31;hpb=cfbe7dcd3318ee846cb5890eea4328e36c5aa364;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/KeyCountingRecordHandler.java b/src/main/java/de/juplo/kafka/KeyCountingRecordHandler.java
index 099dcf7..83b3ff2 100644
--- a/src/main/java/de/juplo/kafka/KeyCountingRecordHandler.java
+++ b/src/main/java/de/juplo/kafka/KeyCountingRecordHandler.java
@@ -18,6 +18,10 @@ public class KeyCountingRecordHandler implements RecordHandler<String, Long>
   {
     Integer partition = record.partition();
     String key = record.key() == null ? "NULL" : record.key().toString();
+
+    if (!seen.containsKey(partition))
+      seen.put(partition, new HashMap<>());
+
     Map<String, Long> byKey = seen.get(partition);
 
     if (!byKey.containsKey(key))
@@ -28,16 +32,6 @@ public class KeyCountingRecordHandler implements RecordHandler<String, Long>
     byKey.put(key, seenByKey);
   }
 
-  public void addPartition(Integer partition, Map<String, Long> statistics)
-  {
-    seen.put(partition, statistics);
-  }
-
-  public Map<String, Long> removePartition(Integer partition)
-  {
-    return seen.remove(partition);
-  }
-
 
   public Map<Integer, Map<String, Long>> getSeen()
   {