在Airflow DAG中,如果想要将返回值写入日志文件,可以使用PythonOperator,并在其execute函数中使用TaskInstance对象的xcom_push方法将返回值推送到XCom。然后,在需要使用该返回值的任务中,使用TaskInstance对象的xcom_pull方法获取该返回值。
以下是一个简单的示例:
from airflow import DAG
from airflow.operators.python_operator import PythonOperator
from datetime import datetime
from airflow.models import TaskInstance
def my_task(**context):
# Your task code here
return "my return value"
with DAG(dag_id="my_dag", start_date=datetime(2022, 1, 1)) as dag:
task1 = PythonOperator(task_id="task1", python_callable=my_task)
task2 = PythonOperator(task_id="task2",
python_callable=my_task,
op_kwargs={"my_return_value": TaskInstance(context['ti']).xcom_pull(key=None, task_ids='task1')})
task1 >> task2
在这个示例中,task1操作将返回'my return value”,然后可在task2中使用该值。在task2中,使用TaskInstance对象的xcom_pull方法获取task1的xcom,并将其赋给op_kwargs参数,在task2中可以通过参数'my_return_value”获取该值。