airflow SubDAG怎么使用

Apache Airflow中的SubDAG（子任务图）是一种强大的功能，它允许你在一个主DAG中组织和管理多个独立但相关的任务序列。SubDAGs可以帮助你保持代码结构清晰，便于维护，并且可以灵活地进行调度和扩展。

使用SubDAG的步骤如下：

1. **创建子任务图**:
- 在Airflow DAG文件中，定义一个新的类，继承自`airflow.models.DAG`，并覆盖`__init__`方法，指定子任务的名称、描述、默认起始时间等信息。

   from airflow import DAG
   from airflow.operators.dummy_operator import DummyOperator

   class SubDAGExample(DAG):
       default_args = {
           'owner': 'airflow',
           'start_date': datetime(2023, 1, 1),
       }
       dag_id = 'subdag_example'

   sub_dag = SubDAGExample()

2. **添加任务**:
- 在子任务图类里，你可以像在常规DAG中一样添加各种任务，如`DummyOperator`、`PythonOperator`、`BranchPythonOperator`等。

   task_1 = DummyOperator(task_id='task_1', dag=sub_dag)
   task_2 = DummyOperator(task_id='task_2', dag=sub_dag)

3. **连接任务**:
- 如果需要，可以使用`sub_dag >> task_2`的方式将子任务链起来，或者使用`apply_pressure`函数将任务添加到特定的时间点。

4. **包含在主DAG中**:
- 在主DAG中，调用子DAG实例，并将其添加到调度中，比如使用`DAG.add_subdag`或`DAG.schedule_on_demand`。

   main_dag = DAG('main_dag', default_args={...})
   main_dag.add_subdag(sub_dag)

5. **调度和运行**:
- 配置主DAG的调度规则，然后运行`airflow dags run main_dag`命令启动整个流程。