Add soft limit function #302

This commit is contained in:
Rafa de la Torre
2016-11-10 11:09:40 +01:00
parent 0e38af6ecd
commit 3e059003b6

View File

@@ -37,3 +37,20 @@ RETURNS integer AS $$
else:
raise 'not implemented'
$$ LANGUAGE plpythonu;
CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_soft_limit(
username TEXT,
orgname TEXT,
service TEXT)
RETURNS boolean 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.soft_isolines_limit
else:
raise 'not implemented'
$$ LANGUAGE plpythonu;