Airflow DAG连接本地PostgreSQL遇连接拒绝问题求助

核心原因分析 #

在Airflow环境中使用*localhost*指向的是Airflow运行的容器（如果用Docker部署）本身，而非你的宿主机，这就是本地直接运行代码正常，但DAG执行失败的关键。

具体解决办法 #

• 修改连接地址：
  把代码中的host='localhost'替换成宿主机的实际IP地址（比如192.168.x.x这类局域网IP），或者在Docker部署场景下，使用host.docker.internal（仅适用于Docker Desktop）来指向宿主机。
  修改后的代码示例：

  try:
          conn = psycopg2.connect(database = "postgres", 
                          user = "postgres", 
                          host= '192.168.1.100',  # 替换成你的宿主机实际IP
                          password = "postgres",
                          port = 5432)
          print(conn.closed)
          print("Hola")
  

• 检查PostgreSQL网络配置：
  确保PostgreSQL的postgresql.conf中listen_addresses设置为'*'，允许所有IP访问；同时在pg_hba.conf中添加允许Airflow所在IP段访问的规则，比如Docker默认网段的规则：

  host    all             all             172.17.0.0/16          md5
  

• 使用Airflow内置连接管理：
  建议避免在DAG中硬编码数据库信息，改用Airflow的Connections功能：
  1. 进入Airflow UI（localhost:8080），点击Admin -> Connections
  2. 新建连接，Conn Type选择Postgres，填写宿主机IP、数据库名、用户名、密码、端口等信息，设置一个连接ID（比如postgres_local）
  3. 在DAG中通过连接ID调用，示例：

     from airflow.providers.postgres.hooks.postgres import PostgresHook
     
     def query_postgres():
         pg_hook = PostgresHook(postgres_conn_id='postgres_local')
         conn = pg_hook.get_conn()
         # 执行查询等操作
     

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