From 6f9feb07a038687bfa27edc8bd1762df93e50b12 Mon Sep 17 00:00:00 2001 From: Rafa de la Torre Date: Tue, 8 Nov 2016 18:29:06 +0100 Subject: [PATCH] Add cdb_monthly_quota (WIP) #302 --- server/extension/sql/200_quotas.sql | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 server/extension/sql/200_quotas.sql diff --git a/server/extension/sql/200_quotas.sql b/server/extension/sql/200_quotas.sql new file mode 100644 index 0000000..a2e627f --- /dev/null +++ b/server/extension/sql/200_quotas.sql @@ -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;