Merge branch 'develop' into sign_event

Author: nateprewitt

codegen/gradle/libs.versions.toml

@@ -1,11 +1,11 @@
 [versions]
-junit5 = "5.12.0"
+junit5 = "5.12.1"
 smithy = "1.55.0"
 test-logger-plugin = "4.0.0"
 spotbugs = "6.0.22"
 spotless = "7.0.2"
 smithy-gradle-plugins = "1.2.0"
-dep-analysis = "2.11.0"
+dep-analysis = "2.12.0"
 
 [libraries]
 smithy-model = { module = "software.amazon.smithy:smithy-model", version.ref = "smithy" }

packages/aws-event-stream/src/aws_event_stream/_private/deserializers.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import asyncio
 import datetime
+import logging
 from collections.abc import Callable
 
 from smithy_core.aio.interfaces import AsyncByteStream
@@ -25,6 +26,8 @@
 )
 from smithy_core.traits import EventHeaderTrait
 
+logger = logging.getLogger(__name__)
+
 INITIAL_MESSAGE_TYPES = (INITIAL_REQUEST_EVENT_TYPE, INITIAL_RESPONSE_EVENT_TYPE)
 
 
@@ -55,14 +58,17 @@ async def receive(self) -> E | None:
             raise
 
         if event is None:
+            logger.debug("No event received from the source.")
             return None
+        logger.debug("Received raw event: %s", event)
 
         deserializer = EventDeserializer(
             event=event,
             payload_codec=self._payload_codec,
             is_client_mode=self._is_client_mode,
         )
         result = self._deserializer(deserializer)
+        logger.debug("Successfully deserialized event: %s", result)
         if isinstance(getattr(result, "value"), Exception):
             raise result.value  # type: ignore
         return result

packages/aws-event-stream/src/aws_event_stream/_private/serializers.py

@@ -2,7 +2,8 @@
 # SPDX-License-Identifier: Apache-2.0
 import asyncio
 import datetime
-from collections.abc import Iterator
+import logging
+from collections.abc import Callable, Iterator
 from contextlib import contextmanager
 from io import BytesIO
 from typing import Never, Protocol
@@ -29,6 +30,8 @@
 )
 from smithy_core.traits import ErrorTrait, EventHeaderTrait, MediaTypeTrait
 
+logger = logging.getLogger(__name__)
+
 _DEFAULT_STRING_CONTENT_TYPE = "text/plain"
 _DEFAULT_BLOB_CONTENT_TYPE = "application/octet-stream"
 
@@ -62,6 +65,7 @@ def __init__(
     async def send(self, event: E) -> None:
         if self._closed:
             raise IOError("Attempted to write to closed stream.")
+        logger.debug("Preparing to publish event: %s", event)
         event.serialize(self._serializer)
         result = self._serializer.get_result()
         if result is None:
@@ -78,6 +82,7 @@ async def send(self, event: E) -> None:
         assert isinstance(result, EventMessage)
         encoded_result: bytes = result.encode()
         try:
+            logger.debug("Publishing serialized event: %s", result)
             await self._writer.write(encoded_result)
         except Exception as e:
             await self.close()