sql >> Base de Datos >  >> RDS >> PostgreSQL

OBTENER DIAGNÓSTICOS con instrucción COPY en la función Pl/pgsql

Ahora GET DIAGNOSTIC devolverá el número de filas procesadas por la instrucción COPY en una función Pl/Pgsql.
Instrucción COPY en la función Pl/Pgsql:

CREATE OR REPLACE FUNCTION public.copy_data(fname text) RETURNS integer
AS
$$
declare
copy_qry text;
cnt integer;
Begin
copy_qry := 'copy t from'||quote_literal(fname)||' with CSV HEADER;';
Execute copy_qry;
GET DIAGNOSTICS cnt = ROW_COUNT;
return cnt;
end;
$$ Language plpgsql;

Lanzamiento anterior:

-bash-4.1$ psql
psql.bin (9.2.3)
Type "help" for help.

postgres=# select copy_data('/usr/local/pg93beta/t_load.csv');
copy_data
-----------
0
(1 row)

En PostgreSQL 9.3

-bash-4.1$ ./psql -p 5555
psql (9.3beta1)
Type "help" for help.

postgres=# select copy_data('/usr/local/pg93beta/t_load.csv');
copy_data
-----------
3
(1 row)
or
-bash-4.1$ ./psql -p 5555 -c "select copy_data('/usr/local/pg93beta/t_load.csv');"
copy_data
-----------
3
(1 row)

Gracias al autor, parece simple pero muy eficaz cuando se trabaja con la carga de datos mediante scripts y se desea saber el número de filas procesadas por la instrucción COPY.