More logs on worker shutdown

Author: tconley1428

temporalio/worker/_worker.py

@@ -755,6 +755,8 @@ async def raise_on_shutdown():
         # Initiate core worker shutdown
         self._bridge_worker.initiate_shutdown()
 
+        logger.info("Bridge shut down")
+
         # If any worker task had an exception, replace that task with a queue drain
         for worker, task in tasks.items():
             if worker and task.done() and task.exception():
@@ -768,12 +770,15 @@ async def raise_on_shutdown():
 
         # Wait for all tasks to complete (i.e. for poller loops to stop)
         await asyncio.wait(tasks.values())
+        logger.info("tasks waited")
+
         # Sometimes both workers throw an exception and since we only take the
         # first, Python may complain with "Task exception was never retrieved"
         # if we don't get the others. Therefore we call cancel on each task
         # which suppresses this.
         for task in tasks.values():
             task.cancel()
+        logger.info("tasks cancelled")
 
         # Let all activity / nexus operations completions finish. We cannot guarantee that
         # because poll shutdown completed (which means activities/operations completed)
@@ -782,6 +787,7 @@ async def raise_on_shutdown():
             await self._activity_worker.wait_all_completed()
         if self._nexus_worker:
             await self._nexus_worker.wait_all_completed()
+        logger.info("Waited for workers")
 
         # Do final shutdown
         try:
@@ -790,6 +796,7 @@ async def raise_on_shutdown():
             # Ignore errors here that can arise in some tests where the bridge
             # worker still has a reference
             pass
+        logger.info("Set shutdown complete")
 
         # Mark as shutdown complete and re-raise exception if present
         self._shutdown_complete_event.set()
@@ -847,6 +854,8 @@ async def __aexit__(self, exc_type: Optional[Type[BaseException]], *args) -> Non
         if not self._async_context_run_task:
             raise RuntimeError("Never started")
         await self.shutdown()
+        logger.info("Shutdown complete")
+
         # Cancel our run task
         self._async_context_run_task.cancel()
         # Only re-raise our exception if present and exc_type is cancel