gopidesupavan commented on code in PR #51279:
URL: https://github.com/apache/airflow/pull/51279#discussion_r2124499635
##########
task-sdk/src/airflow/sdk/execution_time/task_runner.py:
##########
@@ -657,14 +660,41 @@ class CommsDecoder(Generic[ReceiveMsgType, SendMsgType]):
lock: aiologic.Lock = attrs.field(factory=aiologic.Lock, repr=False)
+ def _read_stdin_line(self):
+ """
+ Read triggerer workloads.
+
+ Used in ThreadPoolExecutor in triggerer to read blockingly from stdin.
+ """
+ line = None
+ while not line:
+ line = self.input.readline()
+ return line.encode()
+
def get_message(self) -> ReceiveMsgType:
"""
Get a message from the parent.
This will block until the message has been received.
"""
+ global TRIGGERER_SUPERVISOR_COMMS_FUTURE
line = None
+ if TRIGGERER_SUPERVISOR_COMMS_FUTURE is not None:
Review Comment:
Agree, happy to do with subclass..
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
