b0c1948c14
- Instead of modifying cdb_service_quota_info to return max_batch_size, a new type (service_quota_info_batch) and a new function (cdb_service_quota_info_batch) are created. That makes deployment safe. - Fixes geocoding with forced batch size 1. - Improves namespacing for count_estimate (-> cdb_dataservices_client.cdb_count_estimate). - Improves namespacing for jsonb_array_casttext (-> cdb_dataservices_client.cdb_jsonb_array_casttext).
20 lines
703 B
PL/PgSQL
20 lines
703 B
PL/PgSQL
-- Taken from https://wiki.postgresql.org/wiki/Count_estimate
|
|
CREATE FUNCTION cdb_dataservices_client.cdb_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 cdb_dataservices_client.cdb_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; |