plpgsql: nối biến vào mệnh đề FROM

Question

Tôi mới dùng Postgresql và đang cố gắng xây dựng một hàm để lặp qua hàng loạt tệp CSV và tải chúng. Tôi có thể làm cho COPY hoạt động tốt với một tệp duy nhất, nhưng tôi không thể nhận cú pháp FOR LOOP đúng. Tôi đang cố gắng để thay thế một số năm như ruồi của tôi được đặt tên /path/tmp.YEAR.out.csv

Đây là những gì tôi đã bị hack lên:

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'; 

này ném một lỗi ở đầu tiên ||. Vì vậy, tôi nghi ngờ tôi đang quản lý concat của biến i không đúng cách. Có lời khuyên nào không?

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

Tôi không nghĩ mình sẽ sử dụng plpgsql cho điều đó. Một kịch bản shell có thể là nhiều hơn nữa thường hữu ích:

#!/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 

ví dụ sử dụng:

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

Lưu ý: Bằng cách này bạn cũng tránh các vấn đề về đọc file với COPY mà yêu cầu người dùng postgres máy chủ để đã đọc quyền trên tệp.