如何在Snowflake SQL函数中用参数指定information_schema所属数据库？

Snowflake的SQL函数是静态编译的，无法通过参数动态指定数据库名来引用DatabaseName.information_schema.columns这类对象。要实现你的需求，需要改用存储过程——存储过程支持运行时构造动态SQL，从而实现跨数据库的元数据查询。

具体实现代码 #

以下是基于SQL语言的存储过程实现，替换原有的SQL函数：

CREATE OR REPLACE PROCEDURE SP_GET_COLUMNS(DatabaseName VARCHAR, SchemaName VARCHAR, TableName VARCHAR)
RETURNS VARCHAR
LANGUAGE SQL
AS
$$
DECLARE
    sql_stmt STRING;
    result VARCHAR;
BEGIN
    -- 构造动态SQL，用IDENTIFIER处理数据库名的动态引用
    sql_stmt := '
        WITH data_columns AS (
            SELECT UPPER(column_name) AS column_name
            FROM ' || IDENTIFIER(:DatabaseName || '.information_schema.columns') || '
            WHERE table_catalog = UPPER(''' || :DatabaseName || ''')
                AND table_schema = UPPER(''' || :SchemaName || ''')
                AND table_name = UPPER(''' || :TableName || ''')
                AND is_identity != ''YES''
        ) SELECT LISTAGG(''src.''||column_name, '', '') within group(order by column_name) as data_col
        FROM data_columns
        WHERE NOT STARTSWITH(column_name,''__'')';
    
    -- 执行动态SQL并将结果存入变量
    EXECUTE IMMEDIATE :sql_stmt INTO result;
    RETURN result;
END;
$$;

关键说明 #

1. 动态对象引用：使用IDENTIFIER()函数包裹动态拼接的数据库名+information_schema路径，确保Snowflake能正确解析跨库的元数据表引用，同时避免SQL注入风险。
2. 动态SQL构造：通过字符串拼接生成目标查询语句，将输入参数代入到SQL逻辑中。
3. 结果返回：用EXECUTE IMMEDIATE ... INTO语法执行动态SQL，并将查询结果赋值给变量后返回。

调用方式 #

执行存储过程时直接传入参数即可：

CALL SP_GET_COLUMNS('目标数据库名', '目标模式名', '目标表名');

注意事项 #

• 确保执行存储过程的用户拥有目标数据库information_schema.columns的查询权限。
• 如果数据库/模式/表名包含特殊字符，IDENTIFIER()会自动处理引号包裹逻辑，无需额外转义。

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