hexkit 3.2.0__tar.gz → 3.2.2__tar.gz

{hexkit-3.2.0/src/hexkit.egg-info → hexkit-3.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hexkit
-Version: 3.2.0
+Version: 3.2.2
 Summary: A Toolkit for Building Microservices using the Hexagonal Architecture
 Author-email: "German Human Genome Phenome Archive (GHGA)" <contact@ghga.de>
 License: Apache 2.0

{hexkit-3.2.0 → hexkit-3.2.2}/pyproject.toml

@@ -23,7 +23,7 @@ classifiers = [
     "Intended Audience :: Developers",
 ]
 name = "hexkit"
-version = "3.2.0"
+version = "3.2.2"
 description = "A Toolkit for Building Microservices using the Hexagonal Architecture"
 dependencies = [
     "pydantic >=2, <3",

{hexkit-3.2.0 → hexkit-3.2.2}/src/hexkit/correlation.py

@@ -29,6 +29,7 @@ correlation_id_var: ContextVar[str] = ContextVar("correlation_id", default="")
 
 __all__ = [
     "set_correlation_id",
+    "set_new_correlation_id",
     "get_correlation_id",
     "new_correlation_id",
     "validate_correlation_id",
@@ -40,6 +41,9 @@ __all__ = [
 class CorrelationIdContextError(RuntimeError):
     """Raised when the correlation ID ContextVar is unexpectedly not set."""
 
+    def __init__(self):
+        super().__init__("No correlation ID has been set.")
+
 
 class InvalidCorrelationIdError(RuntimeError):
     """Raised when a correlation ID fails validation."""
@@ -68,7 +72,7 @@ def validate_correlation_id(correlation_id: str):
 
 @asynccontextmanager
 async def set_correlation_id(correlation_id: str):
-    """Set the correlation ID for the life of the context.
+    """Set the given correlation ID for the life of the context.
 
     Raises:
         InvalidCorrelationIdError: when the correlation ID is empty or invalid.
@@ -80,6 +84,16 @@ async def set_correlation_id(correlation_id: str):
         yield
 
 
+@asynccontextmanager
+async def set_new_correlation_id():
+    """Set a new correlation ID for the life of the context."""
+    correlation_id = new_correlation_id()
+
+    async with set_context_var(correlation_id_var, correlation_id):
+        log.info("Set context correlation ID to %s", correlation_id)
+        yield correlation_id
+
+
 def get_correlation_id() -> str:
     """Get the correlation ID.
 

{hexkit-3.2.0 → hexkit-3.2.2}/src/hexkit/providers/mongokafka/provider.py

@@ -251,7 +251,7 @@ class MongoKafkaDaoPublisher(Generic[Dto]):
         """
         correlation_id = get_correlation_id()
         document = self._dao._dto_to_document(dto)
-        document["__metadata__.correlation_id"] = correlation_id
+        document.setdefault("__metadata__", {})["correlation_id"] = correlation_id
         result = await self._collection.replace_one(
             {"_id": document["_id"], "__metadata__.deleted": False}, document
         )

{hexkit-3.2.0 → hexkit-3.2.2/src/hexkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hexkit
-Version: 3.2.0
+Version: 3.2.2
 Summary: A Toolkit for Building Microservices using the Hexagonal Architecture
 Author-email: "German Human Genome Phenome Archive (GHGA)" <contact@ghga.de>
 License: Apache 2.0