20 lines
647 B
PL/PgSQL
20 lines
647 B
PL/PgSQL
-- Taken from https://wiki.postgresql.org/wiki/Count_estimate
|
|
CREATE FUNCTION count_estimate(query text) RETURNS INTEGER AS
|
|
$func$
|
|
DECLARE
|
|
rec record;
|
|
ROWS INTEGER;
|
|
BEGIN
|
|
FOR rec IN EXECUTE 'EXPLAIN ' || query LOOP
|
|
ROWS := SUBSTRING(rec."QUERY PLAN" FROM ' rows=([[:digit:]]+)');
|
|
EXIT WHEN ROWS IS NOT NULL;
|
|
END LOOP;
|
|
|
|
RETURN ROWS;
|
|
END
|
|
$func$ LANGUAGE plpgsql;
|
|
|
|
-- Taken from https://stackoverflow.com/a/48013356/351721
|
|
CREATE OR REPLACE FUNCTION jsonb_array_casttext(jsonb) RETURNS text[] AS $f$
|
|
SELECT array_agg(x) || ARRAY[]::text[] FROM jsonb_array_elements_text($1) t(x);
|
|
$f$ LANGUAGE sql IMMUTABLE; |