buz 2.22.0__tar.gz → 2.22.1__tar.gz

{buz-2.22.0 → buz-2.22.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.22.0
+Version: 2.22.1
 Summary: Buz is a set of light, simple and extensible implementations of event, command and query buses.
 License: MIT
 Author: Luis Pintado Lozano

{buz-2.22.0 → buz-2.22.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "buz"
-version = "2.22.0"
+version = "2.22.1"
 description = "Buz is a set of light, simple and extensible implementations of event, command and query buses."
 readme = "README.md"
 authors = [

{buz-2.22.0 → buz-2.22.1}/src/buz/event/infrastructure/buz_kafka/buz_kafka_event_bus.py

@@ -16,6 +16,8 @@ from buz.kafka.domain.models.auto_create_topic_configuration import AutoCreateTo
 from buz.kafka.domain.models.create_kafka_topic import CreateKafkaTopic
 from buz.kafka.domain.services.kafka_admin_client import KafkaAdminClient
 from buz.kafka.domain.services.kafka_producer import KafkaProducer
+from buz.kafka.infrastructure.serializers.implementations.cdc_partition_key_serializer import CDCPartitionKeySerializer
+from buz.kafka.infrastructure.serializers.partitiion_key_generator import PartitionKeySerializer
 
 
 class BuzKafkaEventBus(EventBus):
@@ -28,6 +30,7 @@ class BuzKafkaEventBus(EventBus):
         kafka_admin_client: Optional[KafkaAdminClient] = None,
         publish_middlewares: Optional[list[PublishMiddleware]] = None,
         auto_create_topic_configuration: Optional[AutoCreateTopicConfiguration] = None,
+        partition_key_generator: Optional[PartitionKeySerializer] = None,
     ):
         self.__publish_middleware_chain_resolver = PublishMiddlewareChainResolver(publish_middlewares or [])
         self.__publish_strategy = publish_strategy
@@ -36,6 +39,7 @@ class BuzKafkaEventBus(EventBus):
         self.__kafka_admin_client = kafka_admin_client
         self.__auto_create_topic_configuration = auto_create_topic_configuration
         self.__logger = logger
+        self.__partition_key_generator: PartitionKeySerializer = partition_key_generator or CDCPartitionKeySerializer()
         self.__check_kafka_admin_client_is_needed()
 
     def __check_kafka_admin_client_is_needed(self) -> None:
@@ -74,7 +78,7 @@ class BuzKafkaEventBus(EventBus):
                 message=event,
                 headers=headers,
                 topic=topic,
-                partition_key=event.partition_key(),
+                partition_key=self.__partition_key_generator.generate_key(event),
             )
         except Exception as exc:
             raise EventNotPublishedException(event) from exc

buz-2.22.1/src/buz/kafka/infrastructure/serializers/implementations/cdc_partition_key_serializer.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+from buz.event.event import Event
+from buz.kafka.infrastructure.serializers.implementations.json_byte_serializer import JSONByteSerializer
+from buz.kafka.infrastructure.serializers.partitiion_key_generator import PartitionKeySerializer
+
+# This is a static string because the order matters and we can not trust that json encoder libraries are deterministic
+PAYLOAD_CDC_SCHEMA = r"""{"schema":{"type":"string","optional":true},"payload":"[partition_key]"}"""
+
+
+class CDCPartitionKeySerializer(PartitionKeySerializer):
+    def __init__(self) -> None:
+        self.__json_serializer = JSONByteSerializer()
+
+    def __generate_payload_schema(self, partition_key: str) -> str:
+        return PAYLOAD_CDC_SCHEMA.replace("[partition_key]", partition_key)
+
+    def generate_key(self, event: Event) -> str:
+        return self.__generate_payload_schema(event.partition_key())

buz-2.22.1/src/buz/kafka/infrastructure/serializers/partitiion_key_generator.py

@@ -0,0 +1,9 @@
+from abc import ABC, abstractmethod
+
+from buz.event.event import Event
+
+
+class PartitionKeySerializer(ABC):
+    @abstractmethod
+    def generate_key(self, event: Event) -> str:
+        pass