ferruzzi commented on code in PR #62645:
URL: https://github.com/apache/airflow/pull/62645#discussion_r3066771711
##########
airflow-core/src/airflow/executors/base_executor.py:
##########
@@ -579,6 +582,76 @@ def get_cli_commands() -> list[GroupCommand]:
"""
return []
+ @staticmethod
+ def run_workload(
+ workload: ExecutorWorkload,
+ *,
+ server: str | None = None,
+ dry_run: bool = False,
+ subprocess_logs_to_stdout: bool = False,
+ proctitle: str | None = None,
+ ) -> int:
+ """
+ Pass the workload to the appropriate supervisor based on workload type.
+
+ Workload-specific attributes (log_path, sentry_integration,
bundle_info, etc.) are read from the
+ workload object itself.
+
+ :param workload: The ``ExecutorWorkload`` to execute.
+ :param server: Base URL of the API server (used by task workloads).
+ :param dry_run: If True, execute without actual task execution
(simulate run).
+ :param subprocess_logs_to_stdout: Should task logs also be sent to
stdout via the main logger.
+ :param proctitle: Process title to set for this workload. If not
provided, defaults to
+ ``"airflow supervisor: <workload.display_name>"``.
+ :return: Exit code of the process.
+ """
+ try:
+ from setproctitle import setproctitle
+
+ setproctitle(proctitle or f"airflow supervisor:
{workload.display_name}")
+ except ImportError:
+ pass
Review Comment:
fixed in coming update once local tests pass
--
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]
