Snowflake中SELECT调用存储过程SP_CALC_BUCKET报错，如何修改查询？

你遇到的问题核心是：Snowflake的SELECT语句只能调用函数（UDF/表函数），而存储过程必须通过CALL命令执行，所以直接在SELECT里写存储过程名会被识别为UDF，导致找不到对象。以下是几种可行的解决方法：

方法1：将存储过程改写为用户定义函数（UDF） #

如果SP_CALC_BUCKET仅用于计算并返回单个值，没有数据修改、DDL执行等副作用，这是最优方案。UDF可以直接在SELECT中调用，和你原本的写法几乎一致。

假设原存储过程的逻辑是接收日期和两个数值参数，返回计算后的结果，改写为UDF的示例（以SQL UDF为例）：

CREATE OR REPLACE FUNCTION UDF_CALC_BUCKET(p_date DATE, p_param1 INT, p_param2 INT)
RETURNS <返回数据类型> -- 替换为原存储过程的返回类型
LANGUAGE SQL
AS
$$
    -- 复制原存储过程中的计算逻辑到这里
    -- 例如：CASE WHEN ... THEN ... ELSE ... END
$$;

之后就可以用你原本的查询语句调用UDF：

SELECT UDF_CALC_BUCKET($1, $2, $3) 
FROM (VALUES ('2022-01-01', 5, 4), ('1972-10-01', 1, 6), ('2008-08-08', 1, 7), 
    ('1999-12-31', 2, 8), ('2000-01-01', 0, 10));

方法2：改写为表函数（Table Function） #

如果需要返回多行结果或复杂结构，可以将逻辑封装为表函数，通过TABLE()语法在查询中调用：

CREATE OR REPLACE FUNCTION TF_CALC_BUCKET(p_date DATE, p_param1 INT, p_param2 INT)
RETURNS TABLE(result_col <数据类型>)
LANGUAGE SQL
AS
$$
    -- 编写返回结果集的逻辑，例如：
    SELECT <计算逻辑> AS result_col
$$;

调用方式：

SELECT t.$1, t.$2, t.$3, res.result_col
FROM (VALUES ('2022-01-01', 5, 4), ('1972-10-01', 1, 6), ('2008-08-08', 1, 7), 
    ('1999-12-31', 2, 8), ('2000-01-01', 0, 10)) AS t,
TABLE(TF_CALC_BUCKET(t.$1, t.$2, t.$3)) AS res;

方法3：用存储过程批量处理（保留原存储过程） #

如果必须保留原存储过程（例如包含数据修改等副作用），可以编写一个批量处理的存储过程，通过游标遍历输入数据集，逐行调用原存储过程并收集结果：

CREATE OR REPLACE PROCEDURE SP_BATCH_CALC_BUCKET()
RETURNS TABLE(input_date DATE, param1 INT, param2 INT, calc_result <数据类型>)
LANGUAGE JAVASCRIPT
AS
$$
    var output = [];
    // 定义输入数据集
    var inputStmt = snowflake.createStatement({
        sqlText: `SELECT * FROM (VALUES ('2022-01-01', 5, 4), ('1972-10-01', 1, 6), ('2008-08-08', 1, 7), ('1999-12-31', 2, 8), ('2000-01-01', 0, 10)) AS t(d, p1, p2)`
    });
    var inputRs = inputStmt.execute();
    
    while (inputRs.next()) {
        var dateVal = inputRs.getColumnValue(1);
        var p1Val = inputRs.getColumnValue(2);
        var p2Val = inputRs.getColumnValue(3);
        
        // 调用原存储过程
        var callStmt = snowflake.createStatement({
            sqlText: "CALL SP_CALC_BUCKET(?, ?, ?)",
            binds: [dateVal, p1Val, p2Val]
        });
        var callRs = callStmt.execute();
        callRs.next();
        var resultVal = callRs.getColumnValue(1);
        
        // 收集结果
        output.push({
            INPUT_DATE: dateVal,
            PARAM1: p1Val,
            PARAM2: p2Val,
            CALC_RESULT: resultVal
        });
    }
    return output;
$$;

调用批量存储过程：

CALL SP_BATCH_CALC_BUCKET();

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