WIP

[demos/kafka/training] / src / main / java / de / juplo / kafka / RestGateway.java

diff --git a/src/main/java/de/juplo/kafka/RestGateway.java b/src/main/java/de/juplo/kafka/RestGateway.java
index 4549b8f..53a87df 100644 (file)
--- a/src/main/java/de/juplo/kafka/RestGateway.java
+++ b/src/main/java/de/juplo/kafka/RestGateway.java
@@ -2,7 +2,7 @@ package de.juplo.kafka;
 
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.kafka.clients.producer.KafkaProducer;
+import org.apache.kafka.clients.producer.Producer;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.springframework.http.HttpStatus;
 import org.springframework.web.bind.annotation.*;
@@ -17,7 +17,8 @@ public class RestGateway
 {
   private final String id;
   private final String topic;
-  private final KafkaProducer<String, Integer> producer;
+  private final Integer partition;
+  private final Producer<String, Integer> producer;
 
   private long produced = 0;
 
@@ -33,6 +34,7 @@ public class RestGateway
 
     final ProducerRecord<String, Integer> record = new ProducerRecord<>(
         topic,  // Topic
+        partition, // Partition - Uses default-algorithm, if null
         key,    // Key
         value   // Value
     );