Airflow 中提供了 DAG 类的 .on_success_callback 和 .on_failure_callback 属性，分别用于设置 DAG 成功和失败时的回调函数。这些回调函数将在 DAF 的全部任务成功或全部任务失败时自动触发。

示例代码如下：

from airflow import DAG
from datetime import datetime

def on_success(context):
    print("All tasks success!")

def on_failure(context):
    print("Some tasks failed!")

default_args = {
    'start_date': datetime(2021, 10, 1),
}

dag = DAG(
    'my_dag',
    default_args=default_args,
    schedule_interval='@once',
    on_success_callback=on_success,
    on_failure_callback=on_failure
)

# 定义 DAG 的任务

在上面的示例代码中，我们定义了两个回调函数 on_success 和 on_failure，分别在 DAG 的全部任务成功和至少有一个任务失败时被触发。这些回调函数会在 DAG 中的每个任务执行完成后自动调用，并传入参数 context，其中包含了执行任务的相关信息。