PLPGSQL отличается, упорядочить, ничего не возвращает - PullRequest
Новая
       27

PLPGSQL отличается, упорядочить, ничего не возвращает

0 голосов
/ 13 декабря 2010

Я сделал эту функцию plpgsql, и она мне ничего не возвращает! в то время как если я возьму часть запроса и выполню ее в отдельном окне sql, он вернет правильные строки.

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

CREATE OR REPLACE FUNCTION get_members(in_company_uuid uuid, in_start integer, in_limit integer, in_sort character varying, in_order character varying, OUT out_status integer, OUT out_status_description character varying, OUT out_value character varying[]) RETURNS SETOF record
LANGUAGE plpgsql
AS $$DECLARE

temp_record RECORD;
temp_out_value VARCHAR[];
temp_iterator INTEGER := 0;

BEGIN

FOR temp_record IN EXECUTE '
SELECT DISTINCT ON
(' || in_sort || ')
u.user_uuid,
u.firstname,
u.preposition,
u.lastname,
array(SELECT email FROM emails WHERE user_uuid = u.user_uuid) as emails,
array(SELECT mobilenumber FROM mobilenumbers WHERE user_uuid = u.user_uuid) as mobilenumbers,
array(SELECT c.name FROM targetgroupusers AS tgu LEFT JOIN membercategories as mc ON mc.targetgroup_uuid = tgu.targetgroup_uuid LEFT JOIN categories AS c ON mc.category_uuid = c.category_uuid WHERE tgu.user_uuid = u.user_uuid) as categories,
array(SELECT color FROM membercategories WHERE targetgroup_uuid IN(SELECT targetgroup_uuid FROM targetgroupusers WHERE user_uuid = u.user_uuid)) as colors
FROM
    membercategories AS mc
LEFT JOIN
    targetgroups AS tg
ON
    tg.targetgroup_uuid = mc.targetgroup_uuid
LEFT JOIN
    targetgroupusers AS tgu
ON
    tgu.targetgroup_uuid = tg.targetgroup_uuid
LEFT JOIN
    users AS u
ON
    u.user_uuid = tgu.user_uuid
WHERE
    mc.company_uuid = \'' || in_company_uuid || '\'
ORDER BY
   ' || in_sort || ' ' || in_order || '
OFFSET
    ' || in_start || '
LIMIT
    ' || in_limit

LOOP
  temp_out_value[temp_iterator] = ARRAY[temp_record.user_uuid::VARCHAR(36), temp_record.firstname::CHARACTER VARYING, temp_record.preposition::CHARACTER VARYING, temp_record.lastname::CHARACTER VARYING, temp_record.emails::CHARACTER VARYING, temp_record.mobilenumbers::CHARACTER VARYING, temp_record.categories::CHARACTER VARYING, temp_record.colors::CHARACTER VARYING];
  temp_iterator = temp_iterator+1;
END LOOP;

out_status := 0;
out_status_description := 'Members retrieved';
out_value := temp_out_value;
RETURN;

END$$;

Большое спасибо!

Ответы [ 3 ]

1 голос
/ 13 декабря 2010

Вам нужно использовать RETURN NEXT temp_record вместо "просто" ВОЗВРАТА, чтобы вернуть полный набор результатов.

В этой части руководства объясняется, как обрабатывать функцию «ВОЗВРАТ НАСТРОЙКИ»:

В этом случае отдельные возвращаемые элементы определяются последовательностью команд RETURN NEXT или RETURN QUERY, а затем используется последняя команда RETURN без аргумента, указывающая, что функция завершила выполнение

0 голосов
/ 13 декабря 2010

Вам нужно использовать 

CREATE OR REPLACE FUNCTION get_members(...) RETURNS record LANGUAGE plpgsql AS
...

, то есть удалить setof, что является вашей проблемой, поскольку a_horse упоминает

Но имейте в виду, что вы получаете только 1 строку, подобную этой.Вы можете удалить этот массив, если вам нужно несколько строк, следуя этому примеру: 

create or replace function unnest(anyarray) returns setof anyelement as $$
  select $1[i] from generate_series(array_lower($1,1), array_upper($1,1)) i;
$$ language'sql' immutable;

create function func(out text, out text[]) returns record language plpgsql as $$
begin
  $1='success';
  $2[1]='hello';
  $2[2]='there';
  $2[3]='Koen';
  return;
end;$$;

select * from unnest((select column2 from func()));

 unnest
--------
 hello
 there
 Koen

Если у вас версия 8.4 или выше, то нет необходимости создавать собственную функцию unnest, как я сделал здесь.

0 голосов
/ 13 декабря 2010

Не используйте \ для выхода, используйте кавычки: 

\'' || in_company_uuid || '\'

''' || in_company_uuid || '''

Но я бы не использовал функцию для этого, просто запрос SQL. Используйте EXPLAIN, чтобы увидеть, как он выполняется и где можно найти проблемы с производительностью. Это не очень хороший кусок SQL ...

Ps. Ваша функция также уязвима для SQL-инъекций, вы вводите userinput без выхода в SQL.

...