使用动态文件名进行复制

我正在尝试编写一个函数来将 csv 数据加载到表中。我希望输入参数是文件的路径。

CREATE OR REPLACE FUNCTION public.loaddata(filepathname varchar)
  RETURNS void AS
$BODY$
BEGIN
COPY climatedata(
    climatestationid, 
    date,
    prcp,
    prcpqflag,
    prcpmflag,
    prcpsflag,
    tmax,
    tmaxqflag,
    tmaxmflag,
    tmaxsflag,
    tmin,
    tminqflag,
    tminmflag,
    tminsflag)
  FROM $1
  WITH csv header;
END;
$BODY$
  LANGUAGE plpgsql;

当我尝试创建这个函数时，我得到：

$1 处有语法错误

它出什么问题了？

COPY不允许变量替换。这只能通过核心 DML 命令实现SELECT, INSERT, UPDATE, and DELETE. See:

• 使用 plpgsql 变量设置 n_distinct 时出错 https://stackoverflow.com/questions/36025308/error-when-setting-n-distinct-using-a-plpgsql-variable/36025963#36025963

你需要动态SQL with EXECUTE:

CREATE OR REPLACE FUNCTION loaddata(filepathname text)
  RETURNS void
  LANGUAGE plpgsql AS
$func$
BEGIN
   EXECUTE format ('
   COPY climatedata(climatestationid, date, ..., tminsflag)  -- more columns 
   FROM %L (FORMAT CSV, HEADER)'  -- modern syntax
           -- WITH CSV HEADER'    -- tolerated legacy syntax
   , $1);  -- pass 1st function parameter (filepathname) to format() 
END
$func$;

format() https://www.postgresql.org/docs/current/functions-string.html#FUNCTIONS-STRING-OTHER需要 PostgreSQL 9.1+。
传递文件名而不使用额外的（转义的）单引号：

SELECT loaddata('/absolute/path/to/my/file.csv')

format() with %L安全地引用文件名。会容易受到SQL注入没有它。