PostgreeSQL: добавление функций
Часто чтобы не плодить логику в приложении, проще часть функционала по вычислению чего-либо возложить на сервер БД. Для этого и нужны функции. Причем в Postgree мне они понравились много больше чем в MySQL в плане простоты и удобства.
Любая функция в PostgreeSQL начинается с объявления функции, перечисления используемых переменных, тела функции и в конце возврат результата выполнения функции.
Объявление функции:
CREATE OR REPLACE FUNCTION public.get_calculate_station_status(
station_id integer) -- на входе параметр station_id - число
RETURNS integer -- возвращает число
LANGUAGE 'plpgsql' -- язык
plpgsql
AS $BODY$Объявление переменных, идет после команды DECLARE. перечисляются заранее все переменные с указанием их типов, которые будут использоваться внутри функции. Например:
DECLARE
status integer; — число
st_online boolean; — true/false
stations integer[]; — массив чисел
Сама функция пишется непосредственно между BEGIN и END; Доступны элементарные условия:
IF aa=bb THEN
bb:=1;
ELSE
cc:=2;
END IF;Поместить результат запроса в переменную или в массив:
select zz from xx into aa where vv=1;Перебор массива:
FOREACH st_id IN ARRAY stations
LOOP
connectors_free:=connectors_free+get_count_connectors_free(st_id);
stations_online:=stations_online+1;
END LOOP;Результирующий пример готовой функции:
DECLARE
status integer;
stations integer[];
connectors_free integer;
stations_online integer;
st_id integer;
BEGIN
status:=get_ezs_status(ezs_id);
if status=1 THEN
connectors_free:=0;
stations_online:=0;
stations:=ARRAY (select station from stations_online inner join stations on stations.id=station where stations.ezs=ezs_id and stations.status=1 and deleted=false);
FOREACH st_id IN ARRAY stations
LOOP
connectors_free:=connectors_free+get_count_connectors_free(st_id);
stations_online:=stations_online+1;
END LOOP;
IF stations_online>0 THEN
if connectors_free>0 THEN
status:=1;
ELSE
status:=4;
END IF;
ELSE
status:=5;
END IF;
END IF;
return status;
END;
