More logging

Author: tconley1428

temporalio/bridge/worker.py

@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import logging
 from dataclasses import dataclass
 from typing import (
     TYPE_CHECKING,
@@ -42,6 +43,7 @@
 )
 from temporalio.bridge.temporal_sdk_bridge import PollShutdownError  # type: ignore
 
+logger = logging.getLogger(__name__)
 
 @dataclass
 class WorkerConfig:
@@ -219,6 +221,8 @@ async def poll_activity_task(
         self,
     ) -> temporalio.bridge.proto.activity_task.ActivityTask:
         """Poll for an activity task."""
+        logger.info("Polling for activity task")
+
         return temporalio.bridge.proto.activity_task.ActivityTask.FromString(
             await self._ref.poll_activity_task()
         )

temporalio/worker/_activity.py

@@ -169,10 +169,13 @@ async def raise_from_exception_queue() -> NoReturn:
                 else:
                     raise RuntimeError(f"Unrecognized activity task: {task}")
             except temporalio.bridge.worker.PollShutdownError:  # type: ignore[reportPrivateLocalImportUsage]
+                logger.info("Activity worker finished1")
                 exception_task.cancel()
+                logger.info("Activity worker finished2")
                 return
             except Exception as err:
                 exception_task.cancel()
+                logger.info("Activity worker failed: %s", err)
                 raise RuntimeError("Activity worker failed") from err
 
     def notify_shutdown(self) -> None: