Snowflake SQL存储过程动态添加SELECT字段需求咨询

要实现根据输入的逗号分隔字符串动态生成SELECT字段，核心是将输入字符串拆分后拼接成符合格式的字段列表，再整合到主查询语句中。以下是修改后的完整存储过程代码：

CREATE OR REPLACE PROCEDURE dynamic_query_example(c_input VARCHAR)
RETURNS VARCHAR
LANGUAGE SQL
AS
DECLARE
  select_statement VARCHAR;
  dynamic_fields VARCHAR;
  res RESULTSET;
BEGIN
  -- 拆分输入字符串并拼接成带格式的字段列表
  SELECT LISTAGG(',\n                             ' || VALUE, '') 
  INTO dynamic_fields
  FROM TABLE(SPLIT_TO_TABLE(c_input, ','));

  -- 构建最终的查询语句
  select_statement := 'create or replace temporary table TEST_TBL as 
                       SELECT * ' || dynamic_fields || '
                       FROM c_data 
                       WHERE c_name IN (SELECT VALUE FROM TABLE(SPLIT_TO_TABLE(''' || c_input || ''', '','')))';

  res := (EXECUTE IMMEDIATE :select_statement);
  RETURN 'OK';
END;

关键逻辑说明 #

• 拆分输入：使用SPLIT_TO_TABLE将c_input按逗号拆分为单行记录，每个值对应一个要添加的字段。
• 拼接字段列表：通过LISTAGG函数将每个字段值拼接成带换行和缩进的字符串，保证生成的SQL格式清晰，每个新增字段单独占一行并对齐。
• 整合主语句：将拼接好的字段列表直接追加到SELECT *之后，形成最终的动态查询语句。

效果示例 #

当输入c_input = 'A,B,C'时，生成的select_statement内容如下：

create or replace temporary table TEST_TBL as 
                       SELECT *,
                              A,
                              B,
                              C 
                       FROM c_data 
                       WHERE c_name IN (SELECT VALUE FROM TABLE(SPLIT_TO_TABLE('A,B,C', ',')))

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