Add logging for input and output event streams

Author: jonathan343

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)
 
 
@@ -45,13 +48,15 @@ async def receive(self) -> E | None:
         event = await Event.decode_async(self._source)
         if event is None:
             return None
+        logger.debug("Received raw event message: %s", event)
 
         deserializer = EventDeserializer(
             event=event,
             payload_codec=self._payload_codec,
             is_client_mode=self._is_client_mode,
         )
         result = self._deserializer(deserializer)
+        logger.debug("Deserialized event message: %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,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 import asyncio
 import datetime
+import logging
 from collections.abc import Callable, Iterator
 from contextlib import contextmanager
 from io import BytesIO
@@ -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"
 
@@ -52,6 +55,7 @@ def __init__(
         )
 
     async def send(self, event: E) -> None:
+        logger.debug("Raw event message: %s", event)
         event.serialize(self._serializer)
         result = self._serializer.get_result()
         if result is None:
@@ -60,6 +64,7 @@ async def send(self, event: E) -> None:
             )
         if self._signer is not None:
             result = self._signer(result)
+        logger.debug("Writing event message: %s", result)
         await self._writer.write(result.encode())
 
     async def close(self) -> None: