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

问题描述

| 我是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\';

这将在第一个||处引发错误。因此，我怀疑我在不适当地管理变量ѭ1con的连接。有小费吗？     

解决方法

        

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;

    ,        我不认为我会为此使用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服务器用户具有对该文件的读取权限。