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

functions/benchmark.sql
81 строка3 KB
rin-nas
old pg10 comment removed
13 апр 2026, 11:45
13 апр 2026, 11:45233b866
CREATE OR REPLACE FUNCTION public.benchmark(loop_count int, sql_expr text)
    returns interval
    volatile --!!!
    returns null on null input -- = strict
    parallel unsafe --!!!
    security invoker
    language plpgsql
    set search_path = ''
AS
$$
DECLARE
    sql text;
    started_at timestamptz;
BEGIN
    sql := concat('select (', sql_expr, ') is null from generate_series(1, $1)');
    started_at := clock_timestamp();
    EXECUTE sql USING loop_count;
    RETURN clock_timestamp() - started_at;
END
$$;

comment on function public.benchmark(loop_count int, sql_expr text) is $$
    Measures the speed of expressions and functions for a given number of calls. Returns period of time.
    PostgreSQL equivalent of MySQL's BENCHMARK() function.
$$;

------------------------------------------------------------------------------------------------------------------------
CREATE OR REPLACE FUNCTION public.benchmark(timeout interval, sql_expr text)
    returns int
    volatile --!!!
    returns null on null input -- = strict
    parallel unsafe --!!!
    security invoker
    language plpgsql
    set search_path = ''
AS
$$
DECLARE
    sql text;
    loop_count int;
BEGIN
    sql := concat('with recursive r (i, b) as (
                        select 1, (', sql_expr, ') is null
                        where clock_timestamp() < $1
                        union
                        select i + 1, (', sql_expr, ') is null
                        from r
                        where clock_timestamp() < $1
                   )
                   select coalesce(max(i), 0) from r');
    EXECUTE sql USING timeout + clock_timestamp() INTO loop_count;
    RETURN loop_count;
END
$$;

comment on function public.benchmark(timeout interval, sql_expr text) is $$
    Measures the speed of expressions and functions for a given period of time. Returns number of calls.
$$;

------------------------------------------------------------------------------------------------------------------------
-- TESTS
do $$
    begin
        assert public.benchmark(1000, 'gen_random_uuid()') > '0'::interval;
        assert public.benchmark('10ms'::interval, 'gen_random_uuid()') > 0;
    end;
$$;

-- EXAMPLE 1: parse URL
select public.benchmark(100000, $$substring(format('https://www.domain%s.com/?aaa=1111&b[2]=3#test', (random()*1000)::int::text) from '^[^:]+://([^/]+)')$$);

-- EXAMPLE 2: generate UUID
SELECT public.benchmark(100000, $$uuid_in(overlay(overlay(md5(random()::text || ':' || clock_timestamp()::text) placing '4' from 13) placing to_hex(floor(random()*(11-8+1) + 8)::int)::text from 17)::cstring)$$);
SELECT public.benchmark(100000, $$md5(random()::text || clock_timestamp()::text)::uuid$$);

-- EXAMPLE 3: benchmark generate UUID
SELECT public.benchmark('1s'::interval, 'public.gen_random_uuid()'), public.gen_random_uuid() as guid
union all
SELECT public.benchmark('1s'::interval, 'public.uuid_generate_v7()'), public.uuid_generate_v7()
union all
SELECT public.benchmark('1s'::interval, 'public.uuid_generate_v8()'), public.uuid_generate_v8();