durabletask 0.0.0.dev4__tar.gz → 0.0.0.dev6__tar.gz

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev4
+Version: 0.0.0.dev6
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask/entities/entity_instance_id.py

@@ -1,6 +1,3 @@
-from typing import Optional
-
-
 class EntityInstanceId:
     def __init__(self, entity: str, key: str):
         self.entity = entity
@@ -30,8 +27,13 @@ class EntityInstanceId:
 
         Returns
         -------
-        Optional[EntityInstanceId]
-            The parsed EntityInstanceId object, or None if the input is None.
+        EntityInstanceId
+            The parsed EntityInstanceId object.
+
+        Raises
+        ------
+        ValueError
+            If the input string is not in the correct format.
         """
         try:
             _, entity, key = entity_id.split("@", 2)

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask/entities/entity_metadata.py

@@ -44,8 +44,9 @@ class EntityMetadata:
 
     @staticmethod
     def from_entity_response(entity_response: pb.GetEntityResponse, includes_state: bool):
-        entity_id = EntityInstanceId.parse(entity_response.entity.instanceId)
-        if not entity_id:
+        try:
+            entity_id = EntityInstanceId.parse(entity_response.entity.instanceId)
+        except ValueError:
             raise ValueError("Invalid entity instance ID in entity response.")
         entity_state = None
         if includes_state:

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask/internal/helpers.py

@@ -199,8 +199,9 @@ def new_schedule_task_action(id: int, name: str, encoded_input: Optional[str],
 def new_call_entity_action(id: int,
                            parent_instance_id: str,
                            entity_id: EntityInstanceId,
-                           operation: str, encoded_input: Optional[str],
-                           request_id: str):
+                           operation: str,
+                           encoded_input: Optional[str],
+                           request_id: str) -> pb.OrchestratorAction:
     return pb.OrchestratorAction(id=id, sendEntityMessage=pb.SendEntityMessageAction(entityOperationCalled=pb.EntityOperationCalledEvent(
         requestId=request_id,
         operation=operation,
@@ -216,7 +217,7 @@ def new_signal_entity_action(id: int,
                              entity_id: EntityInstanceId,
                              operation: str,
                              encoded_input: Optional[str],
-                             request_id: str):
+                             request_id: str) -> pb.OrchestratorAction:
     return pb.OrchestratorAction(id=id, sendEntityMessage=pb.SendEntityMessageAction(entityOperationSignaled=pb.EntityOperationSignaledEvent(
         requestId=request_id,
         operation=operation,

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask/task.py

@@ -139,7 +139,8 @@ class OrchestrationContext(ABC):
         pass
 
     @abstractmethod
-    def call_entity(self, entity: EntityInstanceId,
+    def call_entity(self,
+                    entity: EntityInstanceId,
                     operation: str,
                     input: Optional[TInput] = None) -> Task:
         """Schedule entity function for execution.
@@ -264,8 +265,8 @@ class OrchestrationContext(ABC):
 
         The default implementation of this method creates a name-based UUID
         using the algorithm from RFC 4122 §4.3. The name input used to generate
-        this value is a combination of the orchestration instance ID and an
-        internally managed sequence number.
+        this value is a combination of the orchestration instance ID, the current UTC datetime,
+        and an internally managed counter.
 
         Returns
         -------

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask/worker.py

@@ -12,7 +12,7 @@ from datetime import datetime, timedelta, timezone
 from threading import Event, Thread
 from types import GeneratorType
 from enum import Enum
-from typing import Any, Generator, Optional, Sequence, TypeVar, Union
+from typing import Any, Generator, Optional, Sequence, Tuple, TypeVar, Union
 import uuid
 from packaging.version import InvalidVersion, parse
 
@@ -20,7 +20,6 @@ import grpc
 from google.protobuf import empty_pb2
 
 from durabletask.internal import helpers
-from durabletask.internal.proto_task_hub_sidecar_service_stub import ProtoTaskHubSidecarServiceStub
 from durabletask.internal.entity_state_shim import StateShim
 from durabletask.internal.helpers import new_timestamp
 from durabletask.entities import DurableEntity, EntityLock, EntityInstanceId, EntityContext
@@ -632,7 +631,7 @@ class TaskHubGrpcWorker:
     def _execute_orchestrator(
             self,
             req: pb.OrchestratorRequest,
-            stub: Union[stubs.TaskHubSidecarServiceStub, ProtoTaskHubSidecarServiceStub],
+            stub: stubs.TaskHubSidecarServiceStub,
             completionToken,
     ):
         try:
@@ -739,10 +738,9 @@ class TaskHubGrpcWorker:
     def _execute_entity_batch(
             self,
             req: Union[pb.EntityBatchRequest, pb.EntityRequest],
-            stub: Union[stubs.TaskHubSidecarServiceStub, ProtoTaskHubSidecarServiceStub],
+            stub: stubs.TaskHubSidecarServiceStub,
             completionToken,
     ):
-        operation_infos = None
         if isinstance(req, pb.EntityRequest):
             req, operation_infos = helpers.convert_to_entity_batch_request(req)
 
@@ -754,9 +752,10 @@ class TaskHubGrpcWorker:
         for operation in req.operations:
             start_time = datetime.now(timezone.utc)
             executor = _EntityExecutor(self._registry, self._logger)
-            entity_instance_id = EntityInstanceId.parse(instance_id)
-            if not entity_instance_id:
-                raise RuntimeError(f"Invalid entity instance ID '{operation.requestId}' in entity operation request.")
+            try:
+                entity_instance_id = EntityInstanceId.parse(instance_id)
+            except ValueError:
+                raise RuntimeError(f"Invalid entity instance ID '{instance_id}' in entity operation request.")
 
             operation_result = None
 
@@ -798,7 +797,7 @@ class TaskHubGrpcWorker:
             stub.CompleteEntityTask(batch_result)
         except Exception as ex:
             self._logger.exception(
-                f"Failed to deliver entity response for orchestration ID '{instance_id}' to sidecar: {ex}"
+                f"Failed to deliver entity response for '{entity_instance_id}' of orchestration ID '{instance_id}' to sidecar: {ex}"
             )
 
         # TODO: Reset context
@@ -831,7 +830,8 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
         self._pending_actions: dict[int, pb.OrchestratorAction] = {}
         self._pending_tasks: dict[int, task.CompletableTask] = {}
         # Maps entity ID to task ID
-        self._entity_task_id_map: dict[str, tuple[EntityInstanceId, int, Optional[str]]] = {}
+        self._entity_task_id_map: dict[str, tuple[EntityInstanceId, int]] = {}
+        self._entity_lock_task_id_map: dict[str, tuple[EntityInstanceId, int]] = {}
         # Maps criticalSectionId to task ID
         self._entity_lock_id_map: dict[str, int] = {}
         self._sequence_number = 0
@@ -1170,12 +1170,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             raise RuntimeError(error_message)
 
         encoded_input = shared.to_json(input) if input is not None else None
-        action = ph.new_call_entity_action(id,
-                                           self.instance_id,
-                                           entity_id,
-                                           operation,
-                                           encoded_input,
-                                           self.new_uuid())
+        action = ph.new_call_entity_action(id, self.instance_id, entity_id, operation, encoded_input, self.new_uuid())
         self._pending_actions[id] = action
 
         fn_task = task.CompletableTask()
@@ -1262,14 +1257,14 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
         self.set_continued_as_new(new_input, save_events)
 
     def new_uuid(self) -> str:
-        URL_NAMESPACE: str = "9e952958-5e33-4daf-827f-2fa12937b875"
+        NAMESPACE_UUID: str = "9e952958-5e33-4daf-827f-2fa12937b875"
 
         uuid_name_value = \
             f"{self._instance_id}" \
             f"_{self.current_utc_datetime.strftime(DATETIME_STRING_FORMAT)}" \
             f"_{self._new_uuid_counter}"
         self._new_uuid_counter += 1
-        namespace_uuid = uuid.uuid5(uuid.NAMESPACE_OID, URL_NAMESPACE)
+        namespace_uuid = uuid.uuid5(uuid.NAMESPACE_OID, NAMESPACE_UUID)
         return str(uuid.uuid5(namespace_uuid, uuid_name_value))
 
 
@@ -1612,32 +1607,11 @@ class _OrchestrationExecutor:
                     raise TypeError("Unexpected sub-orchestration task type")
             elif event.HasField("eventRaised"):
                 if event.eventRaised.name in ctx._entity_task_id_map:
-                    # This eventRaised represents the result of an entity operation after being translated to the old
-                    # entity protocol by the Durable WebJobs extension
-                    entity_id, task_id, action_type = ctx._entity_task_id_map.get(event.eventRaised.name, (None, None, None))
-                    if entity_id is None:
-                        raise RuntimeError(f"Could not retrieve entity ID for entity-related eventRaised with ID '{event.eventId}'")
-                    if task_id is None:
-                        raise RuntimeError(f"Could not retrieve task ID for entity-related eventRaised with ID '{event.eventId}'")
-                    entity_task = ctx._pending_tasks.pop(task_id, None)
-                    if not entity_task:
-                        raise RuntimeError(f"Could not retrieve entity task for entity-related eventRaised with ID '{event.eventId}'")
-                    result = None
-                    if not ph.is_empty(event.eventRaised.input):
-                        # TODO: Investigate why the event result is wrapped in a dict with "result" key
-                        result = shared.from_json(event.eventRaised.input.value)["result"]
-                    if action_type == "entityOperationCalled":
-                        ctx._entity_context.recover_lock_after_call(entity_id)
-                        entity_task.complete(result)
-                        ctx.resume()
-                    elif action_type == "entityLockRequested":
-                        ctx._entity_context.complete_acquire(event.eventRaised.name)
-                        entity_task.complete(EntityLock(ctx))
-                        ctx.resume()
-                    else:
-                        raise RuntimeError(f"Unknown action type '{action_type}' for entity-related eventRaised "
-                                           f"with ID '{event.eventId}'")
-
+                    entity_id, task_id = ctx._entity_task_id_map.get(event.eventRaised.name, (None, None))
+                    self._handle_entity_event_raised(ctx, event, entity_id, task_id, False)
+                elif event.eventRaised.name in ctx._entity_lock_task_id_map:
+                    entity_id, task_id = ctx._entity_lock_task_id_map.get(event.eventRaised.name, (None, None))
+                    self._handle_entity_event_raised(ctx, event, entity_id, task_id, True)
                 else:
                     # event names are case-insensitive
                     event_name = event.eventRaised.name.casefold()
@@ -1705,10 +1679,11 @@ class _OrchestrationExecutor:
                     raise _get_wrong_action_type_error(
                         entity_call_id, expected_method_name, action
                     )
-                entity_id = EntityInstanceId.parse(event.entityOperationCalled.targetInstanceId.value)
-                if not entity_id:
+                try:
+                    entity_id = EntityInstanceId.parse(event.entityOperationCalled.targetInstanceId.value)
+                except ValueError:
                     raise RuntimeError(f"Could not parse entity ID from targetInstanceId '{event.entityOperationCalled.targetInstanceId.value}'")
-                ctx._entity_task_id_map[event.entityOperationCalled.requestId] = (entity_id, entity_call_id, None)
+                ctx._entity_task_id_map[event.entityOperationCalled.requestId] = (entity_id, entity_call_id)
             elif event.HasField("entityOperationSignaled"):
                 # This history event confirms that the entity signal was successfully scheduled.
                 # Remove the entityOperationSignaled event from the pending action list so we don't schedule it
@@ -1769,7 +1744,7 @@ class _OrchestrationExecutor:
                 ctx.resume()
             elif event.HasField("entityOperationCompleted"):
                 request_id = event.entityOperationCompleted.requestId
-                entity_id, task_id, _ = ctx._entity_task_id_map.pop(request_id, (None, None, None))
+                entity_id, task_id = ctx._entity_task_id_map.pop(request_id, (None, None))
                 if not entity_id:
                     raise RuntimeError(f"Could not parse entity ID from request ID '{request_id}'")
                 if not task_id:
@@ -1802,15 +1777,11 @@ class _OrchestrationExecutor:
                 action = ctx._pending_actions.pop(event.eventId, None)
                 if action and action.HasField("sendEntityMessage"):
                     if action.sendEntityMessage.HasField("entityOperationCalled"):
-                        action_type = "entityOperationCalled"
+                        entity_id, event_id = self._parse_entity_event_sent_input(event)
+                        ctx._entity_task_id_map[event_id] = (entity_id, event.eventId)
                     elif action.sendEntityMessage.HasField("entityLockRequested"):
-                        action_type = "entityLockRequested"
-                    else:
-                        return
-
-                    entity_id = EntityInstanceId.parse(event.eventSent.instanceId)
-                    event_id = json.loads(event.eventSent.input.value)["id"]
-                    ctx._entity_task_id_map[event_id] = (entity_id, event.eventId, action_type)
+                        entity_id, event_id = self._parse_entity_event_sent_input(event)
+                        ctx._entity_lock_task_id_map[event_id] = (entity_id, event.eventId)
             else:
                 eventType = event.WhichOneof("eventType")
                 raise task.OrchestrationStateError(
@@ -1820,6 +1791,44 @@ class _OrchestrationExecutor:
             # The orchestrator generator function completed
             ctx.set_complete(generatorStopped.value, pb.ORCHESTRATION_STATUS_COMPLETED)
 
+    def _parse_entity_event_sent_input(self, event: pb.HistoryEvent) -> Tuple[EntityInstanceId, str]:
+        try:
+            entity_id = EntityInstanceId.parse(event.eventSent.instanceId)
+        except ValueError:
+            raise RuntimeError(f"Could not parse entity ID from instanceId '{event.eventSent.instanceId}'")
+        try:
+            event_id = json.loads(event.eventSent.input.value)["id"]
+        except (json.JSONDecodeError, KeyError, TypeError) as ex:
+            raise RuntimeError(f"Could not parse event ID from eventSent input '{event.eventSent.input.value}'") from ex
+        return entity_id, event_id
+
+    def _handle_entity_event_raised(self,
+                                    ctx: _RuntimeOrchestrationContext,
+                                    event: pb.HistoryEvent,
+                                    entity_id: Optional[EntityInstanceId],
+                                    task_id: Optional[int],
+                                    is_lock_event: bool):
+        # This eventRaised represents the result of an entity operation after being translated to the old
+        # entity protocol by the Durable WebJobs extension
+        if entity_id is None:
+            raise RuntimeError(f"Could not retrieve entity ID for entity-related eventRaised with ID '{event.eventId}'")
+        if task_id is None:
+            raise RuntimeError(f"Could not retrieve task ID for entity-related eventRaised with ID '{event.eventId}'")
+        entity_task = ctx._pending_tasks.pop(task_id, None)
+        if not entity_task:
+            raise RuntimeError(f"Could not retrieve entity task for entity-related eventRaised with ID '{event.eventId}'")
+        result = None
+        if not ph.is_empty(event.eventRaised.input):
+            # TODO: Investigate why the event result is wrapped in a dict with "result" key
+            result = shared.from_json(event.eventRaised.input.value)["result"]
+        if is_lock_event:
+            ctx._entity_context.complete_acquire(event.eventRaised.name)
+            entity_task.complete(EntityLock(ctx))
+        else:
+            ctx._entity_context.recover_lock_after_call(entity_id)
+            entity_task.complete(result)
+        ctx.resume()
+
     def evaluate_orchestration_versioning(self, versioning: Optional[VersioningOptions], orchestration_version: Optional[str]) -> Optional[pb.TaskFailureDetails]:
         if versioning is None:
             return None

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev4
+Version: 0.0.0.dev6
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/durabletask.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ README.md
 pyproject.toml
 durabletask/__init__.py
 durabletask/client.py
+durabletask/py.typed
 durabletask/task.py
 durabletask/worker.py
 durabletask.egg-info/PKG-INFO
@@ -24,5 +25,4 @@ durabletask/internal/orchestration_entity_context.py
 durabletask/internal/orchestrator_service_pb2.py
 durabletask/internal/orchestrator_service_pb2.pyi
 durabletask/internal/orchestrator_service_pb2_grpc.py
-durabletask/internal/proto_task_hub_sidecar_service_stub.py
 durabletask/internal/shared.py

{durabletask-0.0.0.dev4 → durabletask-0.0.0.dev6}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durabletask"
-version = "0.0.0.dev4"
+version = "0.0.0.dev6"
 description = "A Durable Task Client SDK for Python"
 keywords = [
     "durable",

durabletask-0.0.0.dev4/durabletask/internal/proto_task_hub_sidecar_service_stub.py

@@ -1,34 +0,0 @@
-from typing import Any, Callable, Protocol
-
-
-class ProtoTaskHubSidecarServiceStub(Protocol):
-    """A stub class roughly matching the TaskHubSidecarServiceStub generated from the .proto file.
-    Used by Azure Functions during orchestration and entity executions to inject custom behavior,
-    as no real sidecar stub is available.
-    """
-    Hello: Callable[..., Any]
-    StartInstance: Callable[..., Any]
-    GetInstance: Callable[..., Any]
-    RewindInstance: Callable[..., Any]
-    WaitForInstanceStart: Callable[..., Any]
-    WaitForInstanceCompletion: Callable[..., Any]
-    RaiseEvent: Callable[..., Any]
-    TerminateInstance: Callable[..., Any]
-    SuspendInstance: Callable[..., Any]
-    ResumeInstance: Callable[..., Any]
-    QueryInstances: Callable[..., Any]
-    PurgeInstances: Callable[..., Any]
-    GetWorkItems: Callable[..., Any]
-    CompleteActivityTask: Callable[..., Any]
-    CompleteOrchestratorTask: Callable[..., Any]
-    CompleteEntityTask: Callable[..., Any]
-    StreamInstanceHistory: Callable[..., Any]
-    CreateTaskHub: Callable[..., Any]
-    DeleteTaskHub: Callable[..., Any]
-    SignalEntity: Callable[..., Any]
-    GetEntity: Callable[..., Any]
-    QueryEntities: Callable[..., Any]
-    CleanEntityStorage: Callable[..., Any]
-    AbandonTaskActivityWorkItem: Callable[..., Any]
-    AbandonTaskOrchestratorWorkItem: Callable[..., Any]
-    AbandonTaskEntityWorkItem: Callable[..., Any]