database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-12-12 02:37:31 +03:00
Code Activity

Fix aboriginal mistake in plpython's set-returning-function support.

Browse Source 
We must stay in the function's SPI context until done calling the iterator
that returns the set result.  Otherwise, any attempt to invoke SPI features
in the python code called by the iterator will malfunction.  Diagnosis and
patch by Jan Urbanski, per bug report from Jean-Baptiste Quenot.

Back-patch to 8.2; there was no support for SRFs in previous versions of
plpython.
This commit is contained in:
Tom Lane
2010-11-15 14:26:55 -05:00
parent 3134d8863e
commit add0ea88e7
3 changed files with 48 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/pl/plpython/sql/plpython_setof.sql
View File

@@ -35,6 +35,15 @@ class producer:
return producer(count, content)
$$ LANGUAGE plpythonu;
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 plpythonu;
-- Test set returning functions
SELECT test_setof_as_list(0, 'list');
@@ -51,3 +60,5 @@ 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();