aws-durable-execution-sdk-python-testing 1.0.0__py3-none-any.whl

aws_durable_execution_sdk_python_testing/observer.py

@@ -0,0 +1,144 @@
+"""Checkpoint processors can notify the Execution of notable event state changes. Observer pattern."""
+
+from __future__ import annotations
+
+import threading
+from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
+
+from aws_durable_execution_sdk_python_testing.token import CallbackToken
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+    from aws_durable_execution_sdk_python.lambda_service import (
+        ErrorObject,
+        CallbackOptions,
+    )
+
+
+class ExecutionObserver(ABC):
+    """Observer for execution lifecycle events."""
+
+    @abstractmethod
+    def on_completed(self, execution_arn: str, result: str | None = None) -> None:
+        """Called when execution completes successfully."""
+
+    @abstractmethod
+    def on_failed(self, execution_arn: str, error: ErrorObject) -> None:
+        """Called when execution fails."""
+
+    @abstractmethod
+    def on_timed_out(self, execution_arn: str, error: ErrorObject) -> None:
+        """Called when execution times out."""
+
+    @abstractmethod
+    def on_stopped(self, execution_arn: str, error: ErrorObject) -> None:
+        """Called when execution is stopped."""
+
+    @abstractmethod
+    def on_wait_timer_scheduled(
+        self, execution_arn: str, operation_id: str, delay: float
+    ) -> None:
+        """Called when wait timer scheduled."""
+
+    @abstractmethod
+    def on_step_retry_scheduled(
+        self, execution_arn: str, operation_id: str, delay: float
+    ) -> None:
+        """Called when step retry scheduled."""
+
+    @abstractmethod
+    def on_callback_created(
+        self,
+        execution_arn: str,
+        operation_id: str,
+        callback_options: CallbackOptions | None,
+        callback_token: CallbackToken,
+    ) -> None:
+        """Called when callback is created."""
+
+
+class ExecutionNotifier:
+    """Notifies observers about execution events. Thread-safe."""
+
+    def __init__(self) -> None:
+        self._observers: list[ExecutionObserver] = []
+        self._lock = threading.RLock()
+
+    def add_observer(self, observer: ExecutionObserver) -> None:
+        """Add an observer to be notified of execution events."""
+        with self._lock:
+            self._observers.append(observer)
+
+    def _notify_observers(self, method: Callable, *args, **kwargs) -> None:
+        """Notify all observers by calling the specified method."""
+        with self._lock:
+            observers = self._observers.copy()
+        for observer in observers:
+            getattr(observer, method.__name__)(*args, **kwargs)
+
+    # region event emitters
+    def notify_completed(self, execution_arn: str, result: str | None = None) -> None:
+        """Notify observers about execution completion."""
+        self._notify_observers(
+            ExecutionObserver.on_completed, execution_arn=execution_arn, result=result
+        )
+
+    def notify_failed(self, execution_arn: str, error: ErrorObject) -> None:
+        """Notify observers about execution failure."""
+        self._notify_observers(
+            ExecutionObserver.on_failed, execution_arn=execution_arn, error=error
+        )
+
+    def notify_timed_out(self, execution_arn: str, error: ErrorObject) -> None:
+        """Notify observers about execution timeout."""
+        self._notify_observers(
+            ExecutionObserver.on_timed_out, execution_arn=execution_arn, error=error
+        )
+
+    def notify_stopped(self, execution_arn: str, error: ErrorObject) -> None:
+        """Notify observers about execution being stopped."""
+        self._notify_observers(
+            ExecutionObserver.on_stopped, execution_arn=execution_arn, error=error
+        )
+
+    def notify_wait_timer_scheduled(
+        self, execution_arn: str, operation_id: str, delay: float
+    ) -> None:
+        """Notify observers about wait timer scheduling."""
+        self._notify_observers(
+            ExecutionObserver.on_wait_timer_scheduled,
+            execution_arn=execution_arn,
+            operation_id=operation_id,
+            delay=delay,
+        )
+
+    def notify_step_retry_scheduled(
+        self, execution_arn: str, operation_id: str, delay: float
+    ) -> None:
+        """Notify observers about step retry scheduling."""
+        self._notify_observers(
+            ExecutionObserver.on_step_retry_scheduled,
+            execution_arn=execution_arn,
+            operation_id=operation_id,
+            delay=delay,
+        )
+
+    def notify_callback_created(
+        self,
+        execution_arn: str,
+        operation_id: str,
+        callback_options: CallbackOptions | None,
+        callback_token: CallbackToken,
+    ) -> None:
+        """Notify observers about callback creation."""
+        self._notify_observers(
+            ExecutionObserver.on_callback_created,
+            execution_arn=execution_arn,
+            operation_id=operation_id,
+            callback_options=callback_options,
+            callback_token=callback_token,
+        )
+
+    # endregion event emitters

aws_durable_execution_sdk_python_testing/py.typed

@@ -0,0 +1 @@
+# Marker file that indicates this package supports typing