Snowflake存储过程中创建FOR循环执行指定分组SQL查询求助

原代码存在的问题 #

• 未声明输入参数：调用时传入GQID02，但存储过程未定义对应的参数，导致调用失败。
• 过滤条件错误：WHERE groupquery=''硬编码为空字符串，未关联传入的groupqueryids参数，无法匹配目标查询集。
• 结果集未持久化：循环中执行动态查询后未保留结果，无法累积生成最终输出。
• 返回值不符合需求：原存储过程返回VARCHAR类型，无法输出查询结果集用于CSV导出。

修正后的存储过程 #

以下存储过程会接收groupqueryids参数，循环执行对应分组的查询，将所有结果存入临时表后返回完整结果集：

CREATE OR REPLACE PROCEDURE myprocedure(p_groupqueryids VARCHAR)
  RETURNS RESULTSET
  LANGUAGE SQL
  AS
  $$
    DECLARE
      c1 CURSOR FOR SELECT query FROM qset WHERE groupqueryids = :p_groupqueryids ORDER BY orderid;
      v_query VARCHAR;
      -- 创建临时表存储最终结果（需根据实际查询结果定义列，这里示例用通用结构）
      temp_result TABLE (like (SELECT * FROM qset LIMIT 0)); -- 替换为实际查询结果的结构
    BEGIN
      -- 清空临时表（若存在）
      TRUNCATE TABLE IF EXISTS temp_result;
      
      FOR record IN c1 DO
        v_query := record.query;
        -- 执行动态查询并将结果插入临时表
        EXECUTE IMMEDIATE 'INSERT INTO temp_result ' || v_query;
      END FOR;
      
      -- 返回最终结果集
      RETURN (SELECT * FROM temp_result);
    END;
  $$;

关键说明 #

1. 参数定义：新增p_groupqueryids参数用于接收用户传入的分组ID，调用时需用单引号包裹字符串参数：

   CALL myprocedure('GQID02');
   

2. 临时表结构：示例中使用like (SELECT * FROM qset LIMIT 0)匹配临时表结构，需根据实际查询的结果列调整，确保与动态查询的输出列一致。
3. 结果累积：通过INSERT INTO temp_result将每次动态查询的结果存入临时表，最终返回完整结果集。

导出CSV的方法 #

存储过程返回结果集后，可通过以下方式导出为CSV：

• Snowflake UI：在查询结果面板点击"Download"按钮，选择CSV格式导出。
• COPY INTO语句：将临时表数据导出到外部存储（如S3），再下载CSV文件：

  COPY INTO '@your_stage/path/result.csv'
  FROM temp_result
  FILE_FORMAT = (TYPE = CSV FIELD_OPTIONALLY_ENCLOSED_BY = '"');
  

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