Add cdb_monthly_quota (WIP) #302

This commit is contained in:
Rafa de la Torre
2016-11-08 18:29:06 +01:00
parent 2c76fa4501
commit 6f9feb07a0

View File

@@ -0,0 +1,15 @@
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_monthly_quota(
username TEXT,
orgname TEXT,
service TEXT)
RETURNS integer AS $$
plpy.execute("SELECT cdb_dataservices_server._connect_to_redis('{0}')".format(username))
redis_conn = GD["redis_connection_{0}".format(username)]['redis_metrics_connection']
if service == 'isolines':
plpy.execute("SELECT cdb_dataservices_server._get_isolines_routing_config({0}, {1})".format(plpy.quote_nullable(username), plpy.quote_nullable(orgname)))
user_isolines_config = GD["user_isolines_routing_config_{0}".format(username)]
return user_isolines_config.isolines_quota
else:
raise 'not implemented'
$$ LANGUAGE plpythonu;