[Airflow] Edge Label 사용하기

1. Edge Label 개념

• Task 연결에 대한 설명 (Comment)

.

 

 

2. Edge Label 사용 예제 코드

'''
dags_empty_with_edge_label.py
'''

from airflow import DAG
import pendulum
from airflow.operators.empty import EmptyOperator
from airflow.utils.edgemodifier import Label


with DAG(
    dag_id="dags_empty_with_edge_label",
    schedule=None,
    start_date=pendulum.datetime(2023, 4, 1, tz="Asia/Seoul"),
    catchup=False
) as dag:

    empty_1 = EmptyOperator(
        task_id = 'empty_1'
    )


    empty_2 = EmptyOperator(
        task_id = 'empty_2'
    )

    empty_1 >> Label('1과 2사이') >> empty_2


    empty_3 = EmptyOperator(
        task_id = 'empty_3'
    )

    empty_4 = EmptyOperator(
        task_id = 'empty_4'
    )

    empty_5 = EmptyOperator(
        task_id = 'empty_5'
    )

    empty_6 = EmptyOperator(
        task_id = 'empty_6'
    )

    empty_2 >> Label('start Branch') >> [empty_3, empty_4, empty_5] >> Label('End Branch') >> empty_6

 

실행결과 Graph