Snowflake中JS存储过程遍历表插入数据仅首行生效问题求助

问题原因 #

你的存储过程在循环内部执行了return "Done";，这会导致第一次循环完成后直接终止存储过程，后续行的插入逻辑根本不会执行。此外，依赖getRowCount()的for循环写法在Snowflake中不够可靠，因为ResultSet.getRowCount()需要结果集完全加载后才能返回准确值，而next()方法本身可以用来判断是否还有未处理的行。

修正后的代码 #

create or replace procedure proc3()
returns varchar
language javascript
as
$$
var query1 = 'select * from test2';
var stmt = snowflake.createStatement({sqlText: query1});
var rs = stmt.execute();

// 用while循环遍历所有行，next()返回true表示存在下一行
while (rs.next()) {
    var x = rs.getColumnValue(1);
    var y = rs.getColumnValue(2);
    var z = rs.getColumnValue(3);
    
    var insertSql = "INSERT INTO test3 VALUES (?,?,?)";
    snowflake.execute({sqlText: insertSql, binds: [x, y, z]});
}

// 循环结束后再返回结果
return "All rows inserted successfully";
$$
;

call proc3();

关键修改点 #

• 移除了循环内部的return语句，将返回逻辑移到循环外部，确保所有行都能被处理。
• 改用while (rs.next())的方式遍历结果集，这是Snowflake JS存储过程中遍历结果的标准写法，避免getRowCount()可能带来的问题。
• 规范了变量命名（小写开头，符合JavaScript命名习惯），提升代码可读性。

额外优化建议 #

如果test2表数据量较大，逐行插入效率较低，建议改用批量插入或者直接用SQL的INSERT ... SELECT语句替代存储过程，性能会更优：

INSERT INTO test3
SELECT col1, col2, col3 FROM test2;

（将col1, col2, col3替换为test2表实际的列名）

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