airflow - 使用 "with dag as DAG(...)"子句创建 DAG 有好处吗

Question

我按如下方式创建我的 Airflow DAG:

dag = DAG(...)

但在多个教程和类(class)中，我看到他们像这样使用 with ... as 子句:

with DAG(...) as dag:
    # Code that will use the dag variable.

我猜这样一来，DAG 实例将在调度程序执行代码块后被销毁，但这样做真的有好处吗？我找不到任何关于此的文档。

Answer 1

是的。

如果您了解 with...as 的作用，那么您应该明白它对 Airflow 生态系统的影响确实没有什么不同。

具体来说，它确保非托管资源(在本例中为 DAG 类的实现)得到正确清理，即使抛出异常(无需每次都使用 try/except block 。)

此外，不必为每个单独的代码都添加 dag=dag 也很好。