如何在Airflow日志中打印PostgreSQL的RAISE NOTICE信息

方法1：通过PostgresHook手动设置捕获参数 #

在执行存储过程的任务中，显式配置PostgreSQL客户端捕获NOTICE级别消息，再提取打印：

from airflow.providers.postgres.hooks.postgres import PostgresHook

def run_procedure_and_capture_notices():
    hook = PostgresHook(postgres_conn_id="你的PostgreSQL连接ID")
    conn = hook.get_conn()
    cur = conn.cursor()
    
    # 设置客户端捕获NOTICE消息
    cur.execute("SET client_min_messages TO 'notice';")
    # 调用目标存储过程，替换为你的过程名和参数
    cur.callproc("your_procedure", (param1, param2))
    
    # 遍历并打印所有NOTICE消息
    for notice in conn.notices:
        print(notice.strip())
    
    conn.commit()
    cur.close()
    conn.close()

方法2：在Airflow连接配置中全局生效 #

进入Airflow的连接管理页面，找到目标PostgreSQL连接，在Extra字段添加以下JSON配置，让所有使用该连接的任务自动捕获NOTICE：

{"options": "-c client_min_messages=notice"}

之后执行存储过程时，同样可以通过conn.notices获取并打印消息。

方法3：在SQL语句中前置参数设置 #

直接在调用存储过程的SQL里先设置参数，再执行：

SET client_min_messages TO 'notice';
CALL your_procedure(%s, %s);

用PostgresHook的run方法执行这段SQL后，从连接对象的notices属性提取消息即可。

关键说明 #

• 确认存储过程使用RAISE NOTICE '调试内容: %', 变量名;的正确语法
• Airflow任务的日志级别需设为INFO及以上，才能看到打印的NOTICE内容
• 新版本Airflow的PostgresHook默认可能将client_min_messages设为warning及以上级别，导致NOTICE被过滤，所以需要手动调整参数

内容的提问来源于stack exchange，提问作者Chad H.