Snowflake：如何用本地变量作为表标识符并检测表中是否存在数据

问题场景 #

先创建测试用临时表：

CREATE OR REPLACE TEMPORARY TABLE public.SomeTable
(
    id INT NOT NULL
);
INSERT INTO public.SomeTable (id) values (1);

直接使用字符串作为IDENTIFIER参数时，可正常返回1（表示表存在数据）：

EXECUTE IMMEDIATE $$
BEGIN
  IF (EXISTS(select top 1 1 from identifier('public.SomeTable'))) THEN
    RETURN 1;
  ELSE 
    RETURN 0;
  END IF;
END;
$$;

但使用本地变量存储表名传入IDENTIFIER时，会触发语法错误：

EXECUTE IMMEDIATE $$
BEGIN
  LET tableName string := 'public.SomeTable';
  IF (EXISTS(select top 1 1 from identifier(:tableName))) THEN
    RETURN 1;
  ELSE 
    RETURN 0;
  END IF;
END;
$$;

错误信息：

syntax error line 1 at position 45 unexpected 'parse_json'. 
syntax error line 1 at position 55 unexpected '('. (line 17)

解决方案 #

Snowflake存储过程中无法直接将绑定变量传入IDENTIFIER()作为静态SQL的一部分，需通过动态SQL拼接+执行的方式实现：

EXECUTE IMMEDIATE $$
DECLARE
  tableName string := 'public.SomeTable';
  has_data BOOLEAN;
  dynamic_query STRING;
BEGIN
  -- 拼接动态查询语句，用IDENTIFIER处理表名避免注入风险
  dynamic_query := 'SELECT EXISTS(SELECT TOP 1 1 FROM ' || IDENTIFIER(:tableName) || ')';
  
  -- 执行动态SQL并将结果存入变量
  EXECUTE IMMEDIATE dynamic_query INTO has_data;
  
  -- 根据结果返回1（有数据）或0（无数据）
  RETURN IFF(has_data, 1, 0);
END;
$$;

关键说明 #

• IDENTIFIER(:tableName)会自动处理标识符的转义逻辑，有效避免SQL注入风险
• 通过EXECUTE IMMEDIATE ... INTO语法捕获动态查询的结果，再完成后续判断返回

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