mirror of https://github.com/postgres/postgres
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.2 KiB
109 lines
2.2 KiB
--
|
|
-- Test returning SETOF
|
|
--
|
|
CREATE FUNCTION test_setof_error() RETURNS SETOF text AS $$
|
|
return 37
|
|
$$ LANGUAGE plpythonu;
|
|
SELECT test_setof_error();
|
|
ERROR: returned object cannot be iterated
|
|
DETAIL: PL/Python set-returning functions must return an iterable object.
|
|
CONTEXT: PL/Python function "test_setof_error"
|
|
CREATE FUNCTION test_setof_as_list(count integer, content text) RETURNS SETOF text AS $$
|
|
return [ content ]*count
|
|
$$ LANGUAGE plpythonu;
|
|
CREATE FUNCTION test_setof_as_tuple(count integer, content text) RETURNS SETOF text AS $$
|
|
t = ()
|
|
for i in xrange(count):
|
|
t += ( content, )
|
|
return t
|
|
$$ LANGUAGE plpythonu;
|
|
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 plpythonu;
|
|
-- Test set returning functions
|
|
SELECT test_setof_as_list(0, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_list(1, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
list
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_list(2, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
list
|
|
list
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_list(2, null);
|
|
test_setof_as_list
|
|
--------------------
|
|
|
|
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_tuple(0, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_tuple(1, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
tuple
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_tuple(2, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
tuple
|
|
tuple
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_tuple(2, null);
|
|
test_setof_as_tuple
|
|
---------------------
|
|
|
|
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_iterator(0, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_iterator(1, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
list
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_iterator(2, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
list
|
|
list
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_iterator(2, null);
|
|
test_setof_as_iterator
|
|
------------------------
|
|
|
|
|
|
(2 rows)
|
|
|
|
|