Skip to content

Reduce entity context exposure #71

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Nov 11, 2025
Merged

Reduce entity context exposure #71

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
da6b55a
Reduce entity context exposure
andystaples Nov 6, 2025
6fa6cfa
Merge branch 'main' into andystaples/minimize-entity-state-exposure
andystaples Nov 11, 2025
34da301
Merge branch 'main' into andystaples/minimize-entity-state-exposure
berndverst Nov 11, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 4 additions & 6 deletions durabletask/worker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -819,7 +819,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
_generator: Optional[Generator[task.Task, Any, Any]]
_previous_task: Optional[task.Task]

def __init__(self, instance_id: str, registry: _Registry, entity_context: OrchestrationEntityContext):
def __init__(self, instance_id: str, registry: _Registry):
self._generator = None
self._is_replaying = True
self._is_complete = False
Expand All @@ -834,7 +834,7 @@ def __init__(self, instance_id: str, registry: _Registry, entity_context: Orches
self._current_utc_datetime = datetime(1000, 1, 1)
self._instance_id = instance_id
self._registry = registry
self._entity_context = entity_context
self._entity_context = OrchestrationEntityContext(instance_id)
self._version: Optional[str] = None
self._completion_status: Optional[pb.OrchestrationStatus] = None
self._received_events: dict[str, list[Any]] = {}
Expand Down Expand Up @@ -1271,7 +1271,6 @@ def __init__(self, registry: _Registry, logger: logging.Logger):
self._logger = logger
self._is_suspended = False
self._suspended_events: list[pb.HistoryEvent] = []
self._entity_state: Optional[OrchestrationEntityContext] = None

def execute(
self,
Expand All @@ -1288,13 +1287,12 @@ def execute(
f"{instance_id}: Beginning replay for orchestrator {orchestration_name}..."
)

self._entity_state = OrchestrationEntityContext(instance_id)

if not new_events:
raise task.OrchestrationStateError(
"The new history event list must have at least one event in it."
)
ctx = _RuntimeOrchestrationContext(instance_id, self._registry, self._entity_state)

ctx = _RuntimeOrchestrationContext(instance_id, self._registry)
try:
# Rebuild local state by replaying old history into the orchestrator function
self._logger.debug(
Expand Down
Loading