La til liten util funksjon i kafka-key test for å beregne partisjonsn…

…ummer basert på record key
navikt · Oct 28, 2024 · 3fd5678 · 3fd5678
1 parent 35548ae
commit 3fd5678
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/apps/kafka-key-generator/build.gradle.kts b/apps/kafka-key-generator/build.gradle.kts
@@ -69,6 +69,7 @@ dependencies {
     testImplementation(libs.test.testContainers.postgresql)
     testImplementation(libs.ktor.server.testJvm)
     testImplementation(libs.ktor.client.mock)
+    testImplementation(libs.kafka.clients)
 }
 
 java {

diff --git a/...-generator/src/test/kotlin/no/nav/paw/kafkakeygenerator/RecordKeyToPartitionCalculator.kt b/...-generator/src/test/kotlin/no/nav/paw/kafkakeygenerator/RecordKeyToPartitionCalculator.kt
@@ -0,0 +1,12 @@
+package no.nav.paw.kafkakeygenerator
+
+import org.apache.kafka.clients.producer.internals.BuiltInPartitioner
+import org.apache.kafka.common.serialization.Serdes
+
+const val NUMBER_OF_PARTITIONS = 6
+
+fun main() {
+    val recordKey: Long = 123
+    val partition = BuiltInPartitioner.partitionForKey(Serdes.Long().serializer().serialize("topic", recordKey), NUMBER_OF_PARTITIONS)
+    println("Record key $recordKey is assigned to partition $partition")
+}