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;