要从任务实例中访问正在运行的任务,可以使用Airflow的XCom功能。XCom是Airflow用于任务之间交换数据的机制。
以下是一个使用XCom来访问正在运行的任务的代码示例:
from airflow import DAG
from airflow.operators.python_operator import PythonOperator
from airflow.utils.dates import days_ago
def task1(**context):
# 获取正在运行的任务的task_instance
task_instance = context['task_instance']
# 获取正在运行的任务的任务ID
task_id = task_instance.task_id
# 获取正在运行的任务的DAG ID
dag_id = task_instance.dag_id
# 打印正在运行的任务的信息
print(f"正在运行的任务信息:DAG ID={dag_id}, 任务ID={task_id}")
def task2(**context):
# 获取上一个任务的输出值
previous_task_output = context['ti'].xcom_pull(task_ids='task1')
# 打印上一个任务的输出值
print("上一个任务的输出值:", previous_task_output)
dag = DAG(
'example_dag',
schedule_interval='@once',
start_date=days_ago(1),
)
task1 = PythonOperator(
task_id='task1',
python_callable=task1,
provide_context=True,
dag=dag,
)
task2 = PythonOperator(
task_id='task2',
python_callable=task2,
provide_context=True,
dag=dag,
)
task1 >> task2
在上面的示例中,task1任务获取正在运行的任务的信息,并将其作为XCom输出。然后,task2任务使用xcom_pull方法从task1任务获取XCom输出的值。
请注意,要使用XCom功能,需要将provide_context参数设置为True,以便将上下文传递给任务函数。