Passing a ResultSet into a Postgresql Function

Question

Is it possible to pass the results of a postgres query as an input into another function?

As a very contrived example, say I have one query like

SELECT id, name
FROM users
LIMIT 50

and I want to create a function my_function that takes the resultset of the first query and returns the minimum id. Is this possible in pl/pgsql?

SELECT my_function(SELECT id, name FROM Users LIMIT 50); --returns 50

score 5 · Answer 1 · edited May 23 '17 at 10:31

You could use a cursor, but that very impractical for computing a minimum.

I would use a temporary table for that purpose, and pass the table name for use in dynamic SQL:

CREATE OR REPLACE FUNCTION f_min_id(_tbl regclass, OUT min_id int) AS 
$func$
BEGIN

EXECUTE 'SELECT min(id) FROM ' || _tbl
INTO min_id;

END  
$func$ LANGUAGE plpgsql;

Call:

CREATE TEMP TABLE foo ON COMMIT DROP AS
SELECT id, name
FROM   users
LIMIT  50;

SELECT f_min_id('foo');

Major points

The first parameter is of type regclass to prevent SQL injection. More info in this related answer on dba.SE.
I made the temp table ON COMMIT DROP to limit its lifetime to the current transaction. May or may not be what you want.
You can extend this example to take more parameters. Search for code examples for dynamic SQL with EXECUTE.

-> SQLfiddle demo

score 5 · Answer 2 · answered Oct 30 '13 at 22:14

5

I would take the problem on the other side, calling an aggregate function for each record of the result set. It's not as flexible but can gives you an hint to work on.

As an exemple to follow your sample problem:

CREATE OR REPLACE FUNCTION myMin ( int,int ) RETURNS int AS $$
  SELECT CASE WHEN $1 < $2 THEN $1 ELSE $2 END;
$$ LANGUAGE SQL STRICT IMMUTABLE;

CREATE AGGREGATE my_function ( int ) (
    SFUNC = myMin, STYPE = int, INITCOND = 2147483647 --maxint
);

SELECT my_function(id) from (SELECT * FROM Users LIMIT 50) x;

answered Oct 30 '13 at 22:14

Le Droid

4,534
3
37
32

Just to be clear, in the simple case of min, your function/aggregate definitions wouldn't be necessary, right? You'd get the same effect from ```SELECT min(id) from (SELECT * FROM Users LIMIT 50) x```, I believe. So your answer doesn't answer the general question but shows how to define an arbitrary aggregation function and apply it to a result set. If I've misunderstood, please let me know. – Sigfried Jan 22 '17 at 12:13
@Sigfried Yes, it's just a theoretical sample as the question is. – Le Droid Jan 23 '17 at 21:27

score 3 · Answer 3 · answered Oct 11 '13 at 06:18

3

It is not possible to pass an array of generic type RECORD to a plpgsql function which is essentially what you are trying to do.

What you can do is pass in an array of a specific user defined TYPE or of a particular table row type. In the example below you could also swap out the argument data type for the table name users[] (though this would obviously mean getting all data in the users table row).

CREATE TYPE trivial {
"ID" integer,
"NAME" text
}

CREATE OR REPLACE FUNCTION trivial_func(data trivial[])
  RETURNS integer AS
$BODY$
DECLARE 

BEGIN
    --Implementation here using data
    return 1;

END$BODY$
  LANGUAGE 'plpgsql' VOLATILE;

answered Oct 11 '13 at 06:18

Fast Engy

1,913
1
12
10

2

That `CREATE TYPE` command fails as a syntax error on PostgreSQL version 9.4.4. – Adam Mackler Jul 14 '15 at 15:58
1

Thanks @AdamMackler - CREATE TYPE trivial AS { } in 9+ – Fast Engy Aug 11 '15 at 00:52
2

Can you please provide an example call of this function? – Paweł Sopel Oct 03 '17 at 09:57

score 2 · Answer 4 · answered Aug 14 '13 at 04:41

I think there's no way to pass recordset or table into function (but I'd be glad if i'm wrong). Best I could suggest is to pass array:

create or replace function my_function(data int[])
returns int
as
$$
   select min(x) from unnest(data) as x
$$
language SQL;

sql fiddle demo

Passing a ResultSet into a Postgresql Function

4 Answers4

Major points

Linked