如何在Snowflake的SELECT语句中调用返回表的存储过程？

一、原存储过程调用失败的原因 #

Snowflake的存储过程（Procedure）属于执行型对象，核心用于执行DDL、DML或复杂流程控制，无法通过TABLE()函数直接嵌入SELECT语句返回结果集。你尝试的select * from table(test_sp.test_procedure('john'))写法仅适用于表函数（Table UDF），而非存储过程。

二、创建支持复杂Python逻辑的表UDF（表函数） #

由于你需要保留复杂Python逻辑，同时要能在SELECT语句中直接引用结果集，需创建Python表函数（Table UDF），而非存储过程。以下是适配需求的实现方案：

关键注意事项 #

• 表函数需显式定义返回的表结构
• 禁止用字符串拼接SQL（原代码的f-string存在SQL注入风险），改用Snowpark参数绑定
• 指定匹配的Python运行时版本和依赖包

完整实现代码 #

create or replace function test_sp.test_table_udf(name varchar)
returns table(id varchar, name varchar)
language python
packages = ('pandas', 'snowflake-snowpark-python')
runtime_version = 3.8
handler = 'get_data'
as
$$
def get_data(session, name):
    # 使用参数绑定替代字符串拼接，规避SQL注入风险
    df_sql = session.sql("""
        SELECT ID, NAME
        FROM TEST_SP.TABLE
        WHERE NAME = ?
    """, params=[name])
    # 直接返回Snowpark DataFrame，表函数会自动解析为结果集
    return df_sql
$$;

调用方式 #

现在可以用你原本尝试的语法直接调用：

select * from table(test_sp.test_table_udf('john'));

复杂Python逻辑扩展示例 #

如果需要添加自定义复杂逻辑（如数据清洗、计算、规则过滤），可在函数内对DataFrame处理后返回：

def get_data(session, name):
    df_sql = session.sql("""
        SELECT ID, NAME
        FROM TEST_SP.TABLE
        WHERE NAME = ?
    """, params=[name])
    # 示例复杂逻辑：新增计算列、过滤空值
    df_processed = df_sql.with_column("ID_DOUBLE", df_sql["ID"].cast("int") * 2) \
                         .filter(df_sql["ID"].is_not_null())
    return df_processed

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