Merge pull request #254 from flix-tech/feature-enable-key-serialization

[RdKafka] Enable serializers to serialize message keys

Author: makasim

pkg/rdkafka/RdKafkaProducer.php

@@ -41,8 +41,8 @@ public function send(PsrDestination $destination, PsrMessage $message)
         InvalidMessageException::assertMessageInstanceOf($message, RdKafkaMessage::class);
 
         $partition = $message->getPartition() ?: $destination->getPartition() ?: RD_KAFKA_PARTITION_UA;
-        $key = $message->getKey() ?: $destination->getKey() ?: null;
         $payload = $this->serializer->toString($message);
+        $key = $message->getKey() ?: $destination->getKey() ?: null;
 
         $topic = $this->producer->newTopic($destination->getTopicName(), $destination->getConf());
         $topic->produce($partition, 0 /* must be 0 */, $payload, $key);

pkg/rdkafka/Tests/RdKafkaProducerTest.php

@@ -95,6 +95,45 @@ public function testShouldAllowGetPreviouslySetSerializer()
         $this->assertSame($expectedSerializer, $producer->getSerializer());
     }
 
+    public function testShouldAllowSerializersToSerializeKeys()
+    {
+        $message = new RdKafkaMessage('theBody', ['foo' => 'fooVal'], ['bar' => 'barVal']);
+        $message->setKey('key');
+
+        $kafkaTopic = $this->createKafkaTopicMock();
+        $kafkaTopic
+            ->expects($this->once())
+            ->method('produce')
+            ->with(
+                RD_KAFKA_PARTITION_UA,
+                0,
+                'theSerializedMessage',
+                'theSerializedKey'
+            )
+        ;
+
+        $kafkaProducer = $this->createKafkaProducerMock();
+        $kafkaProducer
+            ->expects($this->once())
+            ->method('newTopic')
+            ->willReturn($kafkaTopic)
+        ;
+
+        $serializer = $this->createSerializerMock();
+        $serializer
+            ->expects($this->once())
+            ->method('toString')
+            ->willReturnCallback(function () use ($message) {
+                $message->setKey('theSerializedKey');
+
+                return 'theSerializedMessage';
+            })
+        ;
+
+        $producer = new RdKafkaProducer($kafkaProducer, $serializer);
+        $producer->send(new RdKafkaTopic('theQueueName'), $message);
+    }
+
     /**
      * @return \PHPUnit_Framework_MockObject_MockObject|ProducerTopic
      */