mirror of
https://github.com/postgres/postgres.git
synced 2025-05-13 01:13:08 -04:00
db23464715
Since 19252e8ec93 we reject Python 2 during build configuration. Now that the dust on the buildfarm has settled, remove regression testing infrastructure dealing with differing output between Python 2 / 3. Reviewed-By: Peter Eisentraut <peter@eisentraut.org> Reviewed-By: Tom Lane <tgl@sss.pgh.pa.us> Discussion: https://postgr.es/m/20211031184548.g4sxfe47n2kyi55r@alap3.anarazel.de
98 lines
2.5 KiB
PL/PgSQL
98 lines
2.5 KiB
PL/PgSQL
--
|
|
-- Test returning SETOF
|
|
--
|
|
|
|
CREATE FUNCTION test_setof_error() RETURNS SETOF text AS $$
|
|
return 37
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT test_setof_error();
|
|
|
|
|
|
CREATE FUNCTION test_setof_as_list(count integer, content text) RETURNS SETOF text AS $$
|
|
return [ content ]*count
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION test_setof_as_tuple(count integer, content text) RETURNS SETOF text AS $$
|
|
t = ()
|
|
for i in range(count):
|
|
t += ( content, )
|
|
return t
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$
|
|
class producer:
|
|
def __init__ (self, icount, icontent):
|
|
self.icontent = icontent
|
|
self.icount = icount
|
|
def __iter__ (self):
|
|
return self
|
|
def __next__ (self):
|
|
if self.icount == 0:
|
|
raise StopIteration
|
|
self.icount -= 1
|
|
return self.icontent
|
|
return producer(count, content)
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION test_setof_spi_in_iterator() RETURNS SETOF text AS
|
|
$$
|
|
for s in ('Hello', 'Brave', 'New', 'World'):
|
|
plpy.execute('select 1')
|
|
yield s
|
|
plpy.execute('select 2')
|
|
$$
|
|
LANGUAGE plpython3u;
|
|
|
|
|
|
-- Test set returning functions
|
|
SELECT test_setof_as_list(0, 'list');
|
|
SELECT test_setof_as_list(1, 'list');
|
|
SELECT test_setof_as_list(2, 'list');
|
|
SELECT test_setof_as_list(2, null);
|
|
|
|
SELECT test_setof_as_tuple(0, 'tuple');
|
|
SELECT test_setof_as_tuple(1, 'tuple');
|
|
SELECT test_setof_as_tuple(2, 'tuple');
|
|
SELECT test_setof_as_tuple(2, null);
|
|
|
|
SELECT test_setof_as_iterator(0, 'list');
|
|
SELECT test_setof_as_iterator(1, 'list');
|
|
SELECT test_setof_as_iterator(2, 'list');
|
|
SELECT test_setof_as_iterator(2, null);
|
|
|
|
SELECT test_setof_spi_in_iterator();
|
|
|
|
-- set-returning function that modifies its parameters
|
|
CREATE OR REPLACE FUNCTION ugly(x int, lim int) RETURNS SETOF int AS $$
|
|
global x
|
|
while x <= lim:
|
|
yield x
|
|
x = x + 1
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT ugly(1, 5);
|
|
|
|
-- interleaved execution of such a function
|
|
SELECT ugly(1,3), ugly(7,8);
|
|
|
|
-- returns set of named-composite-type tuples
|
|
CREATE OR REPLACE FUNCTION get_user_records()
|
|
RETURNS SETOF users
|
|
AS $$
|
|
return plpy.execute("SELECT * FROM users ORDER BY username")
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT get_user_records();
|
|
SELECT * FROM get_user_records();
|
|
|
|
-- same, but returning set of RECORD
|
|
CREATE OR REPLACE FUNCTION get_user_records2()
|
|
RETURNS TABLE(fname text, lname text, username text, userid int)
|
|
AS $$
|
|
return plpy.execute("SELECT * FROM users ORDER BY username")
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT get_user_records2();
|
|
SELECT * FROM get_user_records2();
|