snowflake-cloud-data-platform - 在数组中运行多个 Snowflake UDF 的给定参数

Question

基本上，我真正想做的是

for i in arr:
 foo(i)

其中 foo 是一个雪花脚本，用于查找与 i 匹配的记录并将它们合并到不同的表中。

有办法做到这一点吗？

Answer 1

目前尚不支持 SQL 脚本，但使用如下所示的简单存储过程应该可以解决问题:

create or replace table table1 (a int);
create or replace table table2 (a int);
insert into table1 values (1), (2), (3), (4), (5);
select * from table1;
select * from table2; -- empty at this stage
 
CREATE OR REPLACE PROCEDURE my_proc(INPUT VARCHAR) 
RETURNS BOOLEAN 
LANGUAGE JAVASCRIPT 
AS 
$$
    function my_insert(i) {
        var stmt = snowflake.createStatement( {
            sqlText: "INSERT INTO table2 SELECT * FROM table1 WHERE a = ?",
            binds: [i]
        } ); 
        stmt.execute(); 
    }
    
    arr = INPUT.split(",");
    for (var i=0; i<arr.length; i++) {
      my_insert(arr[i])
    }
    
    return true;
$$ ; 


call my_proc('1,2,5');
select * from table2; -- should have values 1, 2 and 5