plpgsql：将变量串联到 FROM 子句中

Question

我是 Postgresql 新手，正在努力构建一个函数来循环一系列 CSV 文件并加载它们。我可以使 COPY 在单个文件中正常工作，但我无法使 FOR LOOP 语法正确。我正在尝试替换年份数字，因为我的苍蝇被命名为 /path/tmp.YEAR.out.csv

这就是我所破解的：

CREATE OR REPLACE FUNCTION test() RETURNS void as $$
BEGIN
  FOR i IN 1982..1983 LOOP
    COPY myTable 
    FROM '/path/tmp.' || i::VARCHAR || '.out.csv'
    delimiters ','
  END LOOP;
END;
$$ LANGUAGE 'plpgsql';

这会在第一个 || 处引发错误。所以我怀疑我对变量 i 的连接管理不正确。有什么建议吗？

Answer 1

CREATE OR REPLACE FUNCTION test() RETURNS void as $
BEGIN
FOR i IN 1982..1983 LOOP
  EXECUTE 'COPY myTable FROM ''/path/tmp.' || i::text
                                           || '.out.csv'' DELIMITERS '',''; ';
END LOOP;
END;
$ language plpgsql;

Answer 2

我认为我不会为此使用 plpgsql。 shell 脚本可能更普遍有用：

#!/bin/sh

DBHOST=mydbhost
DBNAME=mydbname

files=$1
target=$2

for file in ${files}; do
    psql -h ${DBHOST} ${DBNAME} -c "\copy ${target} FROM '${file}' delimiters ','"
done

示例用法：

csv2psql "$(ls *.out.csv)" someschema.tablename

注意：这样您还可以避免使用 COPY 读取文件的问题，这需要 postgres 服务器用户对文件具有读取权限。