如何在條件下將 PostgresOperator DAG 標記為失敗？

Question

我有一個 Airflow DAG 每天在特定時間運行。 它是一個 PG 運算符並獲取一個表。 如下所示：

select count(*) from my_table where date_insert='{{ds}}'

現在，如果此計數為零，我需要將此任務標記為失敗。 所以我猜我需要將數據從 DAG 返回到 Python。

Answer 1

這是與 PostgreSQL 兼容的SQLCheckOperator的用例：

from airflow.operators.sql import SQLCheckOperator
SQLCheckOperator(
    conn_id='your_conn',
    sql="""select count(*) from my_table where date_insert='{{ds}}'""",
    
)

如果count==0 ，操作員將引發AirflowException 。