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不正确地管理变量的concat .有小费吗？

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服务器用户对文件具有读取权限.