Skip to content

Commit 8b7b12c

Browse files
committed
Clarify optional system Nexus payload visitation
1 parent 9e306a5 commit 8b7b12c

4 files changed

Lines changed: 13 additions & 11 deletions

File tree

scripts/gen_payload_visitor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -225,14 +225,14 @@ async def visit(
225225
finally:
226226
await bounded.drain()
227227
228-
async def _visit_system_nexus_payload(
228+
async def _visit_nexus_operation_input_payload(
229229
self,
230230
fs: VisitorFunctions,
231231
service: str,
232232
operation: str,
233233
payload: Payload,
234234
) -> None:
235-
new_payload = await temporalio.nexus.system.visit_payload(
235+
new_payload = await temporalio.nexus.system.maybe_visit_payload(
236236
service,
237237
operation,
238238
payload,
@@ -446,7 +446,7 @@ def walk(self, desc: Descriptor) -> bool:
446446
_, field_name, service_expr, operation_expr, payload_expr = item
447447
lines.append(
448448
f' if o.HasField("{field_name}"):\n'
449-
" await self._visit_system_nexus_payload(\n"
449+
" await self._visit_nexus_operation_input_payload(\n"
450450
f" fs, {service_expr}, {operation_expr}, {payload_expr}\n"
451451
" )"
452452
)

temporalio/bridge/_visitor.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,14 +107,14 @@ async def visit(self, fs: VisitorFunctions, root: Any) -> None:
107107
finally:
108108
await bounded.drain()
109109

110-
async def _visit_system_nexus_payload(
110+
async def _visit_nexus_operation_input_payload(
111111
self,
112112
fs: VisitorFunctions,
113113
service: str,
114114
operation: str,
115115
payload: Payload,
116116
) -> None:
117-
new_payload = await temporalio.nexus.system.visit_payload(
117+
new_payload = await temporalio.nexus.system.maybe_visit_payload(
118118
service,
119119
operation,
120120
payload,
@@ -528,7 +528,9 @@ async def _visit_coresdk_workflow_commands_ScheduleNexusOperation(
528528
self, fs: VisitorFunctions, o: Any
529529
):
530530
if o.HasField("input"):
531-
await self._visit_system_nexus_payload(fs, o.service, o.operation, o.input)
531+
await self._visit_nexus_operation_input_payload(
532+
fs, o.service, o.operation, o.input
533+
)
532534

533535
async def _visit_coresdk_workflow_commands_WorkflowCommand(
534536
self, fs: VisitorFunctions, o: Any

temporalio/nexus/system/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,14 @@ def _operation(
2828
return workflow_service.__nexus_operation_registry__.get((service, operation))
2929

3030

31-
async def visit_payload(
31+
async def maybe_visit_payload(
3232
service: str,
3333
operation: str,
3434
payload: temporalio.api.common.v1.Payload,
3535
visitor_functions: VisitorFunctions,
3636
skip_search_attributes: bool,
3737
) -> temporalio.api.common.v1.Payload | None:
38-
"""Visit nested payloads inside a recognized system Nexus envelope."""
38+
"""Visit nested payloads if the payload is a recognized system Nexus envelope."""
3939
operation_def = _operation(service, operation)
4040
if operation_def is None:
4141
return None
@@ -69,6 +69,6 @@ def get_payload_converter() -> temporalio.converter.PayloadConverter:
6969
__all__ = [
7070
"get_payload_converter",
7171
"is_system_operation",
72+
"maybe_visit_payload",
7273
"SystemNexusPayloadConverter",
73-
"visit_payload",
7474
]

temporalio/nexus/system/_payload_visitor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,14 +107,14 @@ async def visit(self, fs: VisitorFunctions, root: Any) -> None:
107107
finally:
108108
await bounded.drain()
109109

110-
async def _visit_system_nexus_payload(
110+
async def _visit_nexus_operation_input_payload(
111111
self,
112112
fs: VisitorFunctions,
113113
service: str,
114114
operation: str,
115115
payload: Payload,
116116
) -> None:
117-
new_payload = await temporalio.nexus.system.visit_payload(
117+
new_payload = await temporalio.nexus.system.maybe_visit_payload(
118118
service,
119119
operation,
120120
payload,

0 commit comments

Comments
 (0)