如何在本地Airflow DAG中连接远程PostgreSQL数据库？

先搞定当前的报错问题 #

你现在的错误很明确：代码里指定的postgres_conn_id='inform_status_adb'，对应的Airflow连接配置里把数据库名设成了inform_status_adb，但你要连的远程库明明是asd（从你给的jdbc地址jdbc:postgresql://10.10.10.10:5432/asd能看出来），PostgreSQL找不到这个不存在的库，自然报错了。

连接远程PostgreSQL的完整最佳实践 #

1. 在Airflow里配置正确的连接 #

• 打开Airflow UI，进入Admin > Connections页面
• 点**+ Add a new record**，把这些关键信息填对：
  • Conn Id：就用你代码里的inform_status_adb（或者自己起个好记的唯一ID）
  • Conn Type：选Postgres
  • Host：远程数据库的IP，也就是10.10.10.10
  • Schema：要连接的数据库名，这里填asd
  • Login：远程PostgreSQL的用户名
  • Password：对应用户的密码
  • Port：填5432（PostgreSQL默认端口）
• 保存后检查一遍，别写错字

2. 先验证连接能不能通 #

• 在Connections页面找到刚创建的连接，点Test按钮测一下
• 如果测试失败，按这几个方向排查：
  • 远程服务器的防火墙有没有开5432端口，允许你的Airflow主机访问
  • 远程PostgreSQL的pg_hba.conf文件里，有没有加Airflow主机的IP白名单，允许它连进来
  • 你用的用户名有没有访问asd数据库的权限
  • 远程PostgreSQL里真的存在asd这个数据库吗？可以登到远程库上用\l命令确认

3. 代码里的小规范 #

• 确保PostgresOperator的postgres_conn_id和Airflow里配置的Conn Id完全一致，大小写都不能错
• 你的SQL里用了双引号包裹"content"、"role"这种大小写敏感的字段，这点要保持，不然PostgreSQL会自动转成小写，可能找不到字段
• 要是SQL语句比较长，建议单独存成.sql文件，然后用sql='sql_files/create_report.sql'这种方式引用，代码看着清爽，也方便维护

4. 生产环境的额外注意点 #

• 绝对不要在代码里硬写数据库密码、用户名这些敏感信息，全靠Airflow Connections管理，避免泄露
• 敏感环境里要开Airflow的连接加密功能，确保存的密码是加密状态
• 可以用Airflow默认支持的连接池，减少频繁创建数据库连接的开销，提升性能

内容的提问来源于stack exchange，提问作者Umid Umaraliev