15740f2ef6
NOTE: also makes CDB_TableMetadata triggers still functional
when user quota and varnish invalidation functions are missing
from the database (currently installed by cartodb ruby)
56 lines
1.6 KiB
PL/PgSQL
56 lines
1.6 KiB
PL/PgSQL
-- Return the estimated size of user data. Used for quota checking.
|
|
CREATE OR REPLACE FUNCTION CDB_UserDataSize()
|
|
RETURNS bigint AS
|
|
$$
|
|
-- TODO: double check this query. Maybe use CDB_TableMetadata for lookup ?
|
|
-- also, it's "table_name" sounds sensible to search_path
|
|
--
|
|
-- NOTE: division by 2 is an hack for the_geom_webmercator
|
|
--
|
|
SELECT coalesce(int8(sum(pg_total_relation_size(quote_ident(table_name))) / 2), 0)
|
|
AS quota
|
|
FROM information_schema.tables
|
|
WHERE table_catalog = current_database() AND table_schema = 'public'
|
|
AND table_name != 'spatial_ref_sys'
|
|
AND table_name != 'cdb_tablemetadata'
|
|
AND table_type = 'BASE TABLE';
|
|
$$
|
|
LANGUAGE 'sql' VOLATILE;
|
|
|
|
CREATE OR REPLACE FUNCTION CDB_CheckQuota()
|
|
RETURNS trigger AS
|
|
$$
|
|
DECLARE
|
|
|
|
pbfact float8;
|
|
qmax int8;
|
|
dice float8;
|
|
quota float8;
|
|
BEGIN
|
|
|
|
pbfact := TG_ARGV[0];
|
|
dice := random();
|
|
|
|
-- RAISE DEBUG 'CDB_CheckQuota enter: pbfact=% dice=%', pbfact, dice;
|
|
|
|
IF dice < pbfact THEN
|
|
RAISE DEBUG 'Checking quota on table % (dice:%, needed:<%)', TG_RELID::text, dice, pbfact;
|
|
BEGIN
|
|
qmax := public._CDB_UserQuotaInBytes();
|
|
EXCEPTION WHEN undefined_function THEN
|
|
RAISE WARNING 'Missing _CDB_UserQuotaInBytes(), assuming no quota';
|
|
RETURN NEW;
|
|
END;
|
|
SELECT CDB_UserDataSize() INTO quota;
|
|
IF quota > qmax THEN
|
|
RAISE EXCEPTION 'Quota exceeded by %KB', (quota-qmax)/1024;
|
|
ELSE RAISE DEBUG 'User quota in bytes: % < % (max allowed)', quota, qmax;
|
|
END IF;
|
|
-- ELSE RAISE DEBUG 'Not checking quota on table % (dice:%, needed:<%)', TG_RELID::text, dice, pbfact;
|
|
END IF;
|
|
|
|
RETURN NEW;
|
|
END;
|
|
$$
|
|
LANGUAGE 'plpgsql' VOLATILE;
|