83 lines
2.0 KiB
Plaintext
83 lines
2.0 KiB
Plaintext
-- directory paths are passed to us in environment variables
|
|
\getenv abs_srcdir PG_ABS_SRCDIR
|
|
\getenv abs_builddir PG_ABS_BUILDDIR
|
|
-- set up file names to use
|
|
\set srcfilename :abs_srcdir '/data/copy1.data'
|
|
\set destfilename :abs_builddir '/results/copy1.data'
|
|
CREATE TABLE copy1 (a int, b float);
|
|
-- COPY TO/FROM not authorized from client.
|
|
DO LANGUAGE plpgsql $$
|
|
BEGIN
|
|
COPY copy1 TO stdout;
|
|
END;
|
|
$$;
|
|
ERROR: cannot COPY to/from client in PL/pgSQL
|
|
CONTEXT: PL/pgSQL function inline_code_block line 3 at SQL statement
|
|
DO LANGUAGE plpgsql $$
|
|
BEGIN
|
|
COPY copy1 FROM stdin;
|
|
END;
|
|
$$;
|
|
ERROR: cannot COPY to/from client in PL/pgSQL
|
|
CONTEXT: PL/pgSQL function inline_code_block line 3 at SQL statement
|
|
DO LANGUAGE plpgsql $$
|
|
BEGIN
|
|
EXECUTE 'COPY copy1 TO stdout';
|
|
END;
|
|
$$;
|
|
ERROR: cannot COPY to/from client in PL/pgSQL
|
|
CONTEXT: PL/pgSQL function inline_code_block line 3 at EXECUTE
|
|
DO LANGUAGE plpgsql $$
|
|
BEGIN
|
|
EXECUTE 'COPY copy1 FROM stdin';
|
|
END;
|
|
$$;
|
|
ERROR: cannot COPY to/from client in PL/pgSQL
|
|
CONTEXT: PL/pgSQL function inline_code_block line 3 at EXECUTE
|
|
-- Valid cases
|
|
-- COPY FROM
|
|
\set dobody 'BEGIN COPY copy1 FROM ' :'srcfilename' '; END'
|
|
DO LANGUAGE plpgsql :'dobody';
|
|
SELECT * FROM copy1 ORDER BY 1;
|
|
a | b
|
|
---+-----
|
|
1 | 1.1
|
|
2 | 2.2
|
|
3 | 3.3
|
|
(3 rows)
|
|
|
|
TRUNCATE copy1;
|
|
\set cmd 'COPY copy1 FROM ' :'srcfilename'
|
|
\set dobody 'BEGIN EXECUTE ' :'cmd' '; END'
|
|
DO LANGUAGE plpgsql :'dobody';
|
|
SELECT * FROM copy1 ORDER BY 1;
|
|
a | b
|
|
---+-----
|
|
1 | 1.1
|
|
2 | 2.2
|
|
3 | 3.3
|
|
(3 rows)
|
|
|
|
-- COPY TO
|
|
-- Copy the data externally once, then process it back to the table.
|
|
\set dobody 'BEGIN COPY copy1 TO ' :'destfilename' '; END'
|
|
DO LANGUAGE plpgsql :'dobody';
|
|
TRUNCATE copy1;
|
|
\set dobody 'BEGIN COPY copy1 FROM ' :'destfilename' '; END'
|
|
DO LANGUAGE plpgsql :'dobody';
|
|
\set cmd 'COPY copy1 FROM ' :'destfilename'
|
|
\set dobody 'BEGIN EXECUTE ' :'cmd' '; END'
|
|
DO LANGUAGE plpgsql :'dobody';
|
|
SELECT * FROM copy1 ORDER BY 1;
|
|
a | b
|
|
---+-----
|
|
1 | 1.1
|
|
1 | 1.1
|
|
2 | 2.2
|
|
2 | 2.2
|
|
3 | 3.3
|
|
3 | 3.3
|
|
(6 rows)
|
|
|
|
DROP TABLE copy1;
|