使用Snowflake和Apache Airflow时如何为SqlSensor连接指定数据库和SCHEMA

不需要为每个数据库和Schema单独创建Airflow连接，有两种成熟方案可以实现运行时指定目标环境参数：

方案1：查询中使用完全限定表名（最简方案） #

不需要修改任何连接配置或传感器参数，直接在感知SQL中写明完整的三级表名（数据库.模式.表）即可，全程仅单条SQL语句，不会触发多语句错误。
示例查询：

SELECT 1 FROM PROD_DB.USER_SCHEMA.USER_LOG WHERE DATE(DT) = CURRENT_DATE LIMIT 1

方案2：通过hook_params动态覆盖连接参数 #

如果不想在每条SQL里重复写完整表名，可通过SqlSensor的hook_params参数在实例化时动态指定目标数据库、模式甚至仓库、角色等配置，直接覆盖Snowflake连接中的默认值。
示例代码：

from airflow.sensors.sql import SqlSensor

check_table_sensor = SqlSensor(
    task_id="check_snowflake_user_log",
    conn_id="your_default_snowflake_conn", # 共用同一个基础连接
    sql="SELECT 1 FROM USER_LOG WHERE DATE(DT) = CURRENT_DATE LIMIT 1",
    hook_params={
        "database": "PROD_DB",
        "schema": "USER_SCHEMA",
        # 可选参数，按需添加即可
        # "warehouse": "ANALYSIS_WH",
        # "role": "DATA_ANALYST_ROLE"
    },
    mode="poke",
    timeout=3600
)

注意事项 #

• 两种方案均无需添加USE DATABASE/USE SCHEMA这类切换语句，完全规避多语句SQL执行报错问题
• hook_params支持传入所有SnowflakeHook初始化可接收的参数，可按需灵活调整执行环境

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