气流获取受Postgres操作员影响的行

时间:2019-10-29 15:58:14

标签: postgresql airflow

我的Airflow DAG有一个Postgres运算符,我想知道它影响了多少行。这是我的任务的一个示例:

task4 = PostgresOperator(
    task_id='empty_staging',
    sql='DELETE FROM staging.table',
    postgres_conn_id='some_db_conn',
    autocommit=True,
    dag=dag,
)

执行此任务时,这就是日志中剩下的全部内容:

[2019-10-29 15:08:43,944] {logging_mixin.py:95} INFO - [[34m2019-10-29 15:08:43,943[0m] {[34mlocal_task_job.py:[0m105} INFO[0m - Task exited with return code 0[0m

据我所知,在Postgres运算符中没有记录到受影响行的方法吗?我如何记录发生了什么事?

1 个答案:

答案 0 :(得分:0)

您是否尝试过“返回*”-它会返回所有已删除的数据,因此您可以在此时对数组进行计数以获得计数。

 sql='DELETE FROM staging.table returning *',