Функции PL/pgSQL: как вернуть нормальную таблицу с несколькими столбцами, используя оператор execute

У меня есть функция PL/pgSQL, которая должна возвращать информацию о некоторых пользователях.

CREATE OR REPLACE FUNCTION my_function(
        user_id integer
    ) RETURNS TABLE(
            id integer, 
            firstname character varying,
            lastname  character varying
        ) AS $$
    DECLARE
        ids character varying;
    BEGIN
        ids := '';
        --Some code which build the ids string, not interesting for this issue
        RETURN QUERY 
            EXECUTE 'SELECT 
                        users.id, 
                        users.firstname, 
                        users.lastname
                    FROM public.users 
                    WHERE ids IN (' || ids || ')';
    END;
$$ LANGUAGE plpgsql;

Проблема, с которой я сталкиваюсь, состоит в том, что результатом функции является таблица из одного столбца, например:

╔═══╦═════════════════════╗
║   ║my_function          ║
╠═══╬═════════════════════╣
║ 1 ║ (106,Ned,STARK)     ║
║ 2 ║ (130,Rob,STARK)     ║
╚═══╩═════════════════════╝

Пока я ожидал:

╔═══╦════════════╦════════════╦═════════════╗
║   ║ id         ║ firstname  ║ lastname    ║
╠═══╬════════════╬════════════╬═════════════╣
║ 1 ║ 106        ║ Ned        ║ STARK       ║
║ 2 ║ 103        ║ Rob        ║ STARK       ║
╚═══╩════════════╩════════════╩═════════════╝

Я думаю (но не уверен), что проблема связана с оператором EXECUTE, но я не понимаю, как поступить иначе.

Есть идеи?

Ответы

Ответ 1

Как вы выполняете эту функцию? Это работает как оператор выбора.

Создать таблицу: public.users

create table public.users (id int, firstname varchar, lastname varchar);

Вставьте несколько записей:

insert into public.users values (1, 'aaa','bbb'),(2,'ccc','ddd');

функция: моя_функция

CREATE OR REPLACE FUNCTION my_function(user_id integer) RETURNS TABLE(id integer, firstname character varying, lastname character varying) AS $$
    DECLARE
        ids INTEGER[];
    BEGIN
         ids := ARRAY[1,2];
         RETURN QUERY
             SELECT users.id, users.firstname, users.lastname
             FROM public.users
             WHERE users.id = ANY(ids);
    END;
$$ LANGUAGE plpgsql;

Теперь вы можете использовать с *

select * from my_function(1);

Результат запроса

 id | firstname | lastname 
----+-----------+----------
  1 | aaa       | bbb
  2 | ccc       | ddd

Или с именами столбцов

select id,firstname,lastname from my_function(1);

Результат

 id | firstname | lastname 
----+-----------+----------
  1 | aaa       | bbb
  2 | ccc       | ddd

Ответ 2

Вызов функции так:

select * from  my_function(123);

Не только с выбором. Я сделал, и это работает