use object reference instead of function, picklable

Author: THardy98

temporalio/activity.py

@@ -174,7 +174,7 @@ class _Context:
         temporalio.converter.PayloadConverter,
     ]
     runtime_metric_meter: Optional[temporalio.common.MetricMeter]
-    cancellation_details: Callable[[], Optional[ActivityCancellationDetails]]
+    cancellation_details: Optional[ActivityCancellationDetails] = None
     _logger_details: Optional[Mapping[str, Any]] = None
     _payload_converter: Optional[temporalio.converter.PayloadConverter] = None
     _metric_meter: Optional[temporalio.common.MetricMeter] = None
@@ -289,7 +289,7 @@ def info() -> Info:
 
 def cancellation_details() -> Optional[ActivityCancellationDetails]:
     """Cancellation details of the currenct activity, if any"""
-    return _Context.current().cancellation_details()
+    return _Context.current().cancellation_details
 
 
 def heartbeat(*details: Any) -> None:

temporalio/testing/_activity.py

@@ -168,7 +168,7 @@ def __init__(
             else self.cancel_thread_raiser.shielded,
             payload_converter_class_or_instance=env.payload_converter,
             runtime_metric_meter=env.metric_meter,
-            cancellation_details=lambda: env.cancellation_details,
+            cancellation_details=env.cancellation_details,
         )
         self.task: Optional[asyncio.Task] = None
 

temporalio/worker/_activity.py

@@ -214,9 +214,8 @@ def _cancel(
             warnings.warn(f"Cannot find activity to cancel for token {task_token!r}")
             return
         logger.debug("Cancelling activity %s, reason: %s", task_token, cancel.reason)
-        activity.cancellation_details = (
-            temporalio.activity.ActivityCancellationDetails._fromProto(cancel.details)
-        )
+        activity.cancellation_details.cancelled = cancel.details.is_cancelled
+        activity.cancellation_details.paused = cancel.details.is_paused
         activity.cancel(cancelled_by_request=True)
 
     def _heartbeat(self, task_token: bytes, *details: Any) -> None:
@@ -424,7 +423,8 @@ async def _run_activity(
                 headers=start.header_fields,
             )
 
-            running_activity.cancellation_details = None
+            cancellation_details = temporalio.activity.ActivityCancellationDetails()
+            running_activity.cancellation_details = cancellation_details
 
             # Set the context early so the logging adapter works and
             # interceptors have it
@@ -442,7 +442,7 @@ async def _run_activity(
                     if sync_non_threaded
                     else self._metric_meter,
                     # Function reference to the running activity's cancellation details
-                    cancellation_details=lambda: running_activity.cancellation_details,
+                    cancellation_details=cancellation_details,
                 )
             )
             temporalio.activity.logger.debug("Starting activity")
@@ -553,8 +553,8 @@ class _RunningActivity:
     done: bool = False
     cancelled_by_request: bool = False
     cancelled_due_to_heartbeat_error: Optional[Exception] = None
-    cancellation_details: Optional[temporalio.activity.ActivityCancellationDetails] = (
-        None
+    cancellation_details: temporalio.activity.ActivityCancellationDetails = (
+        temporalio.activity.ActivityCancellationDetails()
     )
 
     def cancel(
@@ -751,9 +751,7 @@ def _execute_sync_activity(
         temporalio.converter.PayloadConverter,
     ],
     runtime_metric_meter: Optional[temporalio.common.MetricMeter],
-    cancellation_details: Callable[
-        [], Optional[temporalio.activity.ActivityCancellationDetails]
-    ],
+    cancellation_details: Optional[temporalio.activity.ActivityCancellationDetails],
     fn: Callable[..., Any],
     *args: Any,
 ) -> Any: