X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2FTestRecordHandler.java;h=d9f4e679ad9a5ced4ce34916cd6aae8293a78489;hb=66ff7d205e66616de8aaca94503dbbcd7d281f6d;hp=4047093b03de41932c10d4e6cd7acc876ab066e4;hpb=2d84eda74475aaffff11ddfebe56d309b9aff2e9;p=demos%2Fkafka%2Ftraining

diff --git a/src/test/java/de/juplo/kafka/TestRecordHandler.java b/src/test/java/de/juplo/kafka/TestRecordHandler.java
index 4047093..d9f4e67 100644
--- a/src/test/java/de/juplo/kafka/TestRecordHandler.java
+++ b/src/test/java/de/juplo/kafka/TestRecordHandler.java
@@ -1,41 +1,52 @@
 package de.juplo.kafka;
 
 import lombok.RequiredArgsConstructor;
-import org.apache.kafka.clients.consumer.ConsumerRecord;
 import org.apache.kafka.common.TopicPartition;
 
+import java.util.List;
+import java.util.Map;
+
 
 @RequiredArgsConstructor
-public abstract class TestRecordHandler<K, V> implements RecordHandler<K, V>
+public class TestRecordHandler implements RecordHandler
 {
-  private final RecordHandler<K, V> handler;
-
+  private final RecordHandler handler;
 
-  public abstract void onNewRecord(ConsumerRecord<K, V> record);
+  Map<TopicPartition, Long> seenOffsets;
+  int receivedMessages;
 
 
-  @Override
-  public void accept(ConsumerRecord<K, V> record)
-  {
-    this.onNewRecord(record);
-    handler.accept(record);
-  }
-  @Override
-
-  public void beforeNextPoll()
+  public void onNewRecord(
+      String topic,
+      Integer partition,
+      Long offset,
+      Message messgage)
   {
-    handler.beforeNextPoll();
+    seenOffsets.put(new TopicPartition(topic, partition), offset);
+    receivedMessages++;
   }
 
   @Override
-  public void onPartitionAssigned(TopicPartition tp)
+  public void addNumber(
+      String topic,
+      Integer partition,
+      Long offset,
+      String user,
+      MessageAddNumber message)
   {
-    handler.onPartitionAssigned(tp);
+    this.onNewRecord(topic, partition, offset, message);
+    handler.addNumber(topic, partition, offset, user, message);
   }
 
   @Override
-  public void onPartitionRevoked(TopicPartition tp)
+  public void calculateSum(
+      String topic,
+      Integer partition,
+      Long offset,
+      String user,
+      MessageCalculateSum message)
   {
-    handler.onPartitionRevoked(tp);
+    this.onNewRecord(topic, partition, offset, message);
+    handler.calculateSum(topic, partition, offset, user, message);
   }
 }