我已经复制了函数(从其中一个门户网站并进行了相应修改)以将数据从 csv 文件复制到表中。
create or replace function public.load_csv_file
(
target_table text,
csv_path text,
col_count integer
)
returns void as $$
declare
iter integer; -- dummy integer to iterate columns with
col text; -- variable to keep the column name at each iteration
col_first text; -- first column name, e.g., top left corner on a csv file or spreadsheet
begin
set schema 'public';
create table insert_from_csv ();
-- add just enough number of columns
for iter in 1..col_count
loop
execute format('alter table insert_from_csv add column col_%s text;', iter);
end loop;
-- copy the data from csv file
execute format('copy insert_from_csv from %L with delimiter '','' quote ''"'' csv ', csv_path);
iter := 1;
col_first := (select col_1 from insert_from_csv limit 1);
-- update the column names based on the first row which has the column names
for col in execute format('select unnest(string_to_array(trim(temp_table::text, ''()''), '','')) from temp_table where col_1 = %L', col_first)
loop
execute format('alter table insert_from_csv rename column col_%s to %s', iter, col);
iter := iter + 1;
end loop;
-- delete the columns row
execute format('delete from insert_from_csv where %s = %L', col_first, col_first);
-- change the temp table name to the name given as parameter, if not blank
if length(target_table) > 0 then
execute format('alter table insert_from_csv rename to %I', target_table);
end if;
end;
$$ language plpgsql;
并将参数传递为
选择 load_csv_file('客户','C:\Insert_postgres.csv' ,4)
但收到错误消息
错误:必须是超级用户才能复制到文件或从文件中复制 提示:任何人都可以复制到标准输出或从标准输入。psql 的 \copy 命令也适用于任何人。
这个想法是,我将创建一个自动化测试,如果我想在不同的实例上进行测试,那么测试应该自动创建函数并从 csv 文件复制数据。没有超级用户可以复制数据吗?