Add update info to logging extra

Author: dandavison

temporalio/workflow.py

@@ -11,7 +11,7 @@
 import warnings
 from abc import ABC, abstractmethod
 from contextlib import contextmanager
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 from datetime import datetime, timedelta, timezone
 from enum import Enum, IntEnum
 from functools import partial
@@ -500,6 +500,14 @@ class UpdateInfo:
     name: str
     """Update type name."""
 
+    @property
+    def logger_details(self) -> Mapping[str, Any]:
+        """Data to be included in string appended to default logging output."""
+        return {
+            "update_id": self.id,
+            "update_name": self.name,
+        }
+
 
 class _Runtime(ABC):
     @staticmethod
@@ -1205,6 +1213,10 @@ class LoggerAdapter(logging.LoggerAdapter):
             dictionary value will be added to the ``extra`` dictionary with some
             workflow info, making it present on the ``LogRecord.__dict__`` for
             use by others. Default is True.
+        update_info_on_extra: Boolean for whether a ``temporal_update``
+            dictionary value will be added to the ``extra`` dictionary with some
+            update info, making it present on the ``LogRecord.__dict__`` for use
+            by others. Default is True.
         full_workflow_info_on_extra: Boolean for whether a ``workflow_info``
             value will be added to the ``extra`` dictionary with the entire
             workflow info, making it present on the ``LogRecord.__dict__`` for
@@ -1224,6 +1236,7 @@ def __init__(
         super().__init__(logger, extra or {})
         self.workflow_info_on_message = True
         self.workflow_info_on_extra = True
+        self.update_info_on_extra = True
         self.full_workflow_info_on_extra = False
         self.log_during_replay = False
 
@@ -1234,10 +1247,11 @@ def process(
         if (
             self.workflow_info_on_message
             or self.workflow_info_on_extra
+            or self.update_info_on_extra
             or self.full_workflow_info_on_extra
         ):
-            extra = {}
-            msg_extra = {}
+            extra: Dict[str, Any] = {}
+            msg_extra: Dict[str, Any] = {}
             runtime = _Runtime.maybe_current()
             if runtime:
                 if self.workflow_info_on_message:
@@ -1246,6 +1260,12 @@ def process(
                     extra["temporal_workflow"] = runtime.logger_details
                 if self.full_workflow_info_on_extra:
                     extra["workflow_info"] = runtime.workflow_info()
+            update_info = current_update_info()
+            if update_info:
+                if self.update_info_on_extra:
+                    extra["temporal_update"] = asdict(update_info)
+                if self.workflow_info_on_message:
+                    msg_extra.update(update_info.logger_details)
 
         kwargs["extra"] = {**extra, **(kwargs.get("extra") or {})}
         if msg_extra: