cartodb4/cartodb-postgresql
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Let CDB_CheckQuota second parameter override configured user quota

Browse Source 
Closes #13
Includes automated test
This commit is contained in:
Sandro Santilli
2014-05-09 12:55:41 +02:00
parent 1643ae2db1
commit b69ded8f66
3 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
scripts-available/CDB_Quota.sql
View File

@@ -35,12 +35,17 @@ BEGIN
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';
qmax := 0;
END;
IF TG_NARGS > 1 THEN
RAISE NOTICE 'Using quota specified via trigger parameter';
qmax := TG_ARGV[1];
ELSE
BEGIN
qmax := public._CDB_UserQuotaInBytes();
EXCEPTION WHEN undefined_function THEN
RAISE WARNING 'Missing _CDB_UserQuotaInBytes(), assuming no quota';
qmax := 0;
END;
END IF;
IF qmax = 0 THEN
RETURN NEW;