functions/use_parallel.sql - master - telnov.ob/postgresql-patterns-library

functions/use_parallel.sql
78 строк3 KB
Rinat Mukhtarov
`public.` added to many functions
07 ноя 2023, 23:34
07 ноя 2023, 23:341b49317
-- Внимание!
-- Вызов этой функции при вызове миллионов раз занимает драгоценное время.
-- Если это очень критично, вместо этой функции используйте выражение `abs(id) % core_max = core_num - 1` в коде условия SQL запроса.

create or replace function public.use_parallel(
    id bigint,
    core_num integer,
    core_max integer
)
    -- Функция предназначена для распараллеливания SQL запросов
    -- Принимает решение, использовать преданный номер ядра процессора или нет
    -- Пример: WHERE public.use_parallel(id, 1, 5)
    returns boolean
    immutable
    returns null on null input
    parallel safe
    language plpgsql
    set search_path = ''
    cost 2
as
$$
BEGIN
    IF core_num NOT BETWEEN 1 AND 256 THEN
        RAISE EXCEPTION 'Argument core_num must be between 1 and 256, but % given!', core_num;
    ELSIF core_max NOT BETWEEN 1 AND 256 THEN
        RAISE EXCEPTION 'Argument core_max must be between 1 and 256, but % given!', core_max;
    ELSIF core_num > core_max THEN
        RAISE EXCEPTION 'Argument core_num must be <= core_max! Given core_num = %, core_max = %', core_num, core_max;
    END IF;

    RETURN abs(id) % core_max = core_num - 1;
END;
$$;

--TEST
select
    sum(public.use_parallel(g, 1, 3)::int), --3333
    sum(public.use_parallel(g, 2, 3)::int), --3334
    sum(public.use_parallel(g, 3, 3)::int) --3333
from generate_series(1, 10000) as g;

------------------------------------------------------------------------------------------------------------------------
create or replace function public.use_parallel(
    str text,
    core_num integer,
    core_max integer
)
    -- Функция предназначена для распараллеливания SQL запросов
    -- Принимает решение, использовать преданный номер процессора или нет
    -- Пример: WHERE public.use_parallel('mike@domain.com', 1, 5)
    returns boolean
    immutable
    returns null on null input
    parallel safe
    language plpgsql
    set search_path = ''
    cost 11
as
$$
BEGIN
    IF core_num NOT BETWEEN 1 AND 256 THEN
        RAISE EXCEPTION 'Argument core_num must be between 1 and 256, but % given!', core_num;
    ELSIF core_max NOT BETWEEN 1 AND 256 THEN
        RAISE EXCEPTION 'Argument core_max must be between 1 and 256, but % given!', core_max;
    ELSIF core_num > core_max THEN
        RAISE EXCEPTION 'Argument core_num must be <= core_max! Given core_num = %, core_max = %', core_num, core_max;
    END IF;

    RETURN abs(public.crc32(str)) % core_max = core_num - 1;
END;
$$;

--TEST
select
    sum(public.use_parallel(g::text, 1, 3)::int), --3307
    sum(public.use_parallel(g::text, 2, 3)::int), --3358
    sum(public.use_parallel(g::text, 3, 3)::int) --3335
from generate_series(1, 10000) as g;