Revert to mapping codec over structure

Author: dandavison

temporalio/bridge/worker.py

@@ -297,21 +297,21 @@ async def visit_payloads(self, payloads: MutableSequence[Payload]) -> None:
 
 async def decode_activation(
     activation: temporalio.bridge.proto.workflow_activation.WorkflowActivation,
-    decode: Callable[[Sequence[Payload]], Awaitable[List[Payload]]],
+    codec: temporalio.converter.PayloadCodec,
     decode_headers: bool,
 ) -> None:
     """Decode all payloads in the activation."""
     await PayloadVisitor(
         skip_search_attributes=True, skip_headers=not decode_headers
-    ).visit(_Visitor(decode), activation)
+    ).visit(_Visitor(codec.decode), activation)
 
 
 async def encode_completion(
     completion: temporalio.bridge.proto.workflow_completion.WorkflowActivationCompletion,
-    encode: Callable[[Sequence[Payload]], Awaitable[List[Payload]]],
+    codec: temporalio.converter.PayloadCodec,
     encode_headers: bool,
 ) -> None:
     """Encode all payloads in the completion."""
     await PayloadVisitor(
         skip_search_attributes=True, skip_headers=not encode_headers
-    ).visit(_Visitor(encode), completion)
+    ).visit(_Visitor(codec.encode), completion)

temporalio/worker/_workflow.py

@@ -293,7 +293,7 @@ async def _handle_activation(
                     )
                 await temporalio.bridge.worker.decode_activation(
                     act,
-                    payload_codec.decode,
+                    payload_codec,
                     decode_headers=self._encode_headers,
                 )
             if not workflow:
@@ -369,7 +369,7 @@ async def _handle_activation(
             try:
                 await temporalio.bridge.worker.encode_completion(
                     completion,
-                    payload_codec.encode,
+                    payload_codec,
                     encode_headers=self._encode_headers,
                 )
             except Exception as err:

tests/worker/test_visitor.py

@@ -236,7 +236,7 @@ async def test_bridge_encoding():
         ),
     )
 
-    await temporalio.bridge.worker.encode_completion(comp, SimpleCodec().encode, True)
+    await temporalio.bridge.worker.encode_completion(comp, SimpleCodec(), True)
 
     cmd = comp.successful.commands[0]
     sa = cmd.schedule_activity