Airflow DAG无法连接Azure SQL数据库的排查求助

核心问题分析 #

你遇到的登录失败问题，本质是Airflow的MsSqlOperator依赖的pymssql库，与你配置的ODBC格式连接字符串不兼容——pymssql不支持ODBC连接串，需要单独配置连接参数。以下是具体排查和解决步骤：

1. 修正Airflow连接配置 #

登录Airflow UI，进入Admin > Connections，找到标识为mssql的连接：

• 连接类型选择Microsoft SQL Server
• 按以下规则填写参数：
  • Host: mydbserver.database.windows.net
  • Schema: mydatabase（你的目标数据库名）
  • Login: loginname@mydbserver（注意：Azure SQL SQL认证需要带服务器后缀，不能只填loginname）
  • Password: 你的数据库登录密码
  • Port: 1433
  • Extra字段添加：{"encrypt": true}（适配Azure SQL的强制加密要求）

2. 验证Airflow环境的pymssql配置 #

直接运行的pymssql代码可能使用的是系统Python环境，而Airflow可能运行在独立虚拟环境中：

• 执行airflow info查看Airflow使用的Python路径
• 用该路径的pip检查pymssql版本：

  /path/to/airflow/python -m pip show pymssql
  

• 确保安装的是2.2.5及以上版本的pymssql（适配Azure SQL的最新加密规则），若版本不符则重新安装：

  /path/to/airflow/python -m pip install pymssql>=2.2.5
  

3. 测试连接参数有效性 #

在Airflow所在虚拟机中，用Airflow的Python环境运行以下测试代码，验证参数是否正确：

import pymssql

try:
    conn = pymssql.connect(
        host='mydbserver.database.windows.net',
        user='loginname@mydbserver',
        password='your_password',
        database='mydatabase',
        port=1433,
        encrypt=True
    )
    print("连接成功")
    conn.close()
except Exception as e:
    print(f"连接失败: {str(e)}")

如果这段代码能成功连接，说明参数正确，再回到Airflow连接配置中对应填写即可。

4. 确认Azure SQL防火墙规则 #

虽然你已添加VM IP，但仍需确认：

• 防火墙规则是否包含VM的公网IP（注意：部分云VM的内部IP和公网IP不同）
• 是否启用了允许Azure服务和资源访问此服务器选项（若Airflow部署在Azure内部资源中）

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