Airflow :如何在非PythonOperator中使用xcom_push和xcom_pull

Question

我看到了很多有关如何在Airflow中将xcom_push和xcom_pull与PythonOperators结合使用的示例。

我需要从非PythonOperator类中执行xcom_pull，但找不到如何执行该操作。

任何指针或示例将不胜感激!

Answer 1

您可以从模板字段中访问XCom变量。例如，从XCom读取:

myOperator = MyOperator(
    message="Operation result: {{ task_instance.xcom_pull(task_ids=['task1', 'task2'], key='result_status') }}",
    ...

也可以不指定任务以使用相同的键名在一个DagRun中获取所有XCom推送

myOperator = MyOperator(
    message="Warning status: {{ task_instance.xcom_pull(task_ids=None, key='warning_status') }}",
    ...

将返回一个数组。