BAEL3889 - Kafka Mock Producer

2020-05-31 18:39:20 +05:30
parent f952b3c747
commit d7723aeca0
4 changed files with 7 additions and 0 deletions
--- a/libraries-data-2/src/main/java/com/baeldung/kafka/EvenOddPartitioner.java
+++ b/libraries-data-2/src/main/java/com/baeldung/kafka/EvenOddPartitioner.java
@@ -0,0 +1,16 @@
+package com.baeldung.kafka;
+
+import org.apache.kafka.clients.producer.internals.DefaultPartitioner;
+import org.apache.kafka.common.Cluster;
+
+public class EvenOddPartitioner extends DefaultPartitioner {
+
+    @Override
+    public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes, Cluster cluster) {
+
+        if (((String)key).length() % 2 == 0)
+            return 0;
+
+        return 1;
+    }
+}