Redshift存储过程中如何传入S3 manifest参数执行COPY命令

Redshift PL/pgSQL 的静态SQL语句不支持直接将变量作为COPY命令的FROM参数，解析器会把输入参数自动替换为$1占位符，而COPY命令要求FROM后必须直接跟路径字面量，因此触发语法错误。

使用EXECUTE执行动态拼接的SQL语句即可实现参数传入，推荐配合format函数处理引号转义，避免语法混乱。

CREATE OR REPLACE PROCEDURE stage.sp_stage_user_activity_page_events(manifest_location varchar(256))
LANGUAGE plpgsql
AS $$
BEGIN
  EXECUTE format(
    'COPY stage.user_activity_event 
     FROM %L
     IAM_ROLE ''arn:aws:iam::XXX:role/redshift-s3-read-only-role''
     IGNOREHEADER 1
     REMOVEQUOTES
     DELIMITER '',''
     LZOP
     MANIFEST;',
    manifest_location
  );
END;
$$;

• format函数中的%L占位符会自动将manifest_location的值转为带单引号的合法字符串字面量，无需手动处理引号，同时可避免SQL注入风险
• 动态SQL字符串内部的单引号需要用两个连续单引号转义，示例中的IAM_ROLE参数值、DELIMITER参数值都遵循该规则
• 存储过程的调用方式不变，示例：CALL stage.sp_stage_user_activity_page_events('s3://your-bucket/manifest.json');
• 排错时可在EXECUTE前添加RAISE NOTICE '%', format(...)语句打印生成的SQL，确认语句正确性

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