mirror of
https://github.com/postgres/postgres.git
synced 2025-06-02 00:01:40 -04:00
d727c54317
If we recursed to a new call of the same function, with a different coldeflist (AS clause), it would fail because the inner call would overwrite the outer call's idea of what to return. This is vaguely like 1d2fe56e4 and c5bec5426, but it's not due to any API decisions: it's just that we computed the actual output rowtype at the start of the call, and saved it in the per-procedure data structure. We can fix it at basically zero cost by doing the computation at the end of each call instead of the start. It's not clear that there's any real-world use-case for such a function, but given that it doesn't cost anything to fix, it'd be silly not to. Per report from Andreas Karlsson. Back-patch to all supported branches. Discussion: https://postgr.es/m/1651a46d-3c15-4028-a8c1-d74937b54e19@proxel.se
236 lines
8.6 KiB
PL/PgSQL
236 lines
8.6 KiB
PL/PgSQL
CREATE FUNCTION multiout_simple(OUT i integer, OUT j integer) AS $$
|
|
return (1, 2)
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT multiout_simple();
|
|
SELECT * FROM multiout_simple();
|
|
SELECT i, j + 2 FROM multiout_simple();
|
|
SELECT (multiout_simple()).j + 3;
|
|
|
|
CREATE FUNCTION multiout_simple_setof(n integer = 1, OUT integer, OUT integer) RETURNS SETOF record AS $$
|
|
return [(1, 2)] * n
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT multiout_simple_setof();
|
|
SELECT * FROM multiout_simple_setof();
|
|
SELECT * FROM multiout_simple_setof(3);
|
|
|
|
CREATE FUNCTION multiout_record_as(typ text,
|
|
first text, OUT first text,
|
|
second integer, OUT second integer,
|
|
retnull boolean) RETURNS record AS $$
|
|
if retnull:
|
|
return None
|
|
if typ == 'dict':
|
|
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' }
|
|
elif typ == 'tuple':
|
|
return ( first, second )
|
|
elif typ == 'list':
|
|
return [ first, second ]
|
|
elif typ == 'obj':
|
|
class type_record: pass
|
|
type_record.first = first
|
|
type_record.second = second
|
|
return type_record
|
|
elif typ == 'str':
|
|
return "('%s',%r)" % (first, second)
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM multiout_record_as('dict', 'foo', 1, 'f');
|
|
SELECT multiout_record_as('dict', 'foo', 1, 'f');
|
|
|
|
SELECT * FROM multiout_record_as('dict', null, null, false);
|
|
SELECT * FROM multiout_record_as('dict', 'one', null, false);
|
|
SELECT * FROM multiout_record_as('dict', null, 2, false);
|
|
SELECT * FROM multiout_record_as('dict', 'three', 3, false);
|
|
SELECT * FROM multiout_record_as('dict', null, null, true);
|
|
|
|
SELECT * FROM multiout_record_as('tuple', null, null, false);
|
|
SELECT * FROM multiout_record_as('tuple', 'one', null, false);
|
|
SELECT * FROM multiout_record_as('tuple', null, 2, false);
|
|
SELECT * FROM multiout_record_as('tuple', 'three', 3, false);
|
|
SELECT * FROM multiout_record_as('tuple', null, null, true);
|
|
|
|
SELECT * FROM multiout_record_as('list', null, null, false);
|
|
SELECT * FROM multiout_record_as('list', 'one', null, false);
|
|
SELECT * FROM multiout_record_as('list', null, 2, false);
|
|
SELECT * FROM multiout_record_as('list', 'three', 3, false);
|
|
SELECT * FROM multiout_record_as('list', null, null, true);
|
|
|
|
SELECT * FROM multiout_record_as('obj', null, null, false);
|
|
SELECT * FROM multiout_record_as('obj', 'one', null, false);
|
|
SELECT * FROM multiout_record_as('obj', null, 2, false);
|
|
SELECT * FROM multiout_record_as('obj', 'three', 3, false);
|
|
SELECT * FROM multiout_record_as('obj', null, null, true);
|
|
|
|
SELECT * FROM multiout_record_as('str', 'one', 1, false);
|
|
SELECT * FROM multiout_record_as('str', 'one', 2, false);
|
|
|
|
SELECT *, s IS NULL AS snull FROM multiout_record_as('tuple', 'xxx', NULL, 'f') AS f(f, s);
|
|
SELECT *, f IS NULL AS fnull, s IS NULL AS snull FROM multiout_record_as('tuple', 'xxx', 1, 't') AS f(f, s);
|
|
SELECT * FROM multiout_record_as('obj', NULL, 10, 'f');
|
|
|
|
CREATE FUNCTION multiout_setof(n integer,
|
|
OUT power_of_2 integer,
|
|
OUT length integer) RETURNS SETOF record AS $$
|
|
for i in range(n):
|
|
power = 2 ** i
|
|
length = plpy.execute("select length('%d')" % power)[0]['length']
|
|
yield power, length
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM multiout_setof(3);
|
|
SELECT multiout_setof(5);
|
|
|
|
CREATE FUNCTION multiout_return_table() RETURNS TABLE (x integer, y text) AS $$
|
|
return [{'x': 4, 'y' :'four'},
|
|
{'x': 7, 'y' :'seven'},
|
|
{'x': 0, 'y' :'zero'}]
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM multiout_return_table();
|
|
|
|
CREATE FUNCTION multiout_array(OUT integer[], OUT text) RETURNS SETOF record AS $$
|
|
yield [[1], 'a']
|
|
yield [[1,2], 'b']
|
|
yield [[1,2,3], None]
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM multiout_array();
|
|
|
|
CREATE FUNCTION singleout_composite(OUT type_record) AS $$
|
|
return {'first': 1, 'second': 2}
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION multiout_composite(OUT type_record) RETURNS SETOF type_record AS $$
|
|
return [{'first': 1, 'second': 2},
|
|
{'first': 3, 'second': 4 }]
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM singleout_composite();
|
|
SELECT * FROM multiout_composite();
|
|
|
|
-- composite OUT parameters in functions returning RECORD not supported yet
|
|
CREATE FUNCTION multiout_composite(INOUT n integer, OUT type_record) AS $$
|
|
return (n, (n * 2, n * 3))
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION multiout_table_type_setof(typ text, returnnull boolean, INOUT n integer, OUT table_record) RETURNS SETOF record AS $$
|
|
if returnnull:
|
|
d = None
|
|
elif typ == 'dict':
|
|
d = {'first': n * 2, 'second': n * 3, 'extra': 'not important'}
|
|
elif typ == 'tuple':
|
|
d = (n * 2, n * 3)
|
|
elif typ == 'list':
|
|
d = [ n * 2, n * 3 ]
|
|
elif typ == 'obj':
|
|
class d: pass
|
|
d.first = n * 2
|
|
d.second = n * 3
|
|
elif typ == 'str':
|
|
d = "(%r,%r)" % (n * 2, n * 3)
|
|
for i in range(n):
|
|
yield (i, d)
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM multiout_composite(2);
|
|
SELECT * FROM multiout_table_type_setof('dict', 'f', 3);
|
|
SELECT * FROM multiout_table_type_setof('dict', 'f', 7);
|
|
SELECT * FROM multiout_table_type_setof('tuple', 'f', 2);
|
|
SELECT * FROM multiout_table_type_setof('tuple', 'f', 3);
|
|
SELECT * FROM multiout_table_type_setof('list', 'f', 2);
|
|
SELECT * FROM multiout_table_type_setof('list', 'f', 3);
|
|
SELECT * FROM multiout_table_type_setof('obj', 'f', 4);
|
|
SELECT * FROM multiout_table_type_setof('obj', 'f', 5);
|
|
SELECT * FROM multiout_table_type_setof('str', 'f', 6);
|
|
SELECT * FROM multiout_table_type_setof('str', 'f', 7);
|
|
SELECT * FROM multiout_table_type_setof('dict', 't', 3);
|
|
|
|
-- check what happens if a type changes under us
|
|
|
|
CREATE TABLE changing (
|
|
i integer,
|
|
j integer
|
|
);
|
|
|
|
CREATE FUNCTION changing_test(OUT n integer, OUT changing) RETURNS SETOF record AS $$
|
|
return [(1, {'i': 1, 'j': 2}),
|
|
(1, (3, 4))]
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM changing_test();
|
|
ALTER TABLE changing DROP COLUMN j;
|
|
SELECT * FROM changing_test();
|
|
SELECT * FROM changing_test();
|
|
ALTER TABLE changing ADD COLUMN j integer;
|
|
SELECT * FROM changing_test();
|
|
|
|
-- tables of composite types
|
|
|
|
CREATE FUNCTION composite_types_table(OUT tab table_record[], OUT typ type_record[] ) RETURNS SETOF record AS $$
|
|
yield {'tab': [('first', 1), ('second', 2)],
|
|
'typ': [{'first': 'third', 'second': 3},
|
|
{'first': 'fourth', 'second': 4}]}
|
|
yield {'tab': [('first', 1), ('second', 2)],
|
|
'typ': [{'first': 'third', 'second': 3},
|
|
{'first': 'fourth', 'second': 4}]}
|
|
yield {'tab': [('first', 1), ('second', 2)],
|
|
'typ': [{'first': 'third', 'second': 3},
|
|
{'first': 'fourth', 'second': 4}]}
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM composite_types_table();
|
|
|
|
-- check what happens if the output record descriptor changes
|
|
CREATE FUNCTION return_record(t text) RETURNS record AS $$
|
|
return {'t': t, 'val': 10}
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM return_record('abc') AS r(t text, val integer);
|
|
SELECT * FROM return_record('abc') AS r(t text, val bigint);
|
|
SELECT * FROM return_record('abc') AS r(t text, val integer);
|
|
SELECT * FROM return_record('abc') AS r(t varchar(30), val integer);
|
|
SELECT * FROM return_record('abc') AS r(t varchar(100), val integer);
|
|
SELECT * FROM return_record('999') AS r(val text, t integer);
|
|
|
|
CREATE FUNCTION return_record_2(t text) RETURNS record AS $$
|
|
return {'v1':1,'v2':2,t:3}
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM return_record_2('v3') AS (v3 int, v2 int, v1 int);
|
|
SELECT * FROM return_record_2('v3') AS (v2 int, v3 int, v1 int);
|
|
SELECT * FROM return_record_2('v4') AS (v1 int, v4 int, v2 int);
|
|
SELECT * FROM return_record_2('v4') AS (v1 int, v4 int, v2 int);
|
|
-- error
|
|
SELECT * FROM return_record_2('v4') AS (v1 int, v3 int, v2 int);
|
|
-- works
|
|
SELECT * FROM return_record_2('v3') AS (v1 int, v3 int, v2 int);
|
|
SELECT * FROM return_record_2('v3') AS (v1 int, v2 int, v3 int);
|
|
|
|
-- recursion with a different inner result type didn't use to work
|
|
CREATE FUNCTION return_record_3(t text) RETURNS record AS $$
|
|
if t == "text":
|
|
plpy.execute("SELECT * FROM return_record_3('int') AS (a int)");
|
|
return { "a": "x" }
|
|
elif t == "int":
|
|
return { "a": 1 }
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
SELECT * FROM return_record_3('text') AS (a text);
|
|
|
|
-- multi-dimensional array of composite types.
|
|
CREATE FUNCTION composite_type_as_list() RETURNS type_record[] AS $$
|
|
return [[('first', 1), ('second', 1)], [('first', 2), ('second', 2)], [('first', 3), ('second', 3)]];
|
|
$$ LANGUAGE plpython3u;
|
|
SELECT * FROM composite_type_as_list();
|
|
|
|
-- Starting with PostgreSQL 10, a composite type in an array cannot be
|
|
-- represented as a Python list, because it's ambiguous with multi-dimensional
|
|
-- arrays. So this throws an error now. The error should contain a useful hint
|
|
-- on the issue.
|
|
CREATE FUNCTION composite_type_as_list_broken() RETURNS type_record[] AS $$
|
|
return [['first', 1]];
|
|
$$ LANGUAGE plpython3u;
|
|
SELECT * FROM composite_type_as_list_broken();
|