Make cdb_enough_quota check the table got from cdb_service_params #302

This commit is contained in:
Rafa de la Torre
2016-11-10 19:23:39 +01:00
parent 603fbbbc3f
commit 9f9034f4f1

View File

@@ -59,13 +59,15 @@ $$ LANGUAGE plpythonu;
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_enough_quota(
username TEXT,
orgname TEXT,
service TEXT,
service_ cdb_dataservices_server.service_type,
input_size NUMERIC)
returns BOOLEAN AS $$
DECLARE
params cdb_dataservices_server.service_params;
BEGIN
SELECT * INTO params FROM cdb_dataservices_server.cdb_service_params(username, orgname, service);
SELECT * INTO params
FROM cdb_dataservices_server.cdb_service_params(username, orgname) AS p
WHERE p.service = service_;
RETURN params.soft_limit OR ((params.used_quota + input_size) <= params.monthly_quota);
END
$$ LANGUAGE plpgsql;